社区讨论

HackWA求调...

P10113[GESP202312 八级] 大量的工作沟通参与者 2已保存回复 3

讨论操作

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

当前回复
3 条
当前快照
1 份
快照标识符
@mhj3ig0t
此快照首次捕获于
2025/11/03 20:07
4 个月前
此快照最后确认于
2025/11/03 20:07
4 个月前
查看原帖
如题,求调
CPP
#include<bits/stdc++.h>
using namespace std;
#define int long long

const int MAX = 5e5+5;
int n, q, fa[MAX][35], dep[MAX];
vector<int> mp[MAX];

void dfs(int x, int f, int depth)
{
    dep[x] = depth;
    fa[x][0] = f;
    for(int i=1; i<=30; i++)
        fa[x][i] = fa[fa[x][i-1]][i-1];
    
    for(int i=0; i<mp[x].size(); i++)
        if(mp[x][i] != f) dfs(mp[x][i], x, depth+1);
}

int up(int x, int h)
{
    int i=0;
    while(h)
    {
        if(h & 1) x = fa[x][i];
        h >>= 1; i++;
    }
    return x;
}

int LCA(int a, int b)
{
    if(dep[a] < dep[b]) swap(a, b);
    a = up(a, dep[a]-dep[b]);
    if(a == b) return a;

    for(int i=30; i>=0; i--)
        if(fa[a][i] != fa[b][i])
            a = fa[a][i],
            b = fa[b][i];
    return fa[a][0];
}

signed main()
{
    // freopen("a.in", "r", stdin);

    cin >> n;
    for(int i=1; i<n; i++)
    {
        int f; cin >> f;
        mp[f].push_back(i);
        mp[i].push_back(f);
    }

    dfs(0, 0, 1);
    cin >> q; 
    for(int i=1; i<=q; i++)
    {
        int m, lca=0; cin >> m;
        for(int j=1; j<=m; j++)
        {
            int a; cin >> a;
            if(j == 1) lca = a;
            lca = LCA(a, lca);
        }
        cout << lca << endl;
    }

    return 0;
}

回复

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

正在加载回复...