问题描述
有一个 10≤长度≤1,000,000 的字符串,仅由小写字母构成。求有多少个子串,包含有至少k(1≤k≤26)个不同的字母?
输入描述
输入包含多组数据. 第一行有一个整数T(1≤T≤10), 表示测试数据的组数. 对于每组数据: 第一行输入字符串S。 第二行输入一个整数k。
输出描述
对于每组数据,输出符合要求的子串的个数。
输入样例
2 abcabcabca 4 abcabcabcabc 3
输出样例
0 55解题思路:尺取法,st表示子串的起点,en表示终点,len为给你的字符串的长度
1.先将en向后查找,当查找到字符子串有k个不同字母时停止,答案数加上len-en+1
2.然后再将st向后推,st每向后推一位如果st到en的子串仍有k个不同字母,则答案数再加len-en+1,否则把en往后推。
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
char str[1000005];
int main()
{
int T;
scanf("%d", &T);
while(T--)
{
int n;
scanf("%s%d", str, &n);
int cou[26] = {0};
int len = strlen(str);
int st = 0, en = 0, num = 0;
long long ans = 0;
while(true)
{
while(en < len && num < n)
{
if(cou[str[en++]-'a']++ == 0)
num++;
}
if(num < n)
break;
ans += len - en + 1;
if(--cou[str[st++]-'a'] == 0)
num--;
}
printf("%I64d\n", ans);
}
return 0;
}