FJWC2019 原样输出 SAM+拓扑排序

Description

---

nealchen 是一只 copycat。
它会把输入按行读入，原封不动地复制到输出中去。
但是在一次更新以后，它的程序出了一些问题。
它没法输出换行符了。
并且，读入的时候，总会莫名其妙地随机漏掉开头和结尾的若干个字符，甚至整行都会漏掉。
比如 orznight 可能会变成 rzni ，orz，h 或者空串。
现在你找到一份输入文件丢给 nealchen，你想知道它的输出可能有多少种情况，以及每种情况分别是什么。
由于你找到的输入文件全部来自之前的福建省选，所以所有的输入文件每行只可能包含 ACGT 四种字符。
保证输入文件不超过1MB，输出文件不超过100MB

Solution

---

感受到了出题人深深的恶意。。

40分很好做，我们搞出一个串的SAM，然后数数就可以了。输出方案的话，SAM都出来了贪心dfs就行

70分和90分是一档的。我们只需要对每个串建SAM，然后对于(走不下去的状态)向[后面第一个根节点能走的]SAM转移状态连边，这样子就变成了dag路径数数+dfs贪心输出方案。正确性的话大概就是先把能走的走完，最后才考虑跨过串之间的转移这样

100分的话非常毒瘤，题目并没有告诉我们输入n的具体范围，以及|S|的具体范围，因此可能很大。并且最终的输出可能非常大，我比较蠢只会std:: cin这样，大概是卡不过第5个点了。。
除此之外，由于某OJ的奇妙限制，输出太大会有OLE的报错，这个我也没办法啊。。

综上，如果哪位神仙能全A的希望可以赐教OTZ

Code

---

#include <bits/stdc++.h>
#define LL long long
using namespace std;
const int N=3e6+5,P=1e9+7;
int d[N],ans,sz,lst,k,n,m[N],las[4],rt[N],in[N];
string s[N];queue<int>q;vector<int>gg[N];char g[N];
struct SAM{int len,link,mp[4];}sam[N];
void sam_con(int d,int ch){
	int np=++sz,p=lst;
	sam[np].len=sam[p].len+1;
	while(p && !sam[p].mp[ch])
		sam[p].mp[ch]=np,p=sam[p].link;
	if (!p) sam[np].link=rt[d];
	else{
		int q=sam[p].mp[ch];
		if (sam[q].len==sam[p].len+1)
			sam[np].link=q;
		else{
			int nq=++sz;
			sam[nq].len=sam[p].len+1;
			sam[nq].link=sam[q].link;
			for (int i=0;i<4;i++)
				sam[nq].mp[i]=sam[q].mp[i];
			sam[q].link=sam[np].link=nq;
			while(p && sam[p].mp[ch]==q)
				sam[p].mp[ch]=nq,p=sam[p].link;
		}
	}
	lst=np;
}
void dfs(int id,int x,int j){
	if (j==0) g[x]='A';if (j==1) g[x]='C';
	if (j==2) g[x]='G';if (j==3) g[x]='T';
	for (int i=0;i<=x;i++)
		putchar(g[i]);putchar('\n');
	for (int i=0;i<4;i++)
		if (sam[id].mp[i])
			dfs(sam[id].mp[i],x+1,i);
}
int main(){
	freopen("data.in","r",stdin);
	freopen("std.out","w",stdout);
	scanf("%d",&n);
	for (int i=1;i<=n;i++)
		cin>>s[i],m[i]=s[i].length();
	for (int i=n;i;i--){
		lst=++sz;rt[i]=sz;
		for (int y,j=0;j<m[i];j++){
			if (s[i][j]=='A') y=0;
			if (s[i][j]=='C') y=1;
			if (s[i][j]=='G') y=2;
			if (s[i][j]=='T') y=3;
			sam_con(i,y);
		}
		for (int j=rt[i];j<=sz;j++)
			for (int l=0;l<4;l++){
				if (!sam[j].mp[l])
					sam[j].mp[l]=las[l];
				if (sam[j].mp[l])
					gg[sam[j].mp[l]].push_back(j),
					in[j]++;
			}
		for (int l=0;l<4;l++)
			if (sam[rt[i]].mp[l])
				las[l]=sam[rt[i]].mp[l];
	}
	scanf("%d",&k);
	if (k){
		puts("");ans++;
		for (int i=0;i<4;i++)
			if (sam[rt[1]].mp[i])
				dfs(sam[rt[1]].mp[i],0,i);
	}
	for (int i=1;i<=sz;i++){
		d[i]=1;
		if (!in[i]) q.push(i);
	}
	while(!q.empty()){
		int x=q.front();q.pop();
		for (int j,i=gg[x].size()-1;~i;i--){
			j=gg[x][i];(d[j]+=d[x])%=P;in[j]--;
			if (!in[j]) q.push(j);
		}
	}
	printf("%d\n",d[rt[1]]);
	return 0;
}