度度熊找子串（百度2017秋招真题）

题目描述
度度熊收到了一个只有小写字母的字符串S，他对S的子串产生了兴趣，S的子串为S中任意连续的一段。他发现，一些子串只由一种字母构成，他想知道在S中一共有多少种这样的子串。
例如在串”aaabbaa”中，度度熊想找的子串有”a”,”aa”,”aaa”,”b”,”bb”五种。

代码（使用set.add()只添加不重复的元素）

str=input()
s=''
s1=''
res=set()#set不包含重复的内容
for ele in str:
    if ele==s1:#当后一个元素等于前一个元素时，
        s=s+ele
    else:
        s=ele#罗列结果需要的字串
        s1=ele#用于标记前一个元素
    res.add(s)

print(len(res))