#2718. 「NOI2018」归程 kruskal重构树

链接

https://loj.ac/problem/2718

思路

我们希望x所在的连通块尽量的大,而且尽量走高处
离线的话可以询问排序,kruskal过程中更新答案
在线就要用kruskal重构树
这kruskal重构树的话,看图就明白了
叶子节点都是原树节点
非叶子节点都是边
按照从大到小的顺序依次加边(是深度不是长度)
如果连通块已经在一起就不联通,其他两个最大节点和这个边(新建节点)连边
看图就是很明白
graph.png
我们发现,重构树的根到任意节点是单调的,也就是说,这是个二叉堆啊
那两点间联通的最小需要深度就是lca(x,y)这条边的深度
询问就是s这个点最远能向上跳到的最远点的子树答案
可能我讲的听不懂,那就看图自己想想吧
复杂度最多一个log

大体流程

多组数据,进入solve
dij预处理dis
kruskal利用并查集重构出树来
lca树剖预处理重构树
再dfs统计子树ans
之后在线查询lca
输出lca这个点的ans

细节

并查集最好优化下,因为他重构出来的树很容易感觉成哈夫曼树的形状
并查集按秩合并和随机rand时间差别不大
树剖链外跳,链内二分

错误

写了比较慢、、(不过一遍60TLE还是很嗨的)
错误只有一个,dij的堆忘记了默认大顶堆,导致复杂度不对,T死了

代码

