题意分析:

             给你一个数字,要求从右往左依次吧数字往前面放,求新的数字比原来数字小的个数,相等的个数,还有大于的个数。如果用暴力的话坑定会超时。如果熟悉扩展KMP算法的话可能会想到吧这个数字看成一个字符串S,然后把这个字符串扩展成2倍T,在依次求出扩展后的字符串的后缀依次与原来字符串前缀的公共长度。判断条件是当extend【i】大于等于原来字符串的长度len时,即extend[i]  >=len,说明新的数字等于原来数,如果T【i+extend[i]】> S【extend[i]】时,说明大于。或者小于。 还有一个容易忽略的问题是S可能有最小循环节,这样就可能的出来的数字重复,所以要去重。

代码如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#define MA 1000000
using namespace std;
int next[MA],extend[MA],nextval[MA];

void getNext(char s[]) {
    int len=strlen(s),a=0;
    next[0]= len;
    while(a<len-1 && s[a] == s[a+1]) a++;
    next[1]  =a;
    a = 1;
    for(int k=2; k<len; k++) {
        int p=a+next[a] -1,L=next[k-a];
        if(k-1+L>=p) {
            int j=p+1-k>0? p+1-k: 0;
            while(k+j<len && s[k+j] == s[j]) j++;
            next[k]  = j;
            a = k;
        }
        else next[k]  = L;
    }
}

void getExtend(char s[],char T[]) {
    int len=strlen(s),lenp=strlen(T),a=0;
    int mi = min(len,lenp);
    while(a<mi && s[a] == T[a]) a++;
    extend[0] = a;
    a=0;
    getNext(T);
    for(int k=1; k<len; k++) {
        int p=a+extend[a] -1,L=next[k-a];
        if(k-1+L>=p) {
            int j=p+1-k>0? p+1-k : 0;
            while(j+k<len&&j<lenp &&s[j+k] == T[j]) j++;
            extend[k] = j;
            a = k;
        }
        else extend[k]  = L;
    }
}

void nextVal(char s[]) {
    int len=strlen(s),j=-1,i=0;
    nextval[0] = -1;
    while(i<len) {
        if(j==-1 || s[i] == s[j]) {
            ++i; ++j;
            if(s[i]!=s[j])
               nextval[i] = j;
            else nextval[i]  = nextval[j] ;
        }
        else j = nextval[j];
    }
}
int main() {
    int t;
    char s[MA],s1[MA];
    scanf("%d",&t);
    int ka=1;
    while(t--) {
        scanf("%s",s1);
        //cout<<t<<endl;
        strcpy(s,s1);
        strcat(s,s1);
        //cout<<s<<endl;
        nextVal(s1);
        int len=strlen(s1);
        int le=len-nextval[len],co=1;
        if(len%le==0) co = len/le;
        //cout<<next[len]<<endl;
        getExtend(s,s1);
        int su1=0,su2=0,su3=0;
        for(int i=0; i<len; i++) {
            if(extend[i] >=len) su1++;
            else if(s[i+extend[i]]>s1[extend[i]]) su2++;
            else su3++;
        }
        printf("Case %d: %d %d %d\n",ka++,su3/co,su1/co,su2/co);
    }
    return 0;
}

Logo

开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!

更多推荐