ACM ICPC China final G Pandaria-CSDN博客

ACM ICPC China final G Pandaria

ACM ICPC China final G Pandaria

题意：给一张$n$个点$m$条边的无向图，$c[i]$代表点$i$的颜色,每条边有一个权值
现在有q个询问如下
u w代表从点u出发，每次只能走权值不超过w的边，经过的颜色中次数最多的是哪种颜色，若有多种，输出编号最小的。
$ 1<=n<=1e5$
$ 1<=m,q<=2e5$
$ 1<=c_i<=n$
$ 1<=w<=1e6$
ps:强制在线

思路：把边从小到大排序，依次合并，现在就是如何维护每个连通块的答案了。
用主席树维护区间最大值和合并操作，查询就是二分最大值的最左位置
如果这题不强制在线的话，那么我们可以把查询按w从小大排序，模拟过去即可
强制在线就多了一些东西，考虑合并的过程，其实是一颗自底向上的树，越往上边权越大
前面主席树已经算出每个结点的答案，如何找到要查询的历史版本，通过pa数组自底向上倍增即可

#include<bits/stdc++.h>
#define P pair<int,int>
#define LL long long
#define ls(i) seg[i].lc
#define rs(i) seg[i].rc
using namespace std;

const int mod = 1e9 + 7;
const int N = 1e5 + 1;
const int M = 2 * N;
const int maxn = 3 * N;
const int maxh = 25;
int n, m,tott;
int c[N];
int ans[maxn];
struct Tree{
    int ls, rs, w;
}tree[maxn];
struct node{
    int lc,rc,cnt;
}seg[maxn * maxh];///主席树维护的区间的最大值，二分查询最大值所在的最左位置
int root[maxn * maxh];
struct Edge{
    int u, v, w;
    Edge(){};
    bool operator<(const Edge&rhs)const{
        return w < rhs.w;
    }
}e[M];
void pushup(int rt){
    seg[rt].cnt = max(seg[ls(rt)].cnt,seg[rs(rt)].cnt);
}
void update(int &rt,int l,int r,int pos,int val){
    seg[++tott] = seg[rt];
    rt = tott;
    if(l == r){
        seg[rt].cnt += val;
        return ;
    }
    int m = (l + r) >> 1;
    if(pos <= m) update(ls(rt),l,m,pos,val);
    else update(rs(rt),m+1,r,pos,val);
    pushup(rt);
}
int query(int rt,int l,int r){
    if(l == r) return l;
    int m = (l + r) >> 1;
    if(seg[ls(rt)].cnt >= seg[rs(rt)].cnt) return query(ls(rt),l,m);
    return query(rs(rt),m+1,r);
}
int Merge(int rt1,int rt2,int l,int r){
    if(!rt1 || !rt2) return rt1 + rt2;///最多只有n个点保证了合并的复杂度科学
    if(l == r){
        seg[rt1].cnt += seg[rt2].cnt;
        return rt1;
    }
    int m = l + r >> 1;
    ls(rt1) = Merge(ls(rt1),ls(rt2),l,m);
    rs(rt1) = Merge(rs(rt1),rs(rt2),m+1,r);
    pushup(rt1);
    return rt1;
}
int pa[maxn][maxh];
int Find(int x){
    return pa[x][0] == x?x:pa[x][0] = Find(pa[x][0]);
}
int ask(int u,int w){ ///倍增找接近的历史版本的答案
    for(int i = maxh - 1;i >= 0;i--) if(pa[u][i] && tree[pa[u][i]].w <= w) u = pa[u][i];
    return ans[u];
}
int main(){

    int T, cas = 1;
    cin>>T;
    while(T--){
        scanf("%d%d",&n,&m);
        tott = 0;
        for(int i = 1;i <= n;i++) {
                scanf("%d",c + i);
                root[i] = root[0];
                update(root[i],1,n,c[i],1);
                pa[i][0] = i;
                tree[i].w = 0; ///叶子结点权值为0
                ans[i] = query(root[i],1,n);
        }
        for(int i = 0;i < m;i++) scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
        int tot = n;
        sort(e, e + m);
        for(int i = 0;i < m;i++){
            int u = Find(e[i].u), v = Find(e[i].v), w = e[i].w;
            if(u == v) continue;
            tree[++tot].w = w;///合并得到新的父结点
            tree[tot].ls = u, tree[tot].rs = v;///两个孩子
            pa[tot][0] = pa[u][0] = pa[v][0] = tot;
            root[tot] = Merge(root[u],root[v],1,n);
            ans[tot] = query(root[tot],1,n);
        }
        for(int i = n + 1;i <= tot;i++) pa[tree[i].ls][0] = pa[tree[i].rs][0] = i;///前面为路径压缩的用途，现在改为上一级祖先用途
        for(int i = 1;i <= tot;i++) pa[i][0] = pa[i][0] == i?0:pa[i][0];
        for(int i = 1;i < maxh;i++)
            for(int j = 1;j <= tot;j++) pa[j][i] = pa[pa[j][i-1]][i-1];
        int q, last = 0, u, w;
        printf("Case #%d:\n",cas++);
        scanf("%d",&q);
        while(q--){
            scanf("%d%d",&u,&w);
            u ^= last, w ^= last;
            printf("%d\n",last = ask(u,w));
        }
    }
    return 0;
}

转载于:https://www.cnblogs.com/jiachinzhao/p/7260494.html