Submission #1349203

#TimeUsernameProblemLanguageResultExecution timeMemory
1349203iamhereforfunMergers (JOI19_mergers)C++20
100 / 100
267 ms85780 KiB
// Starcraft 2 enjoyer //

#include <bits/stdc++.h>

// #pragma GCC target("avx2")
// #pragma GCC optimize("O3")
// #pragma GCC optimize("unroll-loops")

using namespace std;

#define LSOne(X) ((X) & -(X))

const int N = 5e5 + 5;
const int M = 1 << 5;
const int K = 19;
const int LG = 20;
const long long INF = 3e18;
const int C = 26;
const int B = 1000;
const int MOD = 1e9 + 7;

int n, k, sz[N], p[N], par[N], st[N], en[N], cid, num, tot[N], deg[N];
vector<int> adj[N], state[N];
bool gud[N];

void dfs1(int c, int p)
{
    sz[c] = 1;
    par[c] = p;
    st[c] = cid++;
    for (int i : adj[c])
    {
        if (i == p)
            continue;
        dfs1(i, c);
        sz[c] += sz[i];
    }
    en[c] = cid - 1;
}

bool in(int a, int b)
{
    return st[a] <= st[b] && en[b] <= en[a];
}

int parent(int i)
{
    return p[i] == i ? i : p[i] = parent(p[i]);
}

inline void solve()
{
    cin >> n >> k;
    for (int x = 0; x < n - 1; x++)
    {
        int a, b;
        cin >> a >> b;
        adj[a].push_back(b);
        adj[b].push_back(a);
    }
    for (int x = 1; x <= n; x++)
    {
        int i;
        cin >> i;
        state[i].push_back(x);
        p[x] = x;
    }
    cid = 1;
    dfs1(1, 0);
    for (int x = 1; x <= k; x++)
    {
        for (int y = 0; y < state[x].size() - 1; y++)
        {
            int i = state[x][y], j = state[x][y + 1];
            while (!in(i, j))
            {
                int k = parent(par[i]);
                p[i] = k;
                i = k;
            }
            while (!in(j, i))
            {
                int k = parent(par[j]);
                p[j] = k;
                j = k;
            }
        }
    }
    num = 0;
    for (int x = 1; x <= n; x++)
    {
        for (int i : adj[x])
        {
            i = parent(i);
            int j = parent(x);
            if (i != j)
                deg[i]++;
        }
    }
    for (int x = 1; x <= n; x++)
    {
        if (deg[x] == 1 && x == p[x])
        {
            num++;
        }
    }
    cout << (num + 1) / 2 << "\n";
}

signed main()
{
    // freopen("CP.INP", "r", stdin);
    // freopen("CP.OUT", "w", stdout);
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    int t = 1;
    // cin >> t;
    for (int x = 1; x <= t; x++)
    {
        solve();
    }
    return 0;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...