计蒜客习题 旋转数字 拓展KMP

计蒜客习题 旋转数字

问题描述

蒜头君发现了一个很好玩的事情，他对一个数作旋转操作，把该数的最后的数字移动到最前面。比如，数 123 可以得到 312, 231,123这样就可以得到很多个数。 
现在，蒜头君的问题是这些数中，有多少个不同的数小于原数，多少个等于原数，多少个大于原数。 
旋转中可能会出现前导零，两数比较的时候可以忽略前导零的影响。 
输入格式 
输入一个整数 N(0 < N <1e100000)。 
输出格式 
答案在一行中输出三个整数，分别是小于 NN，等于 NN，大于 NN 的个数，中间以空格隔开。 
样例输入 
3 4 1 
样例输出 

1 1 1

题意很简单 主要是注意用拓展kmp求出每个后缀前缀这样你可以直接判断第几个大小就行了

遇到周期串用下KMP的NEXT数组判断你需要弄几个 因为你遇到周期串组成的数字会重复

#include <cstdio>
#include <iostream>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <set>
using namespace std;
#define dbg(x) cout<<#x<<" = "<< (x)<< endl
const int MAX_N = 100024;
int Next[MAX_N],ex[MAX_N],NEXT[MAX_N];
char str[MAX_N],T[MAX_N];
void GetNext(int len){
    int i = 0,j = -1;
    while(i<len){
        if(j==-1||str[i]==str[j]){
            ++i;
            ++j;
            NEXT[i] = j;
        }
        else j = NEXT[j];
    }
    return;
}
void getnext(char *str){
    int i = 0,j,po,len = strlen(str);
    Next[0] = len;
    while(str[i]==str[i+1]&&i+1<len)
        i++;
    Next[1] = i;
    po = 1;
    for(i=2;i<len;++i){
        if(Next[i-po]+i<Next[po]+po)
            Next[i] = Next[i-po];
        else {
            j=Next[po]+po-i;
            if(j<0) j = 0;
            while(i+j<len&&str[j]==str[j+i])
                j++;
            Next[i] = j;
            po = i;
        }
    }
}
void exkmp(char *s1,char *s2){
    int i=0,j,po,len=strlen(s1),l2=strlen(s2);
    getnext(s2);
    while(s1[i]==s2[i]&&i<len)
        i++;
    ex[0]=i;
    po = 0;
    for(i=1;i<len;++i){
        if(Next[i-po]+i<ex[po]+po)
            ex[i] = Next[i-po];
        else {
            j = ex[po]+po-i;
            if(j<0) j = 0;
            while(i+j<len&&j<l2&&s1[j+i]==s2[j])
                j++;
            ex[i] = j;
            po = i;
        }
    }
}
int main(){
    int same=0,dayu = 0,xiaoyu=0;
    scanf("%s",str);
    memcpy(T,str,sizeof(str));
    exkmp(str,T);
    int len = strlen(str);
    //printf("%s",str);
    NEXT[0] = -1;
    GetNext(len);
    int tmp;
    if(len%(len-NEXT[len])!=0) tmp = len;
    else tmp = len-NEXT[len];
    //dbg(tmp);
    for(int i=0;i<tmp;++i){
        //dbg(ex[i]);
        //dbg(str[i+ex[i]]);
        //dbg(T[ex[i]]);
        if(ex[i]==len) {same++;
        }
        else if(str[i+ex[i]]-'0'>T[ex[i]]-'0') {
                dayu++;
        }
        else if(str[i+ex[i]]-'0'<T[ex[i]]-'0') {
                xiaoyu++;
        }
    }
    printf("%d %d %d\n",xiaoyu,same,dayu);
    return 0;
}