LOJ#2718. 「NOI2018」归程

题目描述

题解：

对于权值大于等于或者小于等于某一个值的询问我们可以考虑用kruskal重构树来解决。
kruskal重构树是指在对一张无向图进行kruskal求出最小/最大生成树的同时，把当前的两颗子树合并到新的节点上，作为它的两个子节点，并且把新节点的点权赋为当前这条边的边权，最后变成一颗树。
那么这样合并以后，有一些好的性质：
1.两个点之间的路径经过的最大边最小/最小边最大时，只要求重构树的lca就好了。
2.重构树中某一节点的子树中的边权都大于/小于当前点。

对于这题，我们首先可以轻而易举地求出每个点到1的最短距离。
接下来的重点对于每次询问求出从起始节点s到一个节点，它的海拔足够高并且到1的距离最短。那么我们就可以直接kruskal重构树，对于海拔从大到小加入，那么我们只要统计一个点的子树中最小的到1的距离，就是某一次询问的答案了。求出这个只要用倍增维护一下即可。

代码如下：

#include<bits/stdc++.h>
using namespace std;
const int maxn=400005,maxm=800005;
int n,m,Q,k,S,T,cnt,lnk[maxn],son[2*maxm],nxt[2*maxm],w[2*maxm],h[2*maxm],tot,dis[maxn];
int minn[maxn],f[maxn][21],fa[maxn];
bool vis[maxn];
struct dyt{
	int x,id;
	bool operator <(const dyt &b) const{return x>b.x;}
};
struct node{
	int x,y,h,id;
	bool operator <(const node &b) const{return h>b.h;}
}a[maxm];
priority_queue<dyt> hep;
void clr(){
	tot=0; memset(lnk,0,sizeof(lnk));
}
void add(int x,int y,int z,int z1){
	son[++tot]=y,w[tot]=z,h[tot]=z1,nxt[tot]=lnk[x],lnk[x]=tot;
	son[++tot]=x,w[tot]=z,h[tot]=z1,nxt[tot]=lnk[y],lnk[y]=tot;
}
void adde(int x,int y){
	son[++tot]=y,nxt[tot]=lnk[x],lnk[x]=tot;
}
void Dij(){
	memset(vis,0,sizeof(vis));
	memset(dis,63,sizeof(dis));
	dis[1]=0; hep.push((dyt){0,1});
	while (!hep.empty()){
		dyt x=hep.top(); hep.pop();
		while (vis[x.id]&&(!hep.empty())) {x=hep.top(); hep.pop();}
		vis[x.id]=1;
		for (int j=lnk[x.id];j;j=nxt[j])
			if (!vis[son[j]]&&dis[x.id]+w[j]<dis[son[j]])
				dis[son[j]]=dis[x.id]+w[j],hep.push((dyt){dis[son[j]],son[j]});
	}
}
int getfa(int x){if (fa[x]!=x) fa[x]=getfa(fa[x]); return fa[x];}
void dfs(int x,int fat){
	f[x][0]=fat,minn[x]=dis[x];
	for (int i=1;i<=20;i++) f[x][i]=f[f[x][i-1]][i-1];
	for (int j=lnk[x];j;j=nxt[j])
		if (son[j]!=fat) {
			dfs(son[j],x);
			minn[x]=min(minn[x],minn[son[j]]);
		}
}
void kruskal(){
	tot=0;
	memset(lnk,0,sizeof(lnk));
	memset(w,0,sizeof(w));
	memset(minn,63,sizeof(minn));
	for (int i=1;i<=n;i++) fa[i]=i,w[i]=(1<<30);
	cnt=n; sort(a+1,a+1+m);
	for (int i=1;i<=m;i++){
		int fax=getfa(a[i].x),fay=getfa(a[i].y);
		if (fax!=fay){
			cnt++,fa[cnt]=cnt,w[cnt]=a[i].h,fa[fax]=cnt,fa[fay]=cnt;
			dis[cnt]=min(dis[fax],dis[fay]);
			adde(cnt,fax); adde(cnt,fay);
		}
	}
	dfs(cnt,0); w[0]=-1;
}
int doit(int x,int y){
	int u=x;
	for (int i=20;i>=0;i--)
		if (w[f[u][i]]>y) u=f[u][i];
	return minn[u];
}
int main(){
	freopen("return.in","r",stdin);
	freopen("return.out","w",stdout);
	scanf("%d",&T);
	while (T--){
		clr();
		scanf("%d%d",&n,&m);
		for (int i=1;i<=m;i++) {
			int xx; scanf("%d%d%d%d",&a[i].x,&a[i].y,&xx,&a[i].h);
			add(a[i].x,a[i].y,xx,a[i].h);
		}
		Dij(); kruskal();
		scanf("%d%d%d",&Q,&k,&S);
		int lastans=0;
		for (int i=1;i<=Q;i++){
		 	int x,y; scanf("%d%d",&x,&y);
		 	x=(x+k*lastans-1)%n+1;
		 	y=(y+k*lastans)%(S+1);
		 	lastans=doit(x,y);
		 	printf("%d\n",lastans);
		}
	}
	return 0;
}