题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4552
题意:求字符串中所有前缀的出现的次数。
比赛的时候使用后缀数组写的,后来比完后发现KMP+DP完全可以搞,当时沙茶了- -。然后更神奇的是,这题数据太弱了,暴力完全够了,对于随机数据,基本上就是O(n)的算法,当然如果完全每个字符都一样,那么就是O( n^2 )了。然后...我就缩代码到151B,刷到status第一了~
后缀数组:
1 //STATUS:C++_AC_78MS_3204KB 2 #include <functional> 3 #include <algorithm> 4 #include <iostream> 5 //#include <ext/rope> 6 #include <fstream> 7 #include <sstream> 8 #include <iomanip> 9 #include <numeric> 10 #include <cstring> 11 #include <cassert> 12 #include <cstdio> 13 #include <string> 14 #include <vector> 15 #include <bitset> 16 #include <queue> 17 #include <stack> 18 #include <cmath> 19 #include <ctime> 20 #include <list> 21 #include <set> 22 #include <map> 23 using namespace std; 24 //define 25 #define pii pair<int,int> 26 #define mem(a,b) memset(a,b,sizeof(a)) 27 #define lson l,mid,rt<<1 28 #define rson mid+1,r,rt<<1|1 29 #define PI acos(-1.0) 30 //typedef 31 typedef __int64 LL; 32 typedef unsigned __int64 ULL; 33 //const 34 const int N=100010; 35 const int INF=0x3f3f3f3f; 36 const int MOD=256,STA=8000010; 37 const LL LNF=1LL<<60; 38 const double EPS=1e-8; 39 const double OO=1e15; 40 const int dx[4]={-1,0,1,0}; 41 const int dy[4]={0,1,0,-1}; 42 //Daily Use ... 43 inline int sign(double x){return (x>EPS)-(x<-EPS);} 44 template<class T> T gcd(T a,T b){return b?gcd(b,a%b):a;} 45 template<class T> T lcm(T a,T b){return a/gcd(a,b)*b;} 46 template<class T> inline T Min(T a,T b){return a<b?a:b;} 47 template<class T> inline T Max(T a,T b){return a>b?a:b;} 48 template<class T> inline T Min(T a,T b,T c){return min(min(a, b),c);} 49 template<class T> inline T Max(T a,T b,T c){return max(max(a, b),c);} 50 template<class T> inline T Min(T a,T b,T c,T d){return min(min(a, b),min(c,d));} 51 template<class T> inline T Max(T a,T b,T c,T d){return max(max(a, b),max(c,d));} 52 //End 53 54 int num[N]; 55 int sa[N],t1[N],t2[N],c[N],rank[N],height[N]; 56 int n,m; 57 58 void build_sa(int s[],int n,int m) 59 { 60 int i,k,p,*x=t1,*y=t2; 61 //第一轮基数排序 62 for(i=0;i<m;i++)c[i]=0; 63 for(i=0;i<n;i++)c[x[i]=s[i]]++; 64 for(i=1;i<m;i++)c[i]+=c[i-1]; 65 for(i=n-1;i>=0;i--)sa[--c[x[i]]]=i; 66 for(k=1;k<=n;k<<=1){ 67 p=0; 68 //直接利用sa数组排序第二关键字 69 for(i=n-k;i<n;i++)y[p++]=i; 70 for(i=0;i<n;i++)if(sa[i]>=k)y[p++]=sa[i]-k; 71 //基数排序第一关键字 72 for(i=0;i<m;i++)c[i]=0; 73 for(i=0;i<n;i++)c[x[y[i]]]++; 74 for(i=1;i<m;i++)c[i]+=c[i-1]; 75 for(i=n-1;i>=0;i--)sa[--c[x[y[i]]]]=y[i]; 76 //根据sa和x数组计算新的x数组 77 swap(x,y); 78 p=1;x[sa[0]]=0; 79 for(i=1;i<n;i++) 80 x[sa[i]]=y[sa[i-1]]==y[sa[i]] && y[sa[i-1]+k]==y[sa[i]+k]?p-1:p++; 81 if(p>=n)break; //已经排好序,直接退出 82 m=p; //下次基数排序的最大值 83 } 84 } 85 86 void getHeight(int s[],int n) 87 { 88 int i,j,k=0; 89 for(i=0;i<=n;i++)rank[sa[i]]=i; 90 for(i=0;i<n;i++){ 91 if(k)k--; 92 j=sa[rank[i]-1]; 93 while(s[i+k]==s[j+k])k++; 94 height[rank[i]]=k; 95 } 96 } 97 98 char s[N]; 99 100 int main() 101 { 102 // freopen("in.txt","r",stdin); 103 int i,j,w,low,ans; 104 while(~scanf("%s",s)) 105 { 106 n=strlen(s); 107 for(i=0;i<n;i++){ 108 num[i]=s[i]-'a'+1; 109 } 110 num[n]=0;m=28; 111 build_sa(num,n+1,m); 112 getHeight(num,n); 113 114 low=n;ans=0; 115 for(i=rank[0];i>=2;i--){ 116 low=Min(low,height[i]); 117 if(low==0)break; 118 ans=(ans+low)%MOD; 119 } 120 low=n; 121 for(i=rank[0]+1;i<=n;i++){ 122 low=Min(low,height[i]); 123 if(low==0)break; 124 ans=(ans+low)%MOD; 125 } 126 127 printf("%d\n",(ans+n)%MOD); 128 } 129 return 0; 130 }
KMP+DP:
1 //STATUS:C++_AC_0MS_712KB 2 #include<stdio.h> 3 #include<string.h> 4 int main() 5 { 6 int i,j,n,t[100010],a; 7 char s[100010]; 8 while(~scanf("%s",s)){ 9 n=strlen(s);a=j=0,i=t[0]=-1; 10 while(j<n) 11 if(i==-1 || s[i]==s[j])t[++j]=++i; 12 else i=t[i]; 13 for(i=1;i<=n;i++) 14 for(j=t[i];j!=0;j=t[j])a++; 15 printf("%d\n",(a+n)%256); 16 } 17 return 0; 18 }