码迷,mamicode.com
首页 > 编程语言 > 详细

bzoj3238 [Ahoi2013]差异 后缀数组+单调栈

时间:2018-01-12 20:29:14      阅读:190      评论:0      收藏:0      [点我收藏+]

标签:就是   algorithm   zoj   esc   str   分享图片   geo   algo   ima   

【bzoj3238】[Ahoi2013]差异

Description

技术分享图片

Input

一行,一个字符串S

Output

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

Sample Input

cacao

Sample Output

54

题解:

技术分享图片

 

任意两个字符串的lcp是什么,就是如

a,b  那么若a==b 那么为len(a)

  否则设sa[a]<sa[b] 那么为min(height[sa[a]+1-------sa[b]])

 1 #include<cstring>
 2 #include<iostream>
 3 #include<cmath>
 4 #include<algorithm>
 5 #include<cstdio>
 6 
 7 #define N 500007
 8 #define ll long long
 9 using namespace std;
10 inline int read()
11 {
12     int x=0,f=1;char ch=getchar();
13     while(ch<0||ch>9){if (ch==-)f=-1;ch=getchar();}
14     while(ch>=0&&ch<=9){x=(x<<1)+(x<<3)+ch-0;ch=getchar();}
15     return x*f;
16 }
17 
18 int n;
19 int stk[N],f[N],g[N];
20 struct SA
21 {
22     char s[N];
23     int a[N],b[N],cnta[N],cntb[N],tsa[N],height[N],sa[N],rk[N*2];
24     void Get_SA()
25     {
26         for (int i=0;i<=256;i++)cnta[i]=0;
27         for (int i=1;i<=n;i++)cnta[(int)s[i]]++;
28         for (int i=1;i<=256;i++)cnta[i]+=cnta[i-1];
29         for (int i=n;i>=1;i--)sa[cnta[(int)s[i]]--]=i;
30         rk[sa[1]]=1;
31         for (int i=2;i<=n;i++)rk[sa[i]]=rk[sa[i-1]]+(s[sa[i]]!=s[sa[i-1]]);
32         for (int i=1;rk[sa[n]]!=n;i<<=1)
33         {
34             for (int j=1;j<=n;j++)a[j]=rk[j],b[j]=rk[j+i];
35             for (int j=0;j<=n;j++)cnta[j]=cntb[j]=0;
36             for (int j=1;j<=n;j++)cnta[a[j]]++,cntb[b[j]]++;
37             for (int j=1;j<=n;j++)cnta[j]+=cnta[j-1],cntb[j]+=cntb[j-1];
38             for (int j=n;j>=1;j--)tsa[cntb[b[j]]--]=j;
39             for (int j=n;j>=1;j--)sa[cnta[a[tsa[j]]]--]=tsa[j];
40             rk[sa[1]]=1;
41             for (int j=2;j<=n;j++)
42                 rk[sa[j]]=rk[sa[j-1]]+(a[sa[j]]!=a[sa[j-1]]||b[sa[j]]!=b[sa[j-1]]);
43         }
44     }
45     void Get_Height()
46     {
47         int len=0;
48         for (int i=1;i<=n;i++)
49         {
50             if (len)len--;
51             while(s[i+len]==s[sa[rk[i]-1]+len])len++;
52             height[rk[i]]=len;
53         }
54     }
55 }S;
56 int main()
57 {
58     scanf("%s",S.s+1);
59     n=strlen(S.s+1);
60     ll ans=0;
61     for (int i=1;i<=n;i++)
62     {
63         ans+=(ll)(i-1)*i;
64         ans+=(ll)i*(i-1)/2;
65     }
66     S.Get_SA();
67     S.Get_Height();
68     int tot=0;
69     for (int i=2;i<=n;i++)
70     {
71         while(tot>0&&S.height[i]<S.height[stk[tot]])
72             f[stk[tot--]]=i-1;
73         stk[++tot]=i;
74     }
75     while(tot)f[stk[tot--]]=n;
76     tot=0;
77     for (int i=n;i>=2;i--)
78     {
79         while(tot>0&&S.height[i]<=S.height[stk[tot]])g[stk[tot--]]=i+1;
80         stk[++tot]=i;
81     }
82     while(tot)g[stk[tot--]]=2;
83     for (int i=2;i<=n;i++)
84         ans-=(ll)S.height[i]*(ll)(f[i]-i+1)*(ll)(i-g[i]+1)*2;
85     printf("%lld\n",ans);
86 }

 

bzoj3238 [Ahoi2013]差异 后缀数组+单调栈

标签:就是   algorithm   zoj   esc   str   分享图片   geo   algo   ima   

原文地址:https://www.cnblogs.com/fengzhiyuan/p/8277415.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!