No42_Coded triangle numbers

题目：

The n^th term of the sequence of triangle numbers is given by, t_n = ½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 = t₁₀. 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?

代码为：

#coding=utf-8import io,math,timet=time.time()wordlist={word  for word in  io.open(r'two.txt','r').read()[1:-2].split('","')}matchedword=set()for word in wordlist:    lettersum=sum([ord(letter)-ord('A')+1 for letter in word])#计算和    root=int(math.sqrt(lettersum*2))    if root*root+root==lettersum*2:#计算是否满足条件        #print word,lettersum        matchedword.add(word)"""wordwithsum_list=[(word,sum([ord(letter)-ord('A')+1 for letter in word]))  for word in matchedword]wordwithsum_list.sort(key=lambda x:x[1])for word,tempsum in wordwithsum_list:    print word,tempsum"""print len(matchedword)#print int(1.9)print time.time()-t

运行结果为：
162
0.00800013542175