【UOJ】【kruskal重构树】【NOI2018】归程

按照高度建最大生成树，构造kruskal重构树，每次连边时新建一个节点表示边权连到两端的父亲上。这样一棵树满足小根堆的性质。所以可以倍增跳到最顶端，然后答案就是子树里的最小权值（这里的权值为到1的最短路)。

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
inline char nc(){static char buf[100000],*i=buf,*j=buf;return i==j&&(j=(i=buf)+fread(buf,1,100000,stdin),i==j)?EOF:*i++; }
inline int _read(){char ch=nc();int sum=0;while(!(ch>='0'&&ch<='9'))ch=nc();while(ch>='0'&&ch<='9')sum=sum*10+ch-48,ch=nc();return sum;}
const int maxn=400006,maxe=400006;
int T,n,N,e,tot,ans,fa[maxn],dis[maxn],lnk[maxn],son[maxe*2],w[maxe*2],nxt[maxe*2],f[maxn][20];
bool vis[maxn];
struct side{
    int x,y,h,w;
    bool operator <(const side&b)const{return h>b.h;}
}a[maxe];
struct data{
    int x,s;
    bool operator <(const data&b)const{return s>b.s;}
};
priority_queue<data,vector<data> >heap;
int get(int x){return fa[x]==x?x:fa[x]=get(fa[x]);}
void add(int x,int y,int z){nxt[++tot]=lnk[x];son[tot]=y;lnk[x]=tot;w[tot]=z;}
void add(int x,int y){nxt[++tot]=lnk[x];son[tot]=y;lnk[x]=tot;}
void dij(){
    memset(dis,63,sizeof(dis));memset(vis,0,sizeof(vis));dis[1]=0;
    data x;x.x=1;x.s=0;heap.push(x);
    while(!heap.empty()){
        x=heap.top();heap.pop();while(!heap.empty()&&vis[x.x])x=heap.top(),heap.pop();if(vis[x.x])break;vis[x.x]=1;
        for(int j=lnk[x.x];j;j=nxt[j]) if(x.s+w[j]<dis[son[j]]){
            dis[son[j]]=x.s+w[j];data y;y.x=son[j];y.s=dis[son[j]];heap.push(y);
        }
    }
}
void dfs(int x){
    for(int j=1;j<=19;j++)f[x][j]=f[f[x][j-1]][j-1];
    for(int j=lnk[x];j;j=nxt[j])f[son[j]][0]=x,dfs(son[j]);
}
int find(int x,int dep){
    for(int j=19;j>=0;j--) if(f[x][j]&&w[f[x][j]]>dep)x=f[x][j];
    return dis[x];
}
int main(){
    freopen("return.in","r",stdin);freopen("return.out","w",stdout);
    T=_read();
    while(T--){
        tot=0;memset(lnk,0,sizeof(lnk));
        N=n=_read();e=_read();
        for(int i=1;i<=e;i++){
            a[i].x=_read();a[i].y=_read();a[i].w=_read();a[i].h=_read();
            add(a[i].x,a[i].y,a[i].w);add(a[i].y,a[i].x,a[i].w);
        }dij();tot=0;memset(lnk,0,sizeof(lnk));
        sort(a+1,a+1+e);
        for(int i=1;i<=n;i++)fa[i]=i;
        for(int i=1;i<=e;i++){
            int x=get(a[i].x),y=get(a[i].y);if(x==y)continue;
            fa[x]=++n;fa[y]=n;fa[n]=n;w[n]=a[i].h;dis[n]=min(dis[x],dis[y]);add(n,x);add(n,y);
        }
        f[n][0]=0;dfs(n);ans=0;int Q=_read(),K=_read(),S=_read();
        while(Q--){
            int x=_read(),d=_read();x=(x+K*ans-1)%N+1;d=(d+K*ans)%(S+1);printf("%d\n",ans=find(x,d));
        }
    }
    return 0;
}