http://hihocoder.com/problemset/problem/1457
找不重复子串的和
topo序搞一搞,用父亲更新儿子节点的val,记得乘上节点数
//#pragma comment(linker, "/stack:200000000") //#pragma GCC optimize("Ofast,no-stack-protector") //#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native") //#pragma GCC optimize("unroll-loops") #include<bits/stdc++.h> #define fi first #define se second #define mp make_pair #define pb push_back #define pi acos(-1.0) #define ll long long #define mod (1000000007) #define C 0.5772156649 #define ls l,m,rt<<1 #define rs m+1,r,rt<<1|1 #define pil pair<int,ll> #define pii pair<int,int> #define ull unsigned long long #define base 1000000000000000000 #define fio ios::sync_with_stdio(false);cin.tie(0) using namespace std; const double g=10.0,eps=1e-12; const int N=2000000+10,maxn=1200000+10,inf=0x3f3f3f3f,INF=0x3f3f3f3f3f3f3f3f; inline void sub(ll &x){if(x>=mod)x-=mod;} char s[N]; int c[N],a[N]; ll val[N],sum[N]; struct SAM{ int last,cnt; int ch[N<<1][26],fa[N<<1],l[N<<1]; int mx[N<<1],tmp[N<<1]; void ins(int c) { int p=last,np=++cnt;last=np;l[np]=l[p]+1; for(;p&&!ch[p][c];p=fa[p])ch[p][c]=np; if(!p)fa[np]=1; else { int q=ch[p][c]; if(l[p]+1==l[q])fa[np]=q; else { int nq=++cnt;l[nq]=l[p]+1; memcpy(ch[nq],ch[q],sizeof(ch[q])); fa[nq]=fa[q];fa[q]=fa[np]=nq; sum[nq]=1ll*c; for(;ch[p][c]==q;p=fa[p])ch[p][c]=nq; } } sum[np]=1ll*c; } SAM() { cnt=1; } void build(){ int len=strlen(s+1); last=1; for(int i=1;i<=len;i++)ins(s[i]-‘0‘); // topo(); // for(int i=1;i<=cnt;i++)mx[i]=l[i],tmp[i]=0; } void topo(){ for(int i=1;i<=cnt;i++)c[l[i]]++; for(int i=1;i<=cnt;i++)c[i]+=c[i-1]; for(int i=1;i<=cnt;i++)a[c[l[i]]--]=i; } void debug() { puts(""); for(int i=1;i<=cnt;i++) { printf("%d ",i); for(int j=0;j<26;j++) printf("%d ",ch[i][j]); printf("%d %d %lld %lld\n",fa[i],l[i],val[i],sum[i]); } } void faupdate() { for(int i=cnt;i;i--) tmp[fa[a[i]]]=max(tmp[fa[a[i]]],tmp[a[i]]); } void maupdate() { for(int i=1;i<=cnt;i++) mx[i]=min(mx[i],tmp[i]),tmp[i]=0; } void match() { int len=strlen(s+1),now=1,res=0; for(int i=1;i<=len;i++) { if(ch[now][s[i]-‘a‘]) { now=ch[now][s[i]-‘a‘]; res++; } else { while(now&&!ch[now][s[i]-‘a‘])now=fa[now]; if(!now)now=1,res=0; else { res=l[now]+1; now=ch[now][s[i]-‘a‘]; } } tmp[now]=max(tmp[now],res); } faupdate(); maupdate(); } int maxlen() { int ans=0; for(int i=1;i<=cnt;i++)ans=max(ans,mx[i]); return ans; } void cal() { topo(); for(int i=1;i<=cnt;i++) { int p=a[i]; val[p] = val[p] + 1ll*sum[p] * (l[p] - l[fa[p]])%mod; sub(val[p]); for(int j=0;j<10;j++) { if(ch[p][j]) { val[ch[p][j]]=val[ch[p][j]]+val[p]*10ll%mod; sub(val[ch[p][j]]); } } // cout << val[p] << endl; } // debug(); ll ans=0; for(int i=cnt;i;i--) { ans=ans+val[i]; sub(ans); } printf("%lld\n",ans); } }sam; int main() { int n; scanf("%d",&n); for(int i=1;i<=n;i++) { scanf("%s",s+1); sam.build(); } sam.cal(); return 0; } /******************** 1 101 ********************/
//#pragma comment(linker, "/stack:200000000") //#pragma GCC optimize("Ofast,no-stack-protector") //#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native") //#pragma GCC optimize("unroll-loops") #include<bits/stdc++.h> #define fi first #define se second #define mp make_pair #define pb push_back #define pi acos(-1.0) #define ll long long #define mod (1000000007) #define C 0.5772156649 #define ls l,m,rt<<1 #define rs m+1,r,rt<<1|1 #define pil pair<int,ll> #define pii pair<int,int> #define ull unsigned long long #define base 1000000000000000000 #define fio ios::sync_with_stdio(false);cin.tie(0) using namespace std; const double g=10.0,eps=1e-12; const int N=2000000+10,maxn=1200000+10,inf=0x3f3f3f3f,INF=0x3f3f3f3f3f3f3f3f; inline void sub(ll &x){if(x>=mod)x-=mod;} char s[N]; int c[N],a[N]; ll val[N],sum[N],sz[N]; struct SAM{ int last,cnt; int ch[N<<1][26],fa[N<<1],l[N<<1]; int mx[N<<1],tmp[N<<1]; void ins(int c) { int p=last,np=++cnt;last=np;l[np]=l[p]+1; for(;p&&!ch[p][c];p=fa[p])ch[p][c]=np; if(!p)fa[np]=1; else { int q=ch[p][c]; if(l[p]+1==l[q])fa[np]=q; else { int nq=++cnt;l[nq]=l[p]+1; memcpy(ch[nq],ch[q],sizeof(ch[q])); fa[nq]=fa[q];fa[q]=fa[np]=nq; for(;ch[p][c]==q;p=fa[p])ch[p][c]=nq; } } sum[np]=1ll*c; } SAM() { cnt=1; } void build(){ int len=strlen(s+1); last=1; for(int i=1;i<=len;i++)ins(s[i]-‘0‘); // topo(); // for(int i=1;i<=cnt;i++)mx[i]=l[i],tmp[i]=0; } void topo(){ for(int i=1;i<=cnt;i++)c[l[i]]++; for(int i=1;i<=cnt;i++)c[i]+=c[i-1]; for(int i=1;i<=cnt;i++)a[c[l[i]]--]=i; } void debug() { puts(""); for(int i=1;i<=cnt;i++) { printf("%d ",i); for(int j=0;j<26;j++) printf("%d ",ch[i][j]); printf("%d %d %lld %lld\n",fa[i],l[i],val[i],sum[i]); } } void faupdate() { for(int i=cnt;i;i--) tmp[fa[a[i]]]=max(tmp[fa[a[i]]],tmp[a[i]]); } void maupdate() { for(int i=1;i<=cnt;i++) mx[i]=min(mx[i],tmp[i]),tmp[i]=0; } void match() { int len=strlen(s+1),now=1,res=0; for(int i=1;i<=len;i++) { if(ch[now][s[i]-‘a‘]) { now=ch[now][s[i]-‘a‘]; res++; } else { while(now&&!ch[now][s[i]-‘a‘])now=fa[now]; if(!now)now=1,res=0; else { res=l[now]+1; now=ch[now][s[i]-‘a‘]; } } tmp[now]=max(tmp[now],res); } faupdate(); maupdate(); } int maxlen() { int ans=0; for(int i=1;i<=cnt;i++)ans=max(ans,mx[i]); return ans; } void cal() { topo(); sz[1]=1; for(int i=1;i<=cnt;i++) { int p=a[i]; // val[p] = val[p] + 1ll*sum[p] * (l[p]-l[fa[p]])%mod; // sub(val[p]); for(int j=0;j<10;j++) { if(ch[p][j]) { val[ch[p][j]]=val[ch[p][j]]+val[p]*10ll%mod+1ll*j*sz[p]%mod; sub(val[ch[p][j]]); sz[ch[p][j]]+=sz[p]; } } // cout << val[p] << endl; } // debug(); ll ans=0; for(int i=cnt;i;i--) { ans=ans+val[i]; sub(ans); } printf("%lld\n",ans); } }sam; int main() { int n; scanf("%d",&n); for(int i=1;i<=n;i++) { scanf("%s",s+1); sam.build(); // if(i!=n)sam.ins(10); } sam.cal(); return 0; } /******************** 1 987654321123456789 ********************/