【问题描述】
对于一个字符串 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。
【输出格式】
输出一个整数表示答案。
【样例输入】
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
【评测用例规模与约定】
对于 20% 的评测用例,1 ≤ n ≤ 10;
对于 40% 的评测用例,1 ≤ n ≤ 100;
对于 50% 的评测用例,1 ≤ n ≤ 1000;
对于 60% 的评测用例,1 ≤ n ≤ 10000;
对于所有评测用例,1 ≤ n ≤ 100000。
题解:补一波蓝桥杯的题吧,当时O(n^2)写的,直接暴力没啥好分享的,有手就行。然后分享一下O(n)的解法,涉及到递推算之前字符给予的奉献,因为感觉手写比较清晰,也方便自己之后的复习,就直接手写,如下图(字可能有点辣眼睛):
太难了,要不是有大佬指点,这根本想不到嗷,自己也琢磨挺久的,不过u1s1,这题出的还挺有意思。
代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 1e6 + 7;
int vis[N], next[N];
LL ans = 0, sum = 0, now = 0;
string str;
int main()
{
ios::sync_with_stdio(false);
cin >> str;
memset(vis, -1, sizeof(vis));
memset(next, 0, sizeof(next));
for(int i = 0; i < str.length(); i++)
{
if(vis[str[i] - 'a'] == -1)
{
now++;
sum += now;
vis[str[i] - 'a'] = i;
}
else
{
sum += now;
int lastPos = vis[str[i] - 'a'];
next[lastPos] = i - vis[str[i] - 'a']; //重复出现的字符离最右边相同字符的距离
vis[str[i] - 'a'] = i;
}
}
// cout << sum << endl;
ans += sum;
int n = str.length();
for(int i = 1; i < str.length(); i++)
{
if(!next[i-1]) //第i-1个字符是唯一的
{
sum = sum - (n - (i - 1));
ans += sum;
}
else
{
int dis = next[i-1]; //算出第i-1个字符与它相同的下一个字符的距离
sum = sum - dis;
ans += sum;
}
}
cout << ans << endl;
return 0;
}