You are given a string S consisting of lowercase letters, and your task is counting the number of substring that the number of each lowercase letter in the substring is no more than K.
Input
In the first line there is an integer T , indicates the number of test cases.
For each case, the first line contains a string which only consist of lowercase letters. The second line contains an integer K.
[Technical Specification]
1<=T<= 100
1 <= the length of S <= 100000
1 <= K <= 100000
Output
For each case, output a line contains the answer.
Sample Input
3
abc
1
abcabc
1
abcabc
2
Sample Output
6
15
21
题意:给你一个字符串,字符串只由小写字母组成。问有多少子串满足各个小写字母的个数不超过k。
题解:做这题之前首先明白一个规律:对于字符串“abc”,如果加一个字母“d”,使得字符串变为“abcd”,新的字符串比原来多出了3个子串,即对于一个长度为n的字符串,我们在后面多添一个字符,就会多出n个子串。
该题我们我们确定好选取子串的左端点l和右端点r,每多添一个字母,我们验证当前子串是否满足题意,若满足则答案加上r-l+1,不满足则左端点向右移,直至满足题意。
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
char s[100000+10];
int cnt[30];
int k;
bool check()
{
for(int i=0;i<=25;i++)
{
if(cnt[i]>k) return 0;
}
return 1;
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
memset(cnt,0,sizeof(cnt));
scanf("%s",s+1);
int n=strlen(s+1);
scanf("%d",&k);
long long ans=0;
for(int i=1,j=1;i<=n;i++)
{
cnt[s[i]-'a']++;
while(cnt[s[i]-'a']>k)
{
while(!check())
{
cnt[s[j]-'a']--;
j++;
}
}
ans+=i-j+1;
}
printf("%lld\n",ans);
}
return 0;
}