hdu 5056 Boring count 尺缩法

由于字符串之间可以重复

所以区别字符串的只有起点终点

容易想到尺缩法

对区间[i,j]，设i为满足条件的最左的i

那么答案就加上[i, j]区间的长度j-i+1

如果区间右端向前进一步，j=j+1

此时当前字母的频数>k，则让i向前

直到当前字母频数<=k

#include <iostream>
#include <cstring>
#include <string>
#include <algorithm>
#include <cmath>
#include <vector>
#include <set>
#include <queue>
#include <map>
using namespace std;
#define INF 1e9
#define maxn 100010
#define rep(i,x,y) for(int i=x;i<=y;i++)
#define mset(x) memset(x,0,sizeof(x))
#define idx(x) x-'a'
typedef __int64 ll;

int t, k;
char s[maxn];
int	cc[27];

int main(){
//	freopen("a.txt","r",stdin);
//	freopen(".out","w",stdout);
	cin>>t;
	while(t--){
		scanf("%s", s);
		cin>>k;

		mset(cc);
		int len = strlen(s);
		ll ans=0;

		for(int j=0, i=0; j<len; j++){
			cc[idx(s[j])]++;
			
			while(i<j && cc[idx(s[j])]>k){
				cc[idx(s[i])]--; 
				i++;
			}

			ans = ans + j-i+1;
		}
		cout<<ans<<endl;
	}
	return 0;
}