String
Accepts: 84
Submissions: 373
Time Limit: 2000/1000 MS (Java/Others)
Memory Limit: 65536/65536 K (Java/Others)
问题描述
有一个 10≤10\leq10≤长度≤1,000,000\leq 1,000,000≤1,000,000 的字符串,仅由小写字母构成。求有多少个子串,包含有至少k(1≤k≤26)k(1 \leq k \leq 26)k(1≤k≤26)个不同的字母?
输入描述
输入包含多组数据. 第一行有一个整数T(1≤T≤10)T (1\leq T\leq 10)T(1≤T≤10), 表示测试数据的组数. 对于每组数据:
第一行输入字符串SSS。
第二行输入一个整数kkk。
输出描述
对于每组数据,输出符合要求的子串的个数。
输入样例
2
abcabcabca
4
abcabcabcabc
3
输出样例
0
55
思路
有一个明显的性质:如果子串(i,j)(i,j)(i,j)包含了至少kkk个不同的字符,那么子串(i,k),(j
#include<iostream>
#include<stdio.h>
using namespace std;
int vis[30];
int T;
char s[1000050];
int k;
long long ans;
void solve()
{
ans=0;
int cnt=0;
int l,r=1;
int len=strlen(s);
vis[s[0]-'a']++;
cnt=1;
for(int l=1;l<=len;l++)
{
while(cnt!=k)
{
if(r>=len) return;
r++;
if(!vis[s[r-1]-'a']) cnt++;
vis[s[r-1]-'a']++;
}
ans+=len-r+1;
vis[s[l-1]-'a']--;
if(!vis[s[l-1]-'a'])
cnt--;
}
return ;
}
int main()
{
scanf("%d",&T);
while(T--)
{
scanf("%s",s);
scanf("%d",&k);
memset(vis,0,sizeof(vis));
solve();
printf("%I64d\n",ans);
}
}