问题描述
对于一个字符串S ,我们定义 S 的分值f(S) 为 S中恰好出现一次的字符个数。例如 f(“aba”)=1,f(“abc”)=3, f(“aaa”)=0。
现在给定一个字符串 S(长度为n ),请你计算对于所有 S的非空子串S[i…j] (0<=i<=j<n),f[i…j]的和是多少。
输入格式:
输入一行包含一个由小写字母组成的字符串S 。
输出格式:
输出一个整数表示答案。
样例输入
ababc
样例输出
21
分析:
对于字符串中的每个字符,我们可以考虑它对答案的贡献度,即 它 所能在的所有的字符子串中,它是唯一的个数,设某个字符位置位a,和它相同的前一个字符位置为b,和它相同的后一个字符位置为c,那么他的贡献度为(a-b)*(c-a),其中a-b种取法是在a前面可以取0-a-b-1个字符,c-a取后面的取法,两者相乘。
为了加快时间,我们预处理每个字符的上一个位置和下一个位置。
参考代码:
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=1e5+5,M=1e5+5;
int pre[N],ne[N],ans;
int last[30];//记录某字母最后一次出现的位置
int main(){
string s;
cin>>s;
int len=s.length();
for(int i=0;i<26;i++) last[i]=-1;//前面没有就是-1,这样0-(-1)是1
for(int i=0;i<len;i++){
pre[i]=last[s[i]-'a'];//i下标的前面一个和自己相同的字符是之前最后的此字符位置
last[s[i]-'a']=i; //更新s[i]字符最后一次出现的位置,-'a'是节约空间
}
for(int i=0;i<26;i++) last[i]=len;//后面没有就是len
//为什么要倒着推,因为我们要知道x字符的下一个位置,必然需要提前知道他后面的
for(int i=len-1;i>=0;i--){
ne[i]=last[s[i]-'a'];//i下标的前面一个和自己相同的字符是之前最后的此字符位置
last[s[i]-'a']=i; //更新s[i]字符最后一次出现的位置
}
for(int i=0;i<len;i++)
ans+=(i-pre[i])*(ne[i]-i);
//i点的上一个相同字符距离x,下一个位y,
//左边可以取x种(1-x),右边y种,共x*y
cout<<ans;
return 0;
}