1. 程式人生 > >【UOJ】【kruskal重構樹】【NOI2018】歸程

【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; }