2018.09.26 bzoj5218: [Lydsy2017省队十连测]友好城市(回滚莫队)

传送门
比较简单的一道回滚莫队吧。
每次询问用bitset优化kosaraju统计答案。
就是有点难调。
然后向dzyo学长学习了回滚莫队的一种简洁的实现方式,就是直接建立一个 s q r t ( m ) ∗ s q r t ( m ) sqrt(m)*sqrt(m) sqrt(m)sqrt(m)的动态数组按块存储询问。
这样好写得多。
还有就是学习了korasaju用bitset优化的写法。
代码:

#include
#define N 155
#define M 300005
#define sig 575
using namespace std;
inline int read(){
	int ans=0;
	char ch=getchar();
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
	return ans;
}
bitset<N>vis,e[N],g[N],tmp1[N],tmp2[N];
int n,m,q,U[M],V[M],bg[M],ed[M],tot=0,out[N],cnt,ans[M];
struct Query{int l,r,id;};
inline bool operator<(Query a,Query b){return a.r<b.r;}
vector<Query>qr[1005][1005];
inline void add(int u,int v){e[u].set(v),g[v].set(u);}
inline void dfs1(int p){
	bitset<N>tmp;
	vis[p]=1;
	while((tmp=e[p]^(e[p]&vis)),tmp.any()){
		int v=tmp._Find_first();
		dfs1(v);
	}
	out[++cnt]=p;
}
inline int dfs2(int p){
	bitset<N>tmp;
	vis[p]=1;
	int ret=1;
	while((tmp=g[p]^(g[p]&vis)),tmp.any()){
		int v=tmp._Find_first();
		ret+=dfs2(v);
	}
	return ret;
}
inline int kosaraju(){
	vis.reset(),cnt=0;
	for(int i=1;i<=n;++i)if(!vis[i])dfs1(i);
	vis.reset();
	int ret=0;
	for(int i=n;i;--i){
		if(!vis[out[i]]){
			int tmp=dfs2(out[i]);
			ret+=tmp*(tmp-1)/2;
		}
	}
	return ret;
}
inline void solve(int l,int r,int id){
	for(int i=l;i<=r;++i)add(U[i],V[i]);
	ans[id]=kosaraju();
	for(int i=1;i<=n;++i)e[i].reset(),g[i].reset();
}
int main(){
//	freopen("friend.in","r",stdin);
//	freopen("my.out","w",stdout);
	n=read(),m=read(),q=read();
	for(int i=1;i<=m;++i)U[i]=read(),V[i]=read();
	for(int l=1,r;l<=m;l=r+1)r=min(l+sig-1,m),bg[++tot]=l,ed[tot]=r;
	for(int i=1;i<=q;++i){
		int l=read(),r=read();
		if(r-l+1<=2*sig)solve(l,r,i);
		else qr[(l-1)/sig+2][r/sig].push_back((Query){l,r,i});
	}
	for(int i=1;i<=tot;++i)for(int j=1;j<=tot;++j)sort(qr[i][j].begin(),qr[i][j].end());
	for(int i=1;i<=tot;++i){
		int r=ed[i-1];
		for(int j=i;j<=tot;++j){
			while(r<ed[j])++r,add(U[r],V[r]);
			for(int k=0;k<qr[i][j].size();++k){
				Query pos=qr[i][j][k];
				while(r<pos.r)++r,add(U[r],V[r]);
				for(int l=1;l<=n;++l)tmp1[l]=e[l],tmp2[l]=g[l];
				for(int l=bg[i]-1;l>=pos.l;--l)add(U[l],V[l]);
				ans[pos.id]=kosaraju();
				for(int l=1;l<=n;++l)e[l]=tmp1[l],g[l]=tmp2[l];
			}
		}
		for(int j=1;j<=n;++j)e[j].reset(),g[j].reset();
	}
	for(int i=1;i<=q;++i)printf("%d\n",ans[i]);
	return 0;
}

你可能感兴趣的:(#,分块,#,回滚莫队,#,korasaju)