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-8
import io,math,time
t=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