#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;
}
全部评论

相关推荐

废物一个0offer:认真的吗二本本科找人工智能岗位
点赞 评论 收藏
分享
06-15 18:44
黄淮学院 Java
Lynn012:如果是居民楼还是算了吧,看着有点野呢
点赞 评论 收藏
分享
家人们,我现在真的好纠结。我是26届的,目前还没有实习过。我现在的情况是,想参加秋招,但是感觉自己的简历特别空,没有实习经历会不会秋招直接凉凉啊?可我又听说现在很多公司对26届实习生也不太感冒,说什么不确定性大。而且我最近在准备考公,时间上也有点冲突。要是把时间花在实习上,备考时间就少了。但要是不实习,又怕以后就业有问题😫有没有懂行的友友帮我分析分析:26届现在不实习,秋招找工作真的会很难吗?考公和实习该怎么平衡啊?如果现在不实习,考完公再去找实习还来得及吗?真的太焦虑了,希望大家能给我点建议🙏
小破站_程序员YT:我可能和大家的观点不一样。人的精力是有限的,不能既要还要。你又想实习又想考公最后又要秋招上岸,我觉得哪有那么多的选择。你如果想考上岸,那就全力以赴。如果想秋招上岸,就继续投实习,投没了,就继续准备秋招,秋招不行继续春招。别到最后,考公没上岸,觉得是花了时间浪费在找实习上了, 秋招没上岸,觉得是浪费时间准备考公去了。我是认为很难说可以去平衡 不喜勿喷,可以叫我删除
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务