题意:找出字符串中所有字母出现次数不超过k次的子串数量。
分析:首先明确两点,一:一个字符串的子串数量等于其长度的累积和,即len*(len+1)/2;二:若该字符的一个子串满足条件,则该子串的所有子串都满足条件。知道了这两点,这道题就能做出来了。具体是 设两个指针start和end,代表子串的起始和终点,接着不断将end往后推动,直到子串不满足条件,此时再计算[start,end-1]之间有多少子串,接着将start推动到子串中使得该子串不满足条件的第一次出现的字符处,此时,为了避免有些答案可能重复(至于为什么,自己画图想想就明白了),再将答案减去[start,end]之间的子串数量(注意,此时的start已经是往后推动了的)。
其实很水的一题~~~~
代码如下:
#include <cstdio>
#include <stack>
#include <set>
#include <iostream>
#include <string>
#include <vector>
#include <queue>
#include <list>
#include <functional>
#include <cstring>
#include <algorithm>
#include <cctype>
#include <string>
#include <map>
#include <iomanip>
#include <cmath>
#define LL long long
#define ULL unsigned long long
#define SZ(x) (int)x.size()
#define MP(a, b) make_pair(a, b)
#define MS(arr, num) memset(arr, num, sizeof(arr))
#define PB push_back
#define F first
#define S second
#define ROP freopen("input.txt", "r", stdin);
#define MID(a, b) (a + ((b - a) >> 1))
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define lrt rt << 1
#define rrt rt << 1|1
#define root 1,n,1
#define BitCount(x) __builtin_popcount(x)
#define BitCountll(x) __builtin_popcountll(x)
#define LeftPos(x) 32 - __builtin_clz(x) - 1
#define LeftPosll(x) 64 - __builtin_clzll(x) - 1
const double PI = acos(-1.0);
const int INF = 1e7;
using namespace std;
const double eps = 1e-5;
const int MAXN = 300 + 10;
const int MOD = 1000007;
const double M=1e-8;
const int N=100100;
typedef pair<int, int> pii;
typedef pair<int, string> pis;
map<string,int> hs;
int n,m,vis[200];
char s[N];
LL get(int x,int y)
{
LL len=y-x+1;
return len>0?len*(len+1)/2:0;
}
void slove()
{
MS(vis,0);
int i,j,len=strlen(s);
LL ans=0;
for (i=j=0;i<len;i++) {
vis[s[i]]++;
if (vis[s[i]]>n) { //cout<<"->"<<j<<" "<<i<<endl;
ans+=get(j,i-1);
while (s[j]!=s[i]) vis[s[j++]]--;
vis[s[j++]]--;
ans-=get(j,i-1); // 避免一些答案重复
}
}
ans+=get(j,i-1);
printf("%lld\n",ans);
}
int main()
{
int i,j,T;
cin>>T;
while(T--)
{
scanf("%s",s);
scanf("%d",&n);
slove();
}
}