蓝桥杯2020届C++B组省赛真题 子串分值

【问题描述】
对于一个字符串 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;
}