社区讨论

麻烦帮看一下,是一道主席树模板

学术版参与者 3已保存回复 3

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
3 条
当前快照
1 份
快照标识符
@mhjo7n05
此快照首次捕获于
2025/11/04 05:47
4 个月前
此快照最后确认于
2025/11/04 05:47
4 个月前
查看原帖
题目在这里 1-5个点RE,我看过了讨论版的警示后人但是仍然没找到问题,麻烦帮看一下,谢谢
CPP
#include<bits/stdc++.h>
#define N 1000010
using namespace std;
long long n, m, last=0, cnt, a[N], b[N];
long long root[N], sum[N], ls[N], rs[N], node=0, new_node=0;
long long ans_sum[N], ans_ls[N], ans_rs[N];

struct edge{
    long long to, next;
}e[N<<2];
long long tmp=0, head[N];
void add(int u, int v){
    e[++tmp] = (edge){v, head[u]};
    head[u] = tmp;
}

int modify(int p, int l, int r, int pos){
    int np = ++node;
    sum[np] = sum[p]+1;
    if(l==r) return np;
    ls[np] = ls[p]; rs[np] = rs[p];
    int mid = l+r>>1;
    if(pos <= mid) ls[np] = modify(ls[p], l, mid, pos);
    else rs[np] = modify(rs[p], mid+1, r, pos);
    return np;
}

long long d[N], siz[N], fa[N], son[N];
void dfs1(int x, int f){
    d[x] = d[f]+1;
    fa[x] = f;
    siz[x] = 1;
    for(int i=head[x]; i; i=e[i].next){
        int y = e[i].to;
        if(y==fa[x]) continue;
        root[y] = modify(root[x], 1, cnt, a[y]);
        dfs1(y, x);
        siz[x] += siz[y];
        if(!son[x] || siz[y]>siz[son[x]])
            son[x] = y;
    }
}

long long number=0, top[N];
void dfs2(int x, int tp){
    top[x] = tp;
    if(!son[x]) return ;
    dfs2(son[x], tp);
    for(int i=head[x]; i; i=e[i].next){
        int y = e[i].to;
        if(y==fa[x] || y==son[x]) continue;
        dfs2(y, y);
    }
}

long long LCA(int x, int y){
   while(top[x] != top[y]){
      if(d[top[x]] < d[top[y]]) swap(x, y);
	  x = fa[top[x]];
   }
   return d[x]<d[y]?x:y;
}

long long query(int l, int r, int k, int u, int v, int lca, int fa){
    if(l==r) return l;
    long long mid = l+r>>1, ans;
    long long t=sum[ls[u]]-sum[ls[fa]]+sum[ls[v]]-sum[ls[lca]];
    if(t>=k) ans=query(l, mid, k, ls[u], ls[v], ls[lca], ls[fa]);
    else ans=query(mid+1, r, k-t, rs[u], rs[v], rs[lca], rs[fa]);
    return ans;
}
int main()
{
    freopen("P2633.in", "r", stdin);
    freopen("P2633.out", "w", stdout);
    scanf("%lld%lld", &n, &m);
    for(int i=1; i<=n; i++){
        scanf("%lld", b+i);
        a[i] = b[i];
    }
    sort(b+1, b+n+1);
    cnt = unique(b+1, b+n+1)-b-1;
    for(int i=1; i<=n; i++)
        a[i] = lower_bound(b+1, b+cnt+1, a[i])-b;
    for(int i=1; i<=n-1; i++){
        long long u, v;
        scanf("%lld%lld", &u, &v);
        add(u, v);
        add(v, u);
    }
    d[0]=0; sum[0]=0; ls[0]=0, rs[0]=0; root[0]=0;
    dfs1(1, 0);
    dfs2(1, 1);
    while(m--){    
        long long u, v, k;
        scanf("%lld%lld%lld", &u, &v, &k);
        u = u^last;
        long long lca = LCA(u, v);
        last = b[query(1, cnt, k, root[u], root[v], root[lca], root[fa[lca]])];
        cout << last << endl;
    }
    return 0;
}

回复

3 条回复,欢迎继续交流。

正在加载回复...