/*
大体流程:
多组数据,进入solve
dij预处理dis
kruskal重构出树来
lca树剖预处理重构树(其实就是二分)
再dfs统计子树ans 
之后在线查询
*/
#include <iostream>
#include <ctime>
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <queue>
#include <utility>
#include <algorithm>
using namespace std;
const int N=1e6+7;
int read() {
    int x=0,f=1;char s=getchar();
    for(;s>'9'||s<'0';s=getchar()) if(s=='-') f=-1;
    for(;s>='0'&&s<='9';s=getchar()) x=x*10+s-'0';
    return x*f;
}
int n,m,rt;
int dis[N],w[N<<1],dsr[N<<1];
struct node {
    int u,v,l,a;
    bool operator < (const node &b) const {
        return a>b.a;
    }
} edge[N];
struct noi_2018_t1 {
    int v,nxt;
} e[N<<1];
int head[N<<1],add_tot;
void add(int u,int v) {
//  cout<<u<<" "<<v<<"\n";
    e[++add_tot].v=v;
    e[add_tot].nxt=head[u];
    head[u]=add_tot;
}
namespace bcj {
    int fa[N<<1],siz[N<<1],id[N<<1];
    int find(int x) {
        return fa[x]==x ? x : fa[x]=find(fa[x]);
    }
    void uu(int fx,int fy) {
        fx=find(fx),fy=find(fy);
        id[fx]=id[fy]=max(id[fx],id[fy]);
        if(siz[fx]<=siz[fy]) {
            fa[fx]=fy;
            if(siz[fx]==siz[fy]) siz[fy]++;
        } else {
            fa[fy]=fx;
        }
    }
}
namespace get_lca {
    int fa[N<<1],top[N<<1],siz[N<<1],son[N<<1],dep[N<<1],idx[N<<1],js,frm[N<<1];
    void dfs1(int u,int f) {
        dep[u]=dep[f]+1;
        fa[u]=f;
        siz[u]=1;
        for(int i=head[u];i;i=e[i].nxt) {
            int v=e[i].v;
            dfs1(v,u);
            siz[u]+=siz[v];
            if(siz[son[u]]<siz[v]) son[u]=v;
        }
    }
    void dfs2(int u,int topf) {
        top[u]=topf;
        idx[u]=++js;
        frm[js]=u;
        if(!son[u]) return;
        dfs2(son[u],topf);
        for(int i=head[u];i;i=e[i].nxt) {
            int v=e[i].v;
            if(!idx[v]) dfs2(v,v);
        }
    }
    int lca(int x,int val) {
        int ans=x;
        while(w[top[x]]>val) ans=top[x],x=fa[top[x]];
        int l=idx[top[x]],r=idx[x];
        while(l<=r) {
//          cout<<l<<" "<<r<<"\n";
            int mid=(l+r)>>1;
            if(w[frm[mid]]>val) ans=frm[mid],r=mid-1;
            else l=mid+1;
        }
        return ans;
    }   
    void init() {
        dfs1(rt,0);
        dfs2(rt,rt);
    }
}
namespace dij {
    struct node {
        int v,nxt,q;
    }e[N<<1];
    int head[N<<1],tot;
    void add(int u,int v,int q) {
        e[++tot].v=v;
        e[tot].q=q;
        e[tot].nxt=head[u];
        head[u]=tot;
    }
    void dij() {
        priority_queue<pair<int,int> > q;
        q.push(make_pair(0,1));
        dis[1]=0;
        while(!q.empty()) {
            pair<int,int> u=q.top();
            q.pop();
            if(dis[u.second]!=-u.first) continue;
            for(int i=head[u.second];i;i=e[i].nxt) {
                int v=e[i].v;
                if(dis[v]>dis[u.second]+e[i].q) {
                    dis[v]=dis[u.second]+e[i].q;
                    q.push(make_pair(-dis[v],v));
                }
            }
        }
    }
    void work() {
        memset(dis,0x3f,sizeof(dis));
        memset(head,0,sizeof(head));
        tot=0;
        for(int i=1;i<=m;++i) {
            add(edge[i].u,edge[i].v,edge[i].l);
            add(edge[i].v,edge[i].u,edge[i].l);
        }
        dij();
    }
}
namespace kruskal {
    void work() {
        for(int i=1;i<n+n;++i) bcj::fa[i]=i;
        for(int i=1;i<n+n;++i) bcj::id[i]=i;
        sort(edge+1,edge+1+m);
        for(int i=1;i<=n;++i) w[i]=0x3f3f3f3f;
        for(int i=1,js=n;i<=m;++i) {
            int fx=bcj::find(edge[i].u),fy=bcj::find(edge[i].v);
            if(fx!=fy) {
                rt=++js;
                add(js,bcj::id[fx]),add(js,bcj::id[fy]);
                w[js]=edge[i].a;
                bcj::uu(js,fy);
                bcj::uu(js,fx);
                if(js==n+n-1) return;
            }
        }
    }
}
void clear() {
    add_tot=0;
    dij::tot=0;
    get_lca::js=0;
    memset(dij::head,0,sizeof(dij::head));
    memset(get_lca::siz,0,sizeof(get_lca::siz));
    memset(bcj::siz,0,sizeof(bcj::siz)); 
    memset(head,0,sizeof(head));
    memset(dis,0x3f,sizeof(dis));
    memset(get_lca::idx,0,sizeof(get_lca::idx));
    memset(get_lca::son,0,sizeof(get_lca::son));
}
void dfs(int u,int f) {
    dsr[u]=dis[u];
    for(int i=head[u];i;i=e[i].nxt) {
        int v=e[i].v;
        dfs(v,u);
        dsr[u]=min(dsr[u],dsr[v]);
    }
}
void solve() {
    dij::work();
    // cout<<"1\n";
    kruskal::work();
    // cout<<"2\n";
    get_lca::init();
    // cout<<"3\n";
    dfs(rt,0);
    int Q=read(),k=read();
    int S=read(),lastans=0;
    while(Q--) {
        int vv=read(),pp=read();
        int v=(vv+k*lastans-1)%n+1;
        int p=(pp+k*lastans)%(S+1);
        int tmp=get_lca::lca(v,p);
        lastans=dsr[tmp];
        printf("%d\n",lastans);
    }
}
int main() {
    freopen("return.in","r",stdin);
    freopen("return.out","w",stdout);
    int T=read();
    while(T--) {
        n=read(),m=read();
        for(int i=1;i<=m;++i) {
            edge[i].u=read(),edge[i].v=read();
            edge[i].l=read(),edge[i].a=read();
        }
        clear();
        solve();
    }
    return 0;
}
全部评论

相关推荐

不愿透露姓名的神秘牛友
11-21 19:05
面试官_我太想进步了:混学生会的,难怪简历这么水
点赞 评论 收藏
分享
沉淀一会:1.同学你面试评价不错,概率很大,请耐心等待; 2.你的排名比较靠前,不要担心,耐心等待; 3.问题不大,正在审批,不要着急签其他公司,等等我们! 4.预计9月中下旬,安心过节; 5.下周会有结果,请耐心等待下; 6.可能国庆节前后,一有结果我马上通知你; 7.预计10月中旬,再坚持一下; 8.正在走流程,就这两天了; 9.同学,结果我也不知道,你如果查到了也告诉我一声; 10.同学你出线不明朗,建议签其他公司保底! 11.同学你找了哪些公司,我也在找工作。
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务