problem 42

"""
The nth term of the sequence of triangle numbers is given by, tn = ½n(n+1); so the first ten triangle numbers are:

1, 3, 6, 10, 15, 21, 28, 36, 45, 55, ...

By converting each letter in a word to a number corresponding to its alphabetical position and adding these values we form a word value. For example, the word value for SKY is 19 + 11 + 25 = 55 = t10. If the word value is a triangle number then we shall call the word a triangle word.

Using words.txt (right click and 'Save Link/Target As...'), a 16K text file containing nearly two-thousand common English words, how many are triangle words?
"""
import math
def score(word):
  for letter in word:
    total = sum(ord(c) - 64 for c in word.strip('"'))
  n = int (math.sqrt(total * 2))
  check_if = (n * (n + 1)) / 2  
  if check_if == total:
    return True

total = 0
path = "words.txt"
words_file = open(path,'r')
words = words_file.read().split(',')
words.sort()
#print(score("SKY"))
for word in words:
 if score(word):
   total +=1
print(total)