子串分值和【第十一届】【省赛】【研究生组】python

题目

资源限制
时间限制：1.0s 内存限制：256.0MB

问题描述
对于一个字符串 S ，我们定义 S 的分值 f(S) 为 S 中恰好出现一次的字符个数。例如 f(“aba”)=1，f(“abc”)=3, f(“aaa”)=0。
现在给定一个字符串 S[0…n-1]（长度为 n），请你计算对于所有 S 的非空子串 Si…j，f(S[i…j])的和是多少。

输入格式
输入一行包含一个由小写字母组成的字符串 。

输出格式
输出一个整数表示答案。

样例输入
ababc

样例输出
21

样例说明
子串 f值
a 1
ab 2
aba 1
abab 0
ababc 1
b 1
ba 2
bab 1
babc 2
a 1
ab 2
abc 3
b 1
bc 2
c 1

评测用例规模与约定
对于 20% 的评测用例，1<=n<=10 ；
对于 40% 的评测用例，1<=n<=100 ；
对于 50% 的评测用例，1<=n<=1000 ；
对于 60% 的评测用例，1<=n<=10000 ；
对于所有评测用例，1<=n<=100000 。

代码

S = input()

sum = 0
for i in range(len(S)):
    letter = [0] * 200
    count = 0
    for j in range(i,len(S)):
        loc = ord(S[j])
        # 如果该字符还未出现
        if  letter[loc] == 0:
            count += 1
            # 把该处标记为1
            letter[loc] = 1
        # 如果该字符只出现了1次
        elif letter[loc] == 1:
            count -= 1
            letter[loc] += 1
        # 如果该字符出现了2次以上，不影响计数
        else:
            letter[loc] += 1
        sum += count
print(sum)

双层for循环，能够预料到会超时。。。

评测情况