BZOJ3073 [Pa2011]Journeys

一眼线段树优化建图01bfs

考虑一下怎么优化

搞两棵线段树，线段树上的边都是0，对于一次新建一个单向边（题里建的是双向边，拆成俩单向边即可），新建两个点，两个点之间连一个边权为1的，然后第一颗线段树对应区间的点连第一个新建的点边权0，第二个新建的点连第二棵线段树对应区间的点边权0

第二个线段树的每个叶子向第一个线段树的对应叶子连边权0

然后跑01bfs即可

#include<iostream>
#include<cstring>
#include<ctime>
#include<cmath>
#include<algorithm>
#include<iomanip>
#include<cstdlib>
#include<cstdio>
#include<map>
#include<bitset>
#include<set>
#include<stack>
#include<vector>
#include<queue>
using namespace std;
#define MAXN 5000010
#define MAXM 30000010
#define ll long long
#define eps 1e-8
#define MOD 1000000007
#define INF 1000000000
struct vec{
	int to;
	int fro;
	int v;
};
vec mp[MAXM];
int tai[MAXN],cnt,tot;
int q[MAXN],hd,tl;
int Q[MAXN],HD,TL;
int d[MAXN];
int n,m,s;
int rt1,rt2;
int son[MAXN][2];
int p[MAXN];
inline void be(int x,int y,int z){
	mp[++cnt].to=y;
	mp[cnt].fro=tai[x];
	tai[x]=cnt;
	mp[cnt].v=z;
}
inline void bde(int x,int y,int z){
	be(x,y,z);
	be(y,x,z);
}
void build(int &x,int &xx,int y,int z){
	x=++tot;
	xx=++tot;
	if(y==z){
		be(xx,x,0);
		p[y]=x;
		return ;
	}
	int mid=y+z>>1;
	build(son[x][0],son[xx][0],y,mid);
	build(son[x][1],son[xx][1],mid+1,z);
	be(son[x][0],x,0);
	be(son[x][1],x,0);
	be(xx,son[xx][0],0);
	be(xx,son[xx][1],0);
}
void ins(int x,int y,int z,int l,int r,int f){
	if(y==l&&z==r){
		if(!f){
			be(x,tot,0);
		}else{
			be(tot,x,0);
		}
		return ;
	}
	int mid=y+z>>1;
	if(r<=mid){
		ins(son[x][0],y,mid,l,r,f);
	}else if(l>mid){
		ins(son[x][1],mid+1,z,l,r,f);
	}else{
		ins(son[x][0],y,mid,l,mid,f);
		ins(son[x][1],mid+1,z,mid+1,r,f);
	}
}
void BE(int x,int y,int xx,int yy){
	tot++;
	ins(rt1,1,n,x,y,0);
	tot++;
	ins(rt2,1,n,xx,yy,1);
	be(tot-1,tot,1);
}
int main(){
	int i,x,y,xx,yy;
	scanf("%d%d%d",&n,&m,&s);
	build(rt1,rt2,1,n);
	for(i=1;i<=m;i++){
		scanf("%d%d%d%d",&x,&y,&xx,&yy);
		BE(x,y,xx,yy);
		BE(xx,yy,x,y);
	}
	d[q[tl++]=p[s]]=1;
	while(hd!=tl){
		HD=TL=0;
		Q[TL++]=q[hd++];
		while(HD!=TL){
			x=Q[HD++];
			for(i=tai[x];i;i=mp[i].fro){
				y=mp[i].to;
				if(!d[y]){
					if(mp[i].v){
						d[y]=d[x]+1;
						q[tl++]=y;
					}else{
						d[y]=d[x];
						Q[TL++]=y;
					}
				}
			}
		}
	}
	for(i=1;i<=n;i++){
		printf("%d\n",d[p[i]]-1);
	}
	return 0;
}

/*
5 3 4
1 2 4 5
5 5 4 4
1 1 3 3

*/