p042.py

################################################################################
#                          P42: Coded triangle numbers                         #
################################################################################
#                                                                              #
#        Using p042_words.txt, how many given words are triangle words?        #
#                                                                              #
################################################################################
#                     Problem found at projecteuler.net                        #
#                              Author: ncfgrill                                #
################################################################################

tri_nums, base = {(n*(n+1)//2) for n in range(1, 101)}, ord('A') - 1

def get_total(word):
    s, i = 0, 1
    while word[i] != '"':
        s += ord(word[i]) - base
        i += 1
    return 1 if s in tri_nums else 0

with open('p042_words.txt', 'r') as f:
    print('Triangle words:', sum(map(get_total, f.readline().split(','))))