hdu 5056

来源:互联网 发布:太极软件 政务 编辑:程序博客网 时间:2024/05/17 03:13

     题意:找出字符串中所有字母出现次数不超过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) - 1const 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();    }}


0 0
原创粉丝点击