题意:给一串字符串,求有多少个子串,每个子串最少包含k个不同字符
思路:第一眼就知道是尺取啦,定L,移动R,当L-R中有k个不同的字符时停下,此时从L开始满足的字符串有len-R+1个,L从0到len遍历(证明:如字符串asdfghbvasd,k=3,当L=0,R=2时刚好满足有k个不同的字符,此时R不管是在第2位还是在第3-->len位都满足条件,每多加一个字符都算一种额外的情况,所以是len-R+1)。第一眼看到题目的时候是不是默默定义了一个vis[N]数组呢(反正我是的QAQ),呐,这不就派上用场啦:将每一个出现过的字符标记一下,这样r就不用每次都从头开始啦,只需要找没被标记过的就好啦~
ps:每次移动L的时候都要特判一下哦
代码奉上:
#include<stdio.h>
//#include<bits/stdc++.h>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<map>
#include<stack>
#include<vector>
#include<queue>
using namespace std;
typedef long long ll;
const int N=1e6+10;
const int inf=0x3f3f3f3f;
char a[N];
ll vis[N];
int main()
{
int t;
//cin>>t;
scanf("%d",&t);
while(t--)
{
memset(vis,0,sizeof(vis));
//cin>>a;
scanf("%s",a);
ll n;
//cin>>n;
scanf("%lld",&n);
int len=strlen(a);
int l=0;
int r=0;
ll k=0,ans=0;
while(l<len)
{
while(r<len&&k<n) //k表示找到的不同的字符数量 n表示题目中要求的不同字符数量
{
if(vis[a[r]-'a']==0)
{
k++;
}
vis[a[r]-'a']++;
r++;
}
if(k<n) //如果遍历全部仍不能达到要求则break,最后输出0
{
break;
}
ans+=len-r+1; //每从l到r找到符合要求的最短子序列,则以l-len的子序列共有len-r+1种都成立
//l的特判
//以下四行是为了判断在去除a[l]的情况下,已经找到的最短子序列是否还符合要求
//如果不符合,会继续开始找符合条件的·1最短子序列,反之则不用找,直接求解这种情况就好
vis[a[l]-'a']--;
if(vis[a[l]-'a']==0)
{
k--;
}
l++;
}
//cout<<ans<<endl;
printf("%lld\n",ans);
}
}