【问题描述】
对于一个字符串 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
1
【样例说明】
子串 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。
一开始的思路:
1. 将字母转换为下标(a=0,b=1……),设置一个bool数组用来记录字母是否已经出现
2. f变量用来记录每一轮中的f值
3. 设立双层for循环,第一层循环确立每一轮的起始指针 ,第二层循环从起始指针开始一直遍历到结尾。每一轮都要重置a数组和f变量
代码:
这段代码只过了60%的数据,剩下的超时
#include<iostream>
#include<cstring>
using namespace std;
int main(){
string s = "";
cin >> s;
bool a[26] = {0};
int ans = 0;
for(int i = 0;i < s.length();i++){
memset(a,0,sizeof(a));
int f = 0;
for(int j = i;j < s.length();j++){
int k = s[j] - 97;
if(a[k] == 0){//如果该字母没有出现过
a[k] = 1;
f++;
}
ans += f;
}
}
cout << ans;
return 0;
}
修改的时候发现了规律:
代码:
#include<iostream>
#include<cstring>
using namespace std;
int main(){
string s = "";
cin >> s;
s = "0" + s;//下标从1开始
int a[27] = {0};//更新字母位置
long long ans = 0;
for(int i = 1;i <= s.length();i++){
int k = s[i]-96;//a代表1,b代表2……
ans += (i-a[k])*(s.length()-i);
a[k] = i;
}
cout << ans;
return 0;
}