省选专练之后缀自动机Hihocoder1457

伪广义后缀自动机

利用SA的著名做法加个字符集以外的串然后把串连接起来

选择':'（'0'+10）

然后所有parent树上不经过：边的串都是合法的

然后这里是要从短串转移到长串

基数排序了之后从1-siz暴力转移

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
typedef int INT;
#define int long long 
const int mod=1e9+7;
const int N=2e6+1000;
char S[N];
struct SAM{
	struct Node{
		int vis[12];
		int pre;
		int step;
	}SA[N];
	int last,cnt;
	int C[N];
	int A[N];
	int siz[N];
	int sum[N];
	inline void Insert(char C){
//		cout<<C<<'\n';
		int p=last;
		int np=++cnt;
		last=np;
		SA[np].step=SA[p].step+1;
		for(;!SA[p].vis[C-'0'];p=SA[p].pre)SA[p].vis[C-'0']=np;//cout<<p<<" "<<C<<'\n';
		if(!p)SA[np].pre=1;
		else{
			int q=SA[p].vis[C-'0'];
			if(SA[q].step==SA[p].step+1){
				SA[np].pre=q;
//				cout<<p<<'\n';
//				cout<<SA[q].step<<" "<<SA[p].step<<'\n';
//				cout<<C<<'\n';
			}
			else{
//				cout<<"23908980"<<'\n';
				int nq=++cnt;
				SA[nq].step=SA[p].step+1;
				memcpy(SA[nq].vis,SA[q].vis,sizeof(SA[q].vis));
				SA[nq].pre=SA[q].pre;
				SA[np].pre=SA[q].pre=nq;
				for(;SA[p].vis[C-'0']==q;p=SA[p].pre)SA[p].vis[C-'0']=nq;
			}
		}
	}
	void Solve(){
		int n,all=0;
		scanf("%d",&n);
		last=cnt=1;
		for(int i=1;i<=n;i++){
			scanf("%s",S+1);
			int len=strlen(S+1);
			all+=len;
//			cout<<S+1<<'\n';			
			for(int j=1;j<=len;j++){
				Insert(S[j]);
			}			
			if(i!=n)Insert('0'+10),all++;
		}
//		cout<<"hehe"; 
		for(int i=1;i<=cnt;i++)C[SA[i].step]++;
		for(int i=1;i<=all;i++)C[i]+=C[i-1];
//		cout<<cnt<<'\n';
		for(int i=cnt;i>=1;i--)A[C[SA[i].step]--]=i;
		siz[1]=1;
		for(int i=1;i<=cnt;i++){
			int p=A[i];if(siz[p]==0)continue;
			for(int j=0;j<=9;j++){
				
				if(SA[p].vis[j]){

					siz[SA[p].vis[j]]+=siz[p]%mod;
					siz[SA[p].vis[j]]%=mod;	
//									cout<<sum[p]*10<<'\n';
//					cout<<
					sum[SA[p].vis[j]]+=(sum[p]*10%mod+j*siz[p]%mod)%mod;
					sum[SA[p].vis[j]]%=mod;

//					cout<<sum[SA[p].vis[j]]<<" "<<p<<" "<<sum[p]<<'\n';
				}
			}
		}
		int ans=0;
		for(int i=1;i<=cnt;i++)(ans+=sum[i])%=mod;
		cout<<ans;
	}
}Solution;
INT main(){
//	freopen("hihocoder1457.in","r",stdin);
	Solution.Solve();
}

 

转载于:https://www.cnblogs.com/Leo-JAM/p/10079170.html