HDU4552 怪盗基德的挑战书 KMP | 后缀数组 | 暴力

  题目链接: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 }
View Code

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 }
View Code
原文地址:https://www.cnblogs.com/zhsl/p/3090528.html