【十二省聯考2019】【LOJ3049】【洛谷P5248】【BZOJ5496】字符串問題(SAM優化建圖)(拓撲排序)

LOJ傳送門

洛谷傳送門

BZOJ傳送門


解析:

瞎寫一通成功拿下BZOJ倒數第一(在線丟人

常數實在是太大了沒辦法。。。

首先我們很顯然發現要求的是DAG最長鏈,這是一個圖論問題。只要把圖建出來一切都好說。

如果建出圖有環肯定就咕咕咕了。

顯然不能暴力建圖,考慮怎麼優化前綴那部分的連邊。

建立反串後綴自動機,那麼前綴關係就轉化成了後綴。

樹上倍增找到每個串的對應節點,用一個vector存下來。

將SAM上每個節點拆成入點和出點,中間放的是這個節點存的串的信息。所有父親的出點向兒子的入點連邊。顯然父親表示串全部都是兒子串的後綴。

現在考慮怎麼處理同一個點的連邊。

直接按照長度爲第一關鍵字,是否爲aa類串爲第二關鍵字(aa串如果和某一個bb相同,顯然應該放在bb的出來方向)對所有串排序。

然後把串給連一連就行了。這樣建立的圖顯然是我們需要求的圖


代碼:

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define re register
#define gc getchar
#define cs const

namespace IO{
	inline int getint(){
		re char c;
		while(!isdigit(c=gc()));re int num=c^48;
		while(isdigit(c=gc()))num=((num+(num<<2))<<1)+(c^48);
		return num;
	}
}
using namespace IO;

using std::cout;
using std::cerr;

typedef std::pair<int,int> pii;
#define mp std::make_pair
#define fi first
#define se second

cs int N=8e5+5;

int n,na,nb;

namespace Graph{
	cs int N=::N<<1;
	std::vector<int> G[N];
	int deg[N],n;
	
	inline void addedge(int u,int v){
		G[u].push_back(v);++deg[v];
		n=std::max(n,std::max(u,v));
	}
	
	ll val[N],len[N];
	std::queue<int> q;
	inline void solve(){
		for(int i=1;i<=n;++i)if(!deg[i])q.push(i);
		int tot=0;
		while(!q.empty()){
			int u=q.front();q.pop();
			len[u]+=val[u];++tot;
			for(int re e=G[u].size()-1;~e;--e){
				int v=G[u][e];
				len[v]=std::max(len[v],len[u]);
				if(!--deg[v])q.push(v);
			}
		}
		if(tot!=n)cout<<"-1\n";
		else cout<<*std::max_element(len+1,len+n+1)<<"\n";
	}
	
	inline void clear(){
		while(n){
			G[n].clear();
			val[n]=len[n]=deg[n]=0;
			n--;
		}
	}
}

namespace SAM{
	cs int N=::N<<1;
	cs int B=20;
	std::vector<pii> nd[N];
	int len[N],fa[N],son[N][26],f[N][B+1];
	int pos[N],id[N];
	int last=1,now=1,tot;
	
	inline void push_back(int c,int id){
		c-='a';
		int cur=++now,p=last;
		len[cur]=len[last]+1;pos[id]=cur;
		for(;p&&!son[p][c];p=fa[p])son[p][c]=cur;
		if(!p)fa[cur]=1;
		else if(len[son[p][c]]==len[p]+1)fa[cur]=son[p][c];
		else {
			int clone=++now,q=son[p][c];
			len[clone]=len[p]+1;
			memcpy(son[clone],son[q],sizeof son[q]);
			fa[clone]=fa[q];
			fa[q]=fa[cur]=clone;
			for(;p&&son[p][c]==q;p=fa[p])son[p][c]=clone;
		}
		last=cur;
	}
	
	inline void init(){
		for(int re i=2;i<=now;++i)f[i][0]=fa[i],Graph::addedge(fa[i]+now,i);
		for(int re i=1;i<=B;++i)
		for(int re j=1;j<=now;++j)f[j][i]=f[f[j][i-1]][i-1];
	}
	
	inline void locate(int l,int r,int id){
		int u=pos[l];
		for(int re i=B;~i;--i)if(len[f[u][i]]>=r-l+1)u=f[u][i];
		nd[u].push_back(mp(r-l+1,id));
	}
	
	inline bool cmp(cs pii &a,cs pii &b){
		return a.fi!=b.fi?a.fi<b.fi:a.se>b.se;
	}
	
	inline void build(){
		for(int re i=1;i<=now;++i){
			std::vector<pii> &vec=nd[i];
			if(!vec.size())Graph::addedge(i,i+now);
			else {
				sort(vec.begin(),vec.end(),cmp);
				for(int re j=0;j<(int)vec.size();++j){
					int x=vec[j].se;
					id[x]=x+now*2,x=id[x];
					if(j==0)Graph::addedge(i,x);
					if(j==(int)vec.size()-1)Graph::addedge(x,i+now);
					else Graph::addedge(x,vec[j+1].se+now*2);
					if(x<=now*2+na){
						++tot;
						int u=now*2+na+nb+tot;
						Graph::addedge(x,u);
						Graph::val[u]=vec[j].fi,id[x-now*2]=u;
					}
				}
			}
		}
	}
	
	inline void clear(){
		while(now){
			nd[now].clear();
			len[now]=fa[now]=0;
			memset(son[now],0,sizeof son[now]);
			memset(f[now],0,sizeof f[now]);
			--now;
		}
		last=now=1;tot=0;
	}
}

char s[N];
inline void solve(){
	std::cin>>(s+1);n=strlen(s+1);
	for(int re i=n;i;--i)SAM::push_back(s[i],i);
	SAM::init();
	na=getint();
	for(int re i=1,l,r;i<=na;++i){
		l=getint(),r=getint();
		SAM::locate(l,r,i);
	}
	nb=getint();
	for(int re i=1,l,r;i<=nb;++i){
		l=getint(),r=getint();
		SAM::locate(l,r,na+i);
	}
	SAM::build();
	int m=getint();
	while(m--){
		int x=getint(),y=getint();
		Graph::addedge(SAM::id[x],SAM::id[y+na]);
	}
	Graph::solve();
}

inline void init(){
	memset(SAM::id+1,0,sizeof(int)*(na+nb));
	memset(SAM::pos+1,0,sizeof(int)*n);
	SAM::clear();
	Graph::clear();
}

signed main(){
	int T=getint();
	while(T--)init(),solve();
	return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章