对于一个字符串 S,我们定义 S 的分值 f(S) 为 S
中出现的不同的字符个数。
例如 f(“aba”)=2,f(“abc”)=3,f(“aaa”)=1
。
现在给定一个字符串 S[0..n−1]
(长度为 n),请你计算对于所有 S 的非空子串 S[i..j](0≤i≤j<n),f(S[i..j])
的和是多少。
输入格式
输入一行包含一个由小写字母组成的字符串 S
。
输出格式
输出一个整数表示答案。
数据范围
对于 20%
的评测用例,1≤n≤10;
对于 40% 的评测用例,1≤n≤100;
对于 50% 的评测用例,1≤n≤1000;
对于 60% 的评测用例,1≤n≤10000;
对于所有评测用例,1≤n≤100000
。
输入样例:
ababc
输出样例:
28
样例解释
子串 f值
a 1
ab 2
aba 2
abab 2
ababc 3
b 1
ba 2
bab 2
babc 3
a 1
ab 2
abc 3
b 1
bc 2
c 1
法一
分析:
考虑每个字符在多少种子串中有价值,即在多少种子串中时第一次出现,第一次出现可能出现在任何位置(开头,中间,最后)
用last[]来记录他最后出现的位置,到第i个字符时,包含它的子串的起点可能i-last[s[i]中的任何一个,起点有i-last[s[i]]种,终点可能是包含他以及它后面的任何位置,终点有n-i+1种,则包含他的子串有sum=(i-last[s[i]])*(n-i+1)种,则它在sum个子串中每个提供一点价值
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;
string s;
int last[1010];
int main()
{
cin>>s;
int n=s.size();
s=' '+s;
LL ans=0;
for(int i=1;i<=n;i++)
{
ans+=(LL)(i-last[s[i]])*(n-i+1);
last[s[i]]=i;
}
cout<<ans;
return 0;
}
法二
dp
f[i]:表示以s[i]结尾的子串
f[i]=f[i-1]+i-pos[s[i]]
在s[i]以前不包含s[i]的只有i-pos[s[i]]个起点,终点确定,所以要加上i-pos[s[i]]
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;
char s[101000];
LL f[101000];
int pos[1010];
int main()
{
scanf("%s",s+1);
int n=strlen(s+1);
LL ans=0;
for(int i=1;i<=n;i++)
{
f[i]=f[i-1]+i-pos[s[i]];
pos[s[i]]=i;
ans+=f[i];
}
cout<<ans;
return 0;
}