【poj1743-Musical Theme】不可重叠最长重复子串-后缀数组

http://poj.org/problem?id=1743

这题是一道后缀数组的经典例题:求不可重叠最长重复子串。

题意:

有N(1 <= N <=20000)个音符的序列来表示一首乐曲,每个音符都是1..88范围内的整数,现在要找一个重复的主题。“主题”是整个音符序列的一个子串,它需要满足如下条件:
1.长度至少为5个音符。
2.在乐曲中重复出现。(可能经过转调,“转调”的意思是主题序列中每个音符都被加上或减去了同一个整数值)
3.重复出现的同一主题不能有公共部分。

题解:

因为可能同时加上或减去一个数,所以首先要作差比较。将后一个数减去前一个数得到新的数列c。

注意:例如1 2 3 4 5 2 3 4 5 6 变为 1 1 1 1 1 -3 1 1 1 1 ,重复子串的第一位是会变化的,最后要+1。

二分重复子串的长度k,然后判断它是否可行。

截自论文的图:

注意(我打的时候犯的错误):

1.分组的时候忽略了最后一组的判断。

2.因为判断的重复子串长度比真实长度小1,所以要判断位置x-y>k而不是>=k

对于这个问题的一组debug数据:1 2 3 4 5 1 2 3 4 5 1 ans=5

3.rk[i]不能等于0,至少>=1,因为我的模板没有第二关键字的那些的rk补成0了

4.看到评论里很多人犯了的错误:k最少是5,否则输出0

贴几组数据(debug好帮手):

input:

12
1 2 3 4 5 6 7 8 9 10 11 12
0

11
1 1 1 1 1 1 1 1 1 1 1

9
1 1 1 1 1 1 1 1 1

11
1 2 3 5 4 1 2 3 5 4 1

33
3 2 1 1 2 1 2 3 2 2 3 3 2 2 1 2 1 2 2 2 1 3 1 1 1 2 2 3 1 2 1 1 2

1
100

300
48 15 74 57 17 52 51 20 86 85 24 19 23 34 81 54 12 3 86 41 45 64 23 32 18 17 68 43 83 86
61 22 48 47 50 21 1 12 19 16 78 21 64 27 71 50 65 42 68 11 30 25 45 72 23 44 10 81 36 39
19 46 45 34 8 87 42 45 1 12 67 28 62 13 88 19 71 42 65 42 60 83 86 49 45 72 31 56 10 9
12 35 75 70 45 14 64 55 50 17 17 20 51 28 70 44 45 27 16 26 87 49 58 40 57 35 12 18 35
57 30 56 29 31 40 34 23 53 34 16 73 27 84 54 75 57 6 20 13 35 8 18 63 17 66 52 41 15 20
54 3 57 22 20 85 19 24 54 63 41 82 32 57 43 76 22 11 21 54 16 61 27 16 42 39 25 2 44 1
39 12 34 83 45 14 28 61 19 72 42 79 49 34 56 41 35 36 14 11 17 78 28 44 27 26 49 40 35
18 57 56 31 34 53 16 27 54 57 20 35 18 17 52 15 31 14 13 36 27 22 5 44 43 18 21 40 3 14
41 44 7 22 5 4 39 2 41 44 7 6 41 28 19 30 9 8 3 14 29 12 31 26 21 32 15 6 29 36 43 22 1
4 15 38 13 8 7 22 29 4 19 26 41 28 19 18 21 4 27 22 25 20 39 18 17 28 7 6 37 4

22
1 2 3 2 1 2 3 2 1 1 1 1 1 1 1 1 1 1 1 1 1 1

30
25 27 30 34 39 45 52 60 69 79 69 60 52 45 39 34 30 26 22 18
82 78 74 70 66 67 64 60 65 80
0

output:

6
5
0
5
5
0
22
7
5

代码:

  1 //poj1743
  2 #include<cstdio>
  3 #include<cstdlib>
  4 #include<cstring>
  5 #include<iostream>
  6 using namespace std;
  7 
  8 const int N=200010,INF=(int)1e9;
  9 int n,a[N],c[N],sa[N],rk[N],Rs[N],y[N],wr[N],h[N];
 10 
 11 int minn(int x,int y){return x<y ? x:y;}
 12 int maxx(int x,int y){return x>y ? x:y;}
 13 
 14 void get_sa(int cl,int m)
 15 {
 16     for(int i=1;i<=cl;i++) rk[i]=c[i];//c[i]必须>=1
 17     for(int i=0;i<=m;i++) Rs[i]=0;
 18     for(int i=1;i<=cl;i++) Rs[rk[i]]++;
 19     for(int i=1;i<=m;i++) Rs[i]+=Rs[i-1];
 20     for(int i=cl;i>=1;i--) sa[Rs[rk[i]]--]=i;
 21     
 22     int ln=1,p=0;
 23     while(p<cl)
 24     {
 25         int k=0;
 26         for(int i=cl-ln+1;i<=cl;i++) y[++k]=i;
 27         for(int i=1;i<=cl;i++) 
 28             if(sa[i]>ln) y[++k]=sa[i]-ln;
 29         for(int i=1;i<=cl;i++)
 30             wr[i]=rk[y[i]];
 31 
 32         for(int i=0;i<=m;i++) Rs[i]=0;
 33         for(int i=1;i<=cl;i++) Rs[wr[i]]++;
 34         for(int i=1;i<=m;i++) Rs[i]+=Rs[i-1];
 35         for(int i=cl;i>=1;i--) sa[Rs[wr[i]]--]=y[i];
 36         
 37         for(int i=1;i<=cl;i++) wr[i]=rk[i];
 38         for(int i=cl+1;i<=cl+ln;i++) wr[i]=0;//debug:rk[i]不能等于0的原因:这里给没有第二关键字的补0了。
 39         p=1;rk[sa[1]]=1;
 40         for(int i=2;i<=cl;i++)
 41         {
 42             if(wr[sa[i]]!=wr[sa[i-1]] || wr[sa[i]+ln]!=wr[sa[i-1]+ln]) p++;//debug '!='
 43             rk[sa[i]]=p;
 44         }
 45         ln*=2;m=p;//debug
 46     }
 47     sa[0]=0;rk[0]=0;
 48 }
 49 
 50 void get_height(int cl)
 51 {
 52     int k=0;
 53     for(int i=1;i<=cl;i++) if(rk[i]!=1)
 54     {
 55         int j=sa[rk[i]-1];
 56         if(k) k--;
 57         while(c[i+k]==c[j+k] && i+k<=cl && j+k<=cl) k++;
 58         h[rk[i]]=k;
 59     }
 60     h[1]=0;
 61 }
 62 
 63 bool check(int cl,int k)
 64 {
 65     int fir=1,mn=INF,mx=0;
 66     for(int i=1;i<=cl;i++)
 67     {
 68         if(h[i]<k)
 69         {
 70             if(mx-mn>k) return 1;//debug:>=改成>才能保证差值不重复的情况下音符也不重复(第一个音符没有算进去)
 71             fir=i;mn=INF;mx=0;
 72         }
 73         else 
 74         {    
 75             mn=minn(mn,sa[i-1]);
 76             mx=maxx(mx,sa[i-1]);
 77             
 78         }
 79         mn=minn(mn,sa[i]);
 80         mx=maxx(mx,sa[i]);
 81     }
 82     if(mx-mn>k) return 1;//debug:最后一组不可忽略。
 83     return 0;
 84 }
 85 
 86 int main()
 87 {
 88     freopen("a.in","r",stdin);
 89     freopen("a.out","w",stdout);
 90     while(1)
 91     {
 92         scanf("%d",&n);
 93         if(!n) return 0;
 94         int mx=-1,pre=0,mn=INF;
 95         for(int i=1;i<=n;i++)
 96         {
 97             int x;
 98             scanf("%d",&x);
 99             c[i]=x-pre;
100             pre=x;
101             mx=maxx(mx,c[i]);
102             mn=minn(mn,c[i]);
103         }
104         for(int i=1;i<=n;i++) c[i]=c[i]-mn+1;//debug 因为rk[i]不能=0,c[i]也不能=0
105         mx=mx-mn+1;
106         get_sa(n,mx);
107         get_height(n);
108         int l=0,r=n;
109         while(l<r)
110         {
111             int mid=(l+r+1)/2;
112             if(check(n,mid-1)) l=mid;
113             else r=mid-1;
114         }
115         if(l>=5) printf("%d
",l);
116         else printf("0
");
117     }
118     return 0;
119 }
原文地址:https://www.cnblogs.com/KonjakJuruo/p/5688609.html