BZOJ3238: [Ahoi2013]差异

3238: [Ahoi2013]差异

Time Limit: 20 Sec  Memory Limit: 512 MB
Submit: 437  Solved: 213
[Submit][Status]
Description

Input

一行，一个字符串S

Output

一行，一个整数，表示所求值

Sample Input

cacao
Sample Output

54

HINT

2<=N<=500000,S由小写英文字母组成

题解：

刚看到题目，想着求出height然后搞出每个点作为最小值向右向左拓展的最远距离，然后这段里面随便取两个lcp都是这个height[i]，然后就行了吧。

写完发现不对，两个点必须跨越这个i，发现还是过不了样例。。。

然后想到如果跨越了两个最小值怎么办！！！按照上面的算法我们就计算了两次！！！

然后想不到什么好的解决方法。。。那就乱搞吧

 for1(i,n)l[i]=max(last[h[i]],l[i]),last[h[i]]=i;

总算过了样例，抱着必WA的心态交了居然A了！！！

然后想了想发现这样做是正确的？跨越两次的左端点一定在last[h[i]]左边，那么我们下一次计算的时候把这部分直接去掉就OK了。

代码：

 1 #include<cstdio>
 2 #include<cstdlib>
 3 #include<cmath>
 4 #include<cstring>
 5 #include<algorithm>
 6 #include<iostream>
 7 #include<vector>
 8 #include<map>
 9 #include<set>
10 #include<queue>
11 #include<string>
12 #define inf 1000000000
13 #define maxn 600000
14 #define maxm 500+100
15 #define eps 1e-10
16 #define ll long long
17 #define pa pair<int,int>
18 #define for0(i,n) for(int i=0;i<=(n);i++)
19 #define for1(i,n) for(int i=1;i<=(n);i++)
20 #define for2(i,x,y) for(int i=(x);i<=(y);i++)
21 #define for3(i,x,y) for(int i=(x);i>=(y);i--)
22 #define mod 1000000007
23 using namespace std;
24 inline int read()
25 {
26     int x=0,f=1;char ch=getchar();
27     while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1;ch=getchar();}
28     while(ch>=‘0‘&&ch<=‘9‘){x=10*x+ch-‘0‘;ch=getchar();}
29     return x*f;
30 }
31 int n,m,last[maxn],a[maxn],c[maxn],t1[maxn],t2[maxn],sa[maxn],rk[maxn],h[maxn],sta[maxn],l[maxn],r[maxn];
32 char s[maxn];
33 ll sum[maxn];
34 void getsa(int m)
35 {
36     int *x=t1,*y=t2;
37     for0(i,m)c[i]=0;
38     for0(i,n)c[x[i]=a[i]]++;
39     for1(i,m)c[i]+=c[i-1];
40     for3(i,n,0)sa[--c[x[i]]]=i;
41     for(int k=1;k<n+1;k<<=1)
42     {
43         int p=0;
44         for2(i,n-k+1,n)y[p++]=i;
45         for0(i,n)if(sa[i]>=k)y[p++]=sa[i]-k;
46         for0(i,m)c[i]=0;
47         for0(i,n)c[x[y[i]]]++;
48         for1(i,m)c[i]+=c[i-1];
49         for3(i,n,0)sa[--c[x[y[i]]]]=y[i];
50         swap(x,y);p=0;x[sa[0]]=0;
51         for1(i,n)x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?p:++p;
52         if(p>n)break;
53         m=p;
54     }
55     for1(i,n)rk[sa[i]]=i;
56     for(int i=0,k=0,j;i<n;h[rk[i++]]=k)
57      for(k?k--:0,j=sa[rk[i]-1];a[i+k]==a[j+k];k++);
58 }
59 int main()
60 {
61     freopen("input.txt","r",stdin);
62     freopen("output.txt","w",stdout);
63     scanf("%s",s);n=strlen(s);
64     for0(i,n-1)a[i]=s[i]-‘a‘+1;a[n]=0;
65     getsa(26);
66     h[0]=h[n+1]=-1;
67     int top=0;
68     for1(i,n+1)
69     {
70         while(top&&h[sta[top]]>h[i])r[sta[top--]]=i-1;
71         sta[++top]=i;
72     }
73     top=0;
74     for3(i,n,0)
75     {
76         while(top&&h[sta[top]]>h[i])l[sta[top--]]=i+1;
77         sta[++top]=i;
78     }
79     for1(i,n)l[i]=max(l[i]-1,1);
80     for1(i,n)l[i]=max(last[h[i]],l[i]),last[h[i]]=i;
81     for1(i,n)sum[i]=sum[i-1]+n-sa[i];
82     //for1(i,n)cout<<i<<‘ ‘<<sa[i]<<‘ ‘<<h[i]<<‘ ‘<<l[i]<<‘ ‘<<r[i]<<endl;
83     ll ans=0;
84     for1(i,n)
85     {
86         //cout<<sum[i-1]-sum[l[i]-1]<<‘ ‘<<sum[r[i]]-sum[i-1]<<endl;
87         /*for2(j,l[i],i-1)
88          for2(k,i,r[i])
89           cout<<j<<‘ ‘<<k<<endl;*/
90        ans+=(sum[i-1]-sum[l[i]-1])*(ll)(r[i]-i+1)+(sum[r[i]]-sum[i-1])*(ll)(i-l[i])-(ll)2*(ll)(r[i]-i+1)*(ll)(i-l[i])*(ll)h[i];
91       // cout<<i<<‘ ‘<<ans<<endl;
92     }
93     printf("%lld\n",ans);
94     return 0;
95 }

UPD：看了题解简直吓尿，什么笛卡尔树，后缀树，分治，SAM一个都不会 QAQ