hdu 5671 String(求有多少个子串，包含有至少k(1≤k≤26)个不同的字母)

题意:

有一个 10≤长度≤1,000,000 的字符串，仅由小写字母构成。求有多少个子串，包含有至少k(1≤k≤26)个不同的字母？

思路:

如果l,r中正好包含有k个不同的子串,那么从l开始的包含k个不同字符的串有strlen(s)-r+1

所以我们只需要计算出每个l对应的正好包含k个不同字符的r便可以了,利用双指针便可以完成。

#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <ctime>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
#define INF 0x3f3f3f3f
#define inf -0x3f3f3f3f
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define mem0(a) memset(a,0,sizeof(a))
#define mem1(a) memset(a,-1,sizeof(a))
#define mem(a, b) memset(a, b, sizeof(a))
typedef long long ll;
const int maxn=1010000;
char s[maxn];
int C[27],len,k;

int main(){
    int _;
    scanf("%d",&_);
    while(_--){
        scanf("%s",s+1);
        scanf("%d",&k);
        mem0(C);
        len=strlen(s+1);
        int ed=0,count=0;
        ll ans=0;
        for(int i=1;i<=len;i++){
            if(count>=k){
                 C[s[i]-'a']--;
                 if(C[s[i]-'a']==0)
                     count--;
                ans+=(len-ed+1);
                continue;
            }
            ed++;
            for(;ed<=len;ed++){
                if(C[s[ed]-'a']==0)
                    count++;
                C[s[ed]-'a']++;
                if(count>=k)
                    break;
            }
            if(count<k||ed>len)
                break;
            ans+=(len-ed+1);
            C[s[i]-'a']--;
            if(C[s[i]-'a']==0)
                count--;
        }
        printf("%lld\n",ans);
    }
    return 0;
}