BZOJ_5415_[Noi2018]归程_kruscal重构树+倍增+最短路

BZOJ_5415_[Noi2018]归程_kruscal重构树+倍增

Description

www.lydsy.com/JudgeOnline/upload/noi2018day1.pdf

---

好久不写题解了，趁着周末都更一下吧。

处理出以1为起点的最短路dis。

考虑每次只能走海拔大于p的边，用kruscal重构树维护一个以海拔为关键字的最大生成树。

这样我们每次能走的连通块就是一个重构树中的一个子树。

倍增找一下是哪棵子树，子树取dis的最小值即可。

 

代码：

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125

#include <cstdio> #include <cstring> #include <algorithm> #include <cstdlib> #include <cmath> #include <ext/pb_ds/priority_queue.hpp> using namespace std; using namespace std; #define N 400050 #define M 400050 #define inf 2147483647 typedef long long ll; __attribute__((optimize("-O3")))inline char nc() {     static char buf[100000],*p1,*p2;     return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++; } __attribute__((optimize("-O3")))int rd() {     int x=0; char s=nc();     while(s<'0'||s>'9') s=nc();     while(s>='0'&&s<='9') x=(x<<3)+(x<<1)+s-'0',s=nc();     return x; } char pbuf[100000],*pp=pbuf; __attribute__((optimize("-O3")))void push(const char c) {     if(pp-pbuf==100000) fwrite(pbuf,1,100000,stdout),pp=pbuf;     *pp++=c; } __attribute__((optimize("-O3")))void write(int x) {     static int sta[35];     int top=0;     do{sta[top++]=x%10,x/=10;}while(x);     while(top) push(sta[--top]+'0'); } int head[N],to[M<<1],nxt[M<<1],cnt,f[23][N],mn[N],val[M<<1],n,m,dis[N],vis[N],Lg[N]; int fa[N],w[N]; int find(int x) {return fa[x]==x?x:fa[x]=find(fa[x]);} struct E {     int x,y,z,w;     bool operator < (const E &u) const {         return w>u.w;     } }e[M]; __gnu_pbds::priority_queue<pair<int,int> >q; __attribute__((optimize("-O3")))inline void add(int u,int v) {     to[++cnt]=v; nxt[cnt]=head[u]; head[u]=cnt; } __attribute__((optimize("-O3")))void add(int u,int v,int w) {     to[++cnt]=v; nxt[cnt]=head[u]; head[u]=cnt; val[cnt]=w; } __attribute__((optimize("-O3")))void dij() {     int i;     for(i=1;i<=n;i++) dis[i]=2147483647;     dis[1]=0;     memset(vis,0,sizeof(vis));     q.push(make_pair(0,1));     while(!q.empty()) {         int x=q.top().second,i; q.pop();         if(vis[x]) continue;         vis[x]=1;         if(dis[x]==inf) continue;         for(i=head[x];i;i=nxt[i]) {             if(dis[to[i]]>dis[x]+val[i]) {                 dis[to[i]]=dis[x]+val[i];                 q.push(make_pair(-dis[to[i]],to[i]));             }         }     } } __attribute__((optimize("-O3")))void dfs(int x) {     int i;     if(x<=n) mn[x]=dis[x];     else mn[x]=inf;     for(i=head[x];i;i=nxt[i]) {         f[0][to[i]]=x;         dfs(to[i]);         mn[x]=min(mn[x],mn[to[i]]);     } } __attribute__((optimize("-O3")))void solve() {     memset(head,0,sizeof(head)); cnt=0;     memset(f,0,sizeof(f));     n=rd(); m=rd();     int i,x,y,j;     for(i=1;i<=m;i++) {         e[i].x=rd(); e[i].y=rd(); e[i].z=rd(); e[i].w=rd(); add(e[i].x,e[i].y,e[i].z); add(e[i].y,e[i].x,e[i].z);     }     dij();     memset(head,0,sizeof(head)); cnt=0;     sort(e+1,e+m+1);     for(Lg[0]=-1,i=1;i<=2*n;i++) fa[i]=i,Lg[i]=Lg[i>>1]+1;     int tot=n;     for(i=1;i<=m;i++) {         int dx=find(e[i].x),dy=find(e[i].y);         if(dx==dy) continue;         tot++; fa[dx]=tot; fa[dy]=tot; add(tot,dx); add(tot,dy); w[tot]=e[i].w;     }     dfs(tot);     for(i=1;(1<<i)<=tot;i++) {         for(j=1;j<=tot;j++) {             f[i][j]=f[i-1][f[i-1][j]];         }     }     int opt,cas,S;     cas=rd(); opt=rd(); S=rd();     int ans=0;     while(cas--) {         x=rd(); y=rd();         x=(ll(x)+opt*ans-1)%n+1;         y=(ll(y)+opt*ans)%(S+1);         for(i=Lg[tot];i>=0;i--) {             if(!f[i][x]) continue;             if(w[f[i][x]]>y) x=f[i][x];         }         ans=mn[x];         write(ans); push('\n');     } } __attribute__((optimize("-O3")))int main() {     // freopen("return.in","r",stdin);     // freopen("return.out","w",stdout);     int T;     T=rd();     while(T--) solve();     fwrite(pbuf,1,pp-pbuf,stdout); }