22牛客多校3 A.Ancestor(LCA + 枚举)

传送门

大致题意 : 有两棵节点数为N的树A, B, 树中节点由1~n编号, 每个节点有一权值, 有k可个关键点

问有多少种方案, 仅去掉一个关键点, 使得剩余关键点在树A中lca的权值大于树B的权值

思路 : lca具有可加性, 如 a b 的lca 为 anc, 则 a b c 的lca 为 anc 与 c的 lca, 对于每个关键点, 我们可以处理出该关键点之前和之后的所有关键点的lca, 如关键点key[i] 用 lcap[i] 表示关键点key[i]点(包含该点)之前所有关键点的 lca, lcan[i] 表示关键点key[i]点(包含该点)之后所有关键点的 lca, 那么对于去掉点key[i] (即不看该点, 不是从树中去点)后, 剩余关键点的lca 即为 lca(lcap[i-1], lcan[i+1]), 即key[i] 的前缀lca 和 后缀lca 的 lca, 求出lca后比较权值即可

代码如下: 

#include <bits/stdc++.h>

using namespace std;

stringstream ss;
#define endl "\n"
typedef long long ll;
typedef long double ld;
typedef pair<int, int> PII;
const int N = 1e5+10, M = N*2, mod = 1e9+7;
const int INF = 0x3f3f3f3f;

int n,k;
int ky[N]; // 存储关键点
int h1[N], e1[M], ne1[M], w1[N], idx1; // 树A
int h2[N], e2[M], ne2[M], w2[N], idx2; // 树B
int d1[N], d2[N], fa[N][20], fb[N][20]; // d1[i]表示A中节点i的深度, fa[i][j]表示节点i向上跳2^j步到的节点
int lcapa[N], lcapb[N], lcana[N], lcanb[N]; // 树Alcapa[i]表示关键点ky[i]之前所有节点的lca, lcana表示之后的所有节点

void add(int a, int b, int t)
{
    if(t == 1) e1[idx1] = b, ne1[idx1] = h1[a], h1[a] = idx1++;
    else e2[idx2] = b, ne2[idx2] = h2[a], h2[a] = idx2++;
}
// BFS预处理fa和d
void bfs1()
{
    memset(d1, 0x3f, sizeof d1);
    queue<int> q;
    q.push(1);
    d1[0] = 0, d1[1] = 1;
    while(q.size())
    {
        int t = q.front();
        q.pop();

        for(int i = h1[t]; ~i; i = ne1[i])
        {
            int j = e1[i];
            if(d1[j] > d1[t] + 1)
            {
                d1[j] = d1[t] + 1;
                q.push(j);
                fa[j][0] = t;
                for(int k = 1; k<=16; k++) fa[j][k] = fa[fa[j][k-1]][k-1];
            }
        }
    }
}

void bfs2()
{
    memset(d2, 0x3f, sizeof d2);
    queue<int> q;
    q.push(1);
    d2[0] = 0, d2[1] = 1;
    while(q.size())
    {
        int t = q.front();
        q.pop();

        for(int i = h2[t]; ~i; i = ne2[i])
        {
            int j = e2[i];
            if(d2[j] > d2[t] + 1)
            {
                d2[j] = d2[t] + 1;
                q.push(j);
                fb[j][0] = t;
                for(int k = 1; k<=16; k++) fb[j][k] = fb[fb[j][k-1]][k-1];
            }
        }
    }
}
// lca板子
int lca1(int a, int b)
{
    if(d1[a] < d1[b]) swap(a, b);
    for(int k = 16; k>=0; k--)
    {
        if(d1[fa[a][k]] >= d1[b]) a = fa[a][k];
    }
    if(a == b) return a;
    for(int k = 16; k>=0; k--)
    {
        if(fa[a][k] != fa[b][k]) a = fa[a][k], b = fa[b][k];
    }
    return fa[a][0];
}

int lca2(int a, int b)
{
    if(d2[a] < d2[b]) swap(a, b);
    for(int k = 16; k>=0; k--)
    {
        if(d2[fb[a][k]] >= d2[b]) a = fb[a][k];
    }
    if(a == b) return a;
    for(int k = 16; k>=0; k--)
    {
        if(fb[a][k] != fb[b][k]) a = fb[a][k], b = fb[b][k];
    }
    return fb[a][0];
}

void solve()
{
    cin>>n>>k;
    for(int i = 1; i<=k; i++) cin>>ky[i];
    memset(h1, -1, sizeof h1);
    memset(h2, -1, sizeof h2);
    for(int i = 1; i<=n; i++) cin>>w1[i];
    for(int i = 2; i<=n; i++)
    {
        int p; cin>>p;
        add(p, i, 1), add(i, p, 1);
    }

    for(int i = 1; i<=n; i++) cin>>w2[i];
    for(int i = 2; i<=n; i++)
    {
        int p; cin>>p;
        add(p, i, 2), add(i, p, 2);
    }
    bfs1(), bfs2();
    lcapa[1] = lcapb[1] = ky[1], lcana[k] = lcanb[k] = ky[k]; // 从ky[1]到ky[k]预处理出所有关键点的前缀lca和后缀lca
    for(int i = 2; i<=k; i++)
    {
        lcapa[i] = lca1(lcapa[i-1], ky[i]); // 预处理前缀lca lcapa[i]表示i包括i之前所有关键点的lca
        lcapb[i] = lca2(lcapb[i-1], ky[i]); // b同理
    }
    for(int i = k-1; i>=1; i--)
    {
        lcana[i] = lca1(lcana[i+1], ky[i]); // 预处理后缀lca lcana[i]表示i包括i之后所有关键点的lca
        lcanb[i] = lca2(lcanb[i+1], ky[i]); // b同理
    } 
    int res = 0; // 记录答案
    if(w1[lcana[2]] > w2[lcanb[2]]) res++; // 特判左右端点, 若去点ky[1], 则只用看后缀 求出lca比较权值 
    if(w1[lcapa[k-1]] > w2[lcapb[k-1]]) res++; // 若去点ky[k], 则只用看前缀
    for(int i = 2; i<k; i++)
    {
        // 中间点, 求一边前缀lca和后缀lca 的 lca即为去掉点ky[i]的lca, 比较权值即可
        int a = lca1(lcapa[i-1], lcana[i+1]), b = lca2(lcapb[i-1], lcanb[i+1]);
        if(w1[a] > w2[b]) res++;
    }
    cout<<res<<endl;
}

int main()
{
    ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
    solve();
//    int T;
//    cin>>T;
//    while(T -- )
//    {
//        solve();
//    }
    return 0;
}