http://acm.hdu.edu.cn/showproblem.php?pid=4333
题意
一个数字,依次将第一位放到最后一位,问小于本身的数的个数及等于本身的个数和大于本身的个数,但是要注意重复的不再计算
分析
当这个串有循环节时才会出现重复串,用KMP的next数组来计算循环节:len-next[len]。最后将得到的答案除以循环节个数即可。然后用扩展KMP,求出主串的后缀与模式串的前缀的最长公共前缀,这里的模式串是原串,而主串则是将两个原串连接起来。利用extend数组,当extend[i]>=n时,说明与原串匹配。否则则比较S[extend[i]]和S[i+extend[i]]的大小(比较不匹配的第一个位置)。
#include<iostream> #include<cmath> #include<cstring> #include<queue> #include<vector> #include<cstdio> #include<algorithm> #include<map> #include<set> #define rep(i,e) for(int i=0;i<(e);i++) #define rep1(i,e) for(int i=1;i<=(e);i++) #define repx(i,x,e) for(int i=(x);i<=(e);i++) #define X first #define Y second #define PB push_back #define MP make_pair #define mset(var,val) memset(var,val,sizeof(var)) #define scd(a) scanf("%d",&a) #define scdd(a,b) scanf("%d%d",&a,&b) #define scddd(a,b,c) scanf("%d%d%d",&a,&b,&c) #define pd(a) printf("%d ",a) #define scl(a) scanf("%lld",&a) #define scll(a,b) scanf("%lld%lld",&a,&b) #define sclll(a,b,c) scanf("%lld%lld%lld",&a,&b,&c) #define IOS ios::sync_with_stdio(false);cin.tie(0) #define lc idx<<1 #define rc idx<<1|1 #define lson l,mid,lc #define rson mid+1,r,rc using namespace std; typedef long long ll; template <class T> void test(T a){cout<<a<<endl;} template <class T,class T2> void test(T a,T2 b){cout<<a<<" "<<b<<endl;} template <class T,class T2,class T3> void test(T a,T2 b,T3 c){cout<<a<<" "<<b<<" "<<c<<endl;} const int N = 1e6+10; const int inf = 0x3f3f3f3f; const ll INF = 0x3f3f3f3f3f3f3f3fll; const ll mod = 1000000007; int T; void testcase(){ printf("Case %d:",++T); } const int MAXN = 2e5+10; const int MAXM = 30; const int p = 19; int nxt[MAXN],extend[MAXN]; char s[MAXN],t[MAXN>>1]; void get_next(int m){ int i,j; j=nxt[0]=-1; i=0; while(i<m){ while(-1!=j&&t[i]!=t[j]) j=nxt[j]; if(t[++i]==t[++j]) nxt[i]=nxt[j]; else nxt[i]=j; } } void pre_EKMP(int m){ nxt[0]=m; int j=0; while(j+1<m&&t[j]==t[j+1]) j++; nxt[1]=j; int k=1; for(int i=2;i<m;i++){ int p =nxt[k]+k-1; int L=nxt[i-k]; if(i+L<p+1) nxt[i]=L; else{ j=max(0,p-i+1); while(i+j<m&&t[i+j]==t[j]) j++; nxt[i]=j; k=i; } } } void EKMP(int m,int n){ pre_EKMP(m); int j=0; while(j<n&&j<m&&t[j]==s[j]) j++; extend[0]=j; int k=0; for(int i=1;i<n;i++){ int p =extend[k]+k-1; int L=nxt[i-k]; if(i+L<p+1) extend[i]=L; else{ j=max(0,p-i+1); while(i+j<n&&j<m&&s[i+j]==t[j]) j++; extend[i]=j; k=i; } } } void solve(){ int lt=strlen(t); int ls=strlen(s); get_next(lt); int k=lt-nxt[lt]; k= lt%k==0?(lt/k):1; EKMP(lt,ls); int e=0,g=0,l=0; for(int i=0;i<lt;i++){ if(extend[i]>=lt) e++; else if(s[extend[i]]>s[i+extend[i]]) l++; else g++; } printf(" %d %d %d ",l/k,e/k,g/k); } int main() { #ifdef LOCAL freopen("in.txt","r",stdin); #endif // LOCAL int tt; T=0; scd(tt); while(tt--){ scanf("%s",s); strcpy(t,s); strcat(s,t); testcase(); solve(); } return 0; }