hihocoder1457

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
********************/

版本1

//#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
********************/