P2322 [HNOI2006]最短母串问题(AC自动机)

Posted 2021-12-28 Harris-H

P2322 [HNOI2006]最短母串问题(AC自动机)

AC自动机。

用一个变量st，维护每个结点包含的串(状压)，注意getFail时也要传递st变量。

然后bfs ，回溯输出答案即可。

时间复杂度：$O(cnt\times 2^n)$

// Problem: P2322 [HNOI2006]最短母串问题
// Contest: Luogu
// URL: https://www.luogu.com.cn/problem/P2322
// Memory Limit: 32 MB
// Time Limit: 1000 ms
// Date: 2021-12-02 21:47:26
// --------by Herio--------

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull; 
const int N=655,M=N*(1<<12)+1,inf=0x3f3f3f3f,mod=1e9+7;
const int hashmod[4] = 402653189,805306457,1610612741,998244353;
#define mst(a,b) memset(a,b,sizeof a)
#define PII pair<int,int>
#define PLL pair<ll,ll>
#define x first
#define y second
#define pb emplace_back
#define SZ(a) (int)a.size()
#define rep(i,a,b) for(int i=a;i<=b;++i)
#define per(i,a,b) for(int i=a;i>=b;--i)
#define ios ios::sync_with_stdio(false),cin.tie(nullptr) 
void Print(int *a,int n)
	for(int i=1;i<n;i++)
		printf("%d ",a[i]);
	printf("%d\\n",a[n]); 

template <typename T>		//x=max(x,y)  x=min(x,y)
void cmx(T &x,T y)
	if(x<y) x=y;

template <typename T>
void cmn(T &x,T y)
	if(x>y) x=y;

int n;
struct ACAM
#define il inline
	int son[N][26],fa[N],id,st[N];
		int pre[M],ans[M];
	bool vis[N][(1<<12)];
	il void init()
		id=0;
		mst(fa,0),mst(st,0);
		mst(son,0);
		mst(pre,0);mst(ans,0);
	
	il void ins(char *s,int num)
		int rt = 0, n =strlen(s);
		for(int i=0;i<n;++i)
			int j=s[i]-'A';
			if(!son[rt][j]) son[rt][j]=++id;
			rt = son[rt][j];
		
		st[rt] |= (1<<num-1);
	
	il void getFail()
		queue<int>q;
		for(int i=0;i<26;i++)
			if(son[0][i]) fa[son[0][i]]=0,q.push(son[0][i]);
		while(!q.empty())
			int u=q.front();q.pop();
			//printf("u=%d\\n",u);
			for(int i=0;i<26;i++)
				if(son[u][i])
			fa[son[u][i]] = son[fa[u]][i],st[son[u][i]]|=st[son[fa[u]][i]];
			q.push(son[u][i]);
			
				else son[u][i] = son[fa[u]][i];
			
			
	
	void dfs(int u)
		if(!u) return;
		dfs(pre[u]);
		putchar('A'+ans[u]);
	
	void solve()
		int cnt = 0,tot = 0;
		queue<PII>q;
		q.push(0,0);vis[0][0]=1;
		int all = (1<<n)-1;
		while(!q.empty())
			int u = q.front().x,now =q.front().y;
			q.pop();
		//	printf("u=%d\\n",u);
			if(now == all)
				dfs(cnt);
				return;
			
			for(int i=0;i<26;i++)
				int v = son[u][i],nxt = st[v]|now;
				if(vis[v][nxt]) continue;
				vis[v][nxt] = true;
				q.push(v,nxt);
				pre[++tot] = cnt;
				ans[tot] = i; 
			
			++cnt;
		
	
T;
char s[55];
int main()
	scanf("%d",&n);//T.init();
	rep(i,1,n)
		scanf("%s",s);
		T.ins(s,i);
	
	T.getFail();
	T.solve();
	return 0;