题目描述
对于一个字符串 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 位置上的字母被不同子串所包含的总次数。
![](https://i-blog.csdnimg.cn/blog_migrate/b860199cc5e06b8134c0f219e5d4b3ec.jpeg)
![](https://i-blog.csdnimg.cn/blog_migrate/88a1d2d0c61aa4b74246f413821b6649.jpeg)
代码
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)