2020 C++ A组——H题 子串分值

题目描述

对于一个字符串 SS，我们定义 SS 的分值 f(S)f(S) 为 SS 中恰好出现一次的字符个数。例如 f(aba) = 1，f(abc) = 3, f(aaa) = 0f(aba)=1，f(abc)=3,f(aaa)=0。

现在给定一个字符串 S_{0 \cdots n − 1}S0⋯n−1​（长度为 nn，1 \leq n \leq 10^51≤n≤105），请你计算对于所有 SS 的非空子串 S_{i \cdots j}(0 ≤ i ≤ j < n)Si⋯j​(0≤i≤j<n)，f(S_{i \cdots j})f(Si⋯j​) 的和是多少。

输入描述

输入一行包含一个由小写字母组成的字符串 SS。

输出描述

输出一个整数表示答案。

输入输出样例

示例

输入

ababc

输出

21

运行限制

• 最大运行时间：1s
• 最大运行内存: 256M

解题思路

这道题可以用暴力解法，但是本题的本意并不是在这里。这里其实是求字母的贡献度，只有当字母个数在子串中个数为1才会有贡献度。

我们可以从字母a的左右两边出发，分别计算移动了多远才会出现一个字母与a相同，然后停止遍历，记录下步长pre和suf。然后是算总的贡献度。

当前字母的贡献度=(pre+1) * (suf+1)

代码

通过率40%的情况

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e5+10;
char s[N];//存读入的字符
int main()
{
	cin>>(s + 1);
	int n = strlen(s+1),l,i,r;//l:左端点，r:右端点
	int cnt[26];
	ll ans = 0;
	for(l=1;l<=n;l++)//枚举左端点 
	{
		for(r=l+1;r<=n+1;r++)//枚举右端点 
		{
			memset(cnt,0,sizeof(cnt));
			for(i=l;i<r;i++)
			    cnt[s[i]-'a']++;
			for(i=0;i<26;i++)
			    ans+=(cnt[i]==1);
		}
	 } 
	 cout<<ans<<endl;
	 return 0;
} 

通过率100%

#include<bits/stdc++.h> 
using namespace std;
typedef long long ll;
const int N = 1e5 + 10;
char s[N];//存储读入的字符串
ll pre[N];//pre(i)表示第i个字符上一次出现的位置
ll suf[N];//suf(i)表示第i个字符下一次出现的位置
ll last[30];//辅助数组,两组递归公用 
int main()
{
	cin>>(s+1);//从s[1]开始存
	int n = strlen(s + 1),ans = 0;
	for(int i=1;i<=n;i++)
	{
		pre[i] = last[s[i]-'a'];
		last[s[i]-'a'] = i;
	} 
	
	for(int j=0;j<=25;j++)
        last[j] = n+1;//初始化 
	for(int i=n;i>=1;i--)
	{
		suf[i]=last[s[i]-'a'];
		last[s[i]-'a'] = i ;
	} 
	for(int i=1;i<=n;i++)
	{
		ans+=(i-pre[i])*(suf[i]-i);
	}
	cout<<ans<<endl;
}