子串分值（蓝桥杯杂题，找规律）

题目描述

对于一个字符串 S，我们定义 S 的分值f(S) 为 S 中恰好出现一次的字符个数。例如 f(aba)=1，f(abc)=3,f(aaa)=0。

现在给定一个字符串 S0⋯n−1（长度为 n，1≤n≤10^5），请你计算对于所有 S 的非空子串 Si⋯j(0≤i≤j<n)，f(Si⋯j) 的和是多少。

输入描述

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

输出描述

输出一个整数表示答案。

输入输出样例

示例

输入

ababc

输出

21

运行限制

• 最大运行时间：1s

• 最大运行内存: 256M

解析

• 用 pre[i] 记录第 i 位置上的字母上一次出现的位置；

• 用 next[i] 记录第 i 位置上的字母下一次出现的位置；

• 往左最多能延伸到 pre[i] + 1，其到第 i 个字母一共有 i - pre[i] 个字母；

• 同理往右最多能延伸到 next[i]- 1，其到第 i 个字母一共有 next[i] - i 个字母；

• 二者相乘，就是该 i 位置上的字母被不同子串所包含的总次数。

代码

lst = list(input())
cnt = 0
pre = [-1 for i in range(len(lst))]
next = [len(lst) for i in range(len(lst))]

for i in range(1, len(lst)):  #这里注意从1开始，从0开始的话下一行0-1=-1，lst[-1]是最后一个
    for j in range(i-1, -1, -1):
        if lst[i] == lst[j]:
            pre[i] = j
            break  #只记录上一个相同字母的位置，找到一次后跳出循环

for i in range(len(lst)-1):
    for j in range(i+1, len(lst)):
        if lst[i] == lst[j]:
            next[i] = j
            break

for i in range(len(lst)):
    cnt += (i - pre[i]) * (next[i] - i)

print(cnt)