String
Accepts: 84
Submissions: 373
Time Limit: 2000/1000 MS (Java/Others)
Memory Limit: 65536/65536 K (Java/Others)
问题描述
有一个 10≤长度≤1,000,000 的字符串,仅由小写字母构成。求有多少个子串,包含有至少k(1≤k≤26)个不同的字母?
输入描述
输入包含多组数据. 第一行有一个整数T(1≤T≤10), 表示测试数据的组数. 对于每组数据: 第一行输入字符串S。 第二行输入一个整数k。
输出描述
对于每组数据,输出符合要求的子串的个数。
输入样例
2 abcabcabca 4 abcabcabcabc 3
输出样例
0 55
Statistic | Submit | Clarifications | Back
当时写的复杂度n*n超时了,这个是o(n)的复杂度,控制两个下标,当j满足sun>=n时,i往前挪,直到sun<n,j继续往前挪。
注意结果很大,用long long
代码:
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
char a[1000010];
int main()
{
int t,n;
scanf("%d",&t);
while(t--)
{
scanf("%s",a);
scanf("%d",&n);
int len=strlen(a);
int i=0,j=0,sun=0,num[30],p;
long long sum=0;
memset(num,0,sizeof(num));
while(1)
{
while(sun<n&&j<len)
{
p=a[j++]-97;
if(num[p]==0)
sun++;
num[p]++;
}
if(sun<n)
break;
sum+=(len-j+1);
p=a[i++]-97;
num[p]--;
if(num[p]==0)
sun--;
}
printf("%I64d\n",sum);
}
}