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

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

Description

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

---

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

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

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

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

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

代码：

#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);

}