BZOJ 2725: [Violet 6]故乡的梦

求出最短路径树，对于一个询问(x,y) 若不在树上S->T的链上，则答案不变，若在链上，考虑用一条非树边替换这条边，这条非树边必须跨越x->y这条边，线段树维护区间最小值

#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
#define pr pair<long long,int>
#define mp make_pair
#define sc second
using namespace std;
int cnt,n,m,last[200005],Lv[200005],stack[200005],Fa[200005],vis[200005],pre[200005],Vis[1000005];
long long dis[200005][2],Dep[200005],ANS[200005],tree[800005];
priority_queue<pr,vector<pr>,greater<pr> > q;
struct node{
	int to,next;
	long long val;
}e[400005];
struct node1{
	int x,y,val;
}E[200005];
void add(int a,int b,int c){
	e[++cnt].to=b;
	e[cnt].next=last[a];
	e[cnt].val=c;
	last[a]=cnt;
}
void Dijkstra(int S,int cas){
	for (int i=1; i<=n; i++) dis[i][cas]=1ll<<60,vis[i]=0;
	dis[S][cas]=0;
	q.push(mp(0ll,S));
	while (!q.empty()){
		int x=q.top().sc;
		q.pop();
		if (vis[x]) continue;
		vis[x]=1;
		for (int i=last[x]; i; i=e[i].next){
			int V=e[i].to;
			if (dis[V][cas]>dis[x][cas]+e[i].val){
				dis[V][cas]=dis[x][cas]+e[i].val;
				if (cas==0) pre[V]=x;
				q.push(mp(dis[V][cas],V));
			}
		}
	}
}
void Pre(int x,int fa,long long dep){
	Fa[x]=fa,Dep[x]=dep;
	for (int i=last[x]; i; i=e[i].next){
		int V=e[i].to;
		if (V==fa) continue;
		Pre(V,x,dep+e[i].val);
	}
}
void solve(int x,int lv){
	vis[x]=1;
	Lv[x]=lv;
	for (int i=last[x]; i; i=e[i].next){
		int V=e[i].to;
		if (vis[V]) continue;
		solve(V,lv);
	}
}
bool cmp(node a,node b){
	return a.to<b.to;
}
void insert(int t,int l,int r,int x,long long key){
	if (l==r){
		tree[t]=min(tree[t],key);
		return;
	}
	int mid=(l+r)>>1;
	if (x<=mid) insert(t<<1,l,mid,x,key);
	else insert(t<<1|1,mid+1,r,x,key);
	tree[t]=min(tree[t<<1],tree[t<<1|1]);
}
long long query(int t,int l,int r,int x,int y){
	if (r<x || l>y) return 1ll<<60;
	if (l>=x && r<=y) return tree[t];
	int mid=(l+r)>>1;
	return min(query(t<<1,l,mid,x,y),query(t<<1|1,mid+1,r,x,y));
}
void build(int t,int l,int r){
	if (l==r){
		tree[t]=1ll<<60;
		return;
	}
	int mid=(l+r)>>1;
	build(t<<1,l,mid);
	build(t<<1|1,mid+1,r);
	tree[t]=max(tree[t<<1],tree[t<<1|1]);
}
int main(){
	scanf("%d%d",&n,&m);
	for (int i=1; i<=m; i++){
		scanf("%d%d%d",&E[i].x,&E[i].y,&E[i].val);
		add(E[i].x,E[i].y,E[i].val);
		add(E[i].y,E[i].x,E[i].val);
	}
	int S,T;
	scanf("%d%d",&S,&T);
	Dijkstra(S,0);
	Dijkstra(T,1);
	cnt=0;
	memset(vis,0,sizeof(vis));
	for (int i=1; i<=n; i++) last[i]=0;
	for (int i=1; i<=n; i++)
		if (pre[i]) {
			add(pre[i],i,dis[i][0]-dis[pre[i]][0]);
			add(i,pre[i],dis[i][0]-dis[pre[i]][0]);
		}
	Lv[S]=1;
	Pre(S,0,0);
	int now=T,top=0;
	while (now){
		stack[++top]=now;
		vis[now]=1;
		Vis[now]=top;
		now=Fa[now];
	}
	for (int i=1; i<=top/2; i++) swap(stack[i],stack[top-i+1]);
	for (int i=1; i<=top; i++) solve(stack[i],i);
	int cnt=0;
	for (int i=1; i<=m; i++){
		int x=E[i].x,y=E[i].y;
		if (Vis[x] && Vis[y] && (Vis[y]==Vis[x]+1 || Vis[x]==Vis[y]+1)) continue;
		int X=Lv[x],Y=Lv[y];
		if (X>Y) swap(X,Y),swap(x,y);
		e[++cnt]=(node){X,Y,dis[x][0]+dis[y][1]+E[i].val};
	}
	sort(e+1,e+cnt+1,cmp);
	build(1,1,top);
	int Top=1;
	for (int i=1; i<top; i++){
		while (Top<=cnt && e[Top].to<=i){
			insert(1,1,top,e[Top].next,e[Top].val);
			Top++;
		}
		ANS[i]=query(1,1,top,i+1,top);
	}
	int q;
	scanf("%d",&q);
	int cc=0;
	while (q--){
		cc++;
		int x,y;
		scanf("%d%d",&x,&y);
		if (Vis[x] && Vis[y] && (Vis[x]==Vis[y]+1 || Vis[y]==Vis[x]+1)){
			if (dis[T][0]==1ll<<60) printf("Infinity\n");
			else{
				int X=Lv[x],Y=Lv[y];
				if (X>Y) swap(X,Y);
				if (ANS[X]!=1ll<<60) printf("%lld\n",ANS[X]);
				else printf("Infinity\n");
			}
		}
		else {
			if (dis[T][0]!=1ll<<60) printf("%lld\n",dis[T][0]);
			else printf("Infinity\n");
		}
	}
	return 0;
}