제출 #1327952

#제출 시각아이디문제언어결과실행 시간메모리
1327952aaaaaaaaMergers (JOI19_mergers)C++20
10 / 100
47 ms83056 KiB
#include <bits/stdc++.h>
using namespace std;
#define int long long
const int mxN = 5005;
int dp[mxN][mxN], s[mxN], f[mxN], bad[mxN], exta = 0, cost = 0, ans = 1e18, st, n, k;
vector<int> adj[mxN];
void dfs(int u = 1, int par = 0){
    dp[u][s[u]] = 1, bad[u] = 0;
    int cnt = 0;
    for(auto it : adj[u]){
        if(it ^ par){
            dfs(it, u);
            bad[u] |= bad[it];
            for(int j = 1; j <= k; ++j){
                dp[u][j] += dp[it][j];
            }
            if(bad[it]) cnt += 1;
        }
    }
    bool ok = 1;
    for(int j = 1; j <= k; ++j){
        if(dp[u][j] != 0 && dp[u][j] != f[j]){
            ok = 0;
        }
    }
    if(ok){
        if(!bad[u]){
            if(par != -1) cost += 1;
            bad[u] = 1;
        }else{
            if(cnt > 1 && par != -1) exta = 1;
        }
    }
}
signed main(){
    ios::sync_with_stdio(0);
    cin.tie(nullptr); cout.tie(nullptr);
    cin >> n >> k;
    for(int i = 1, u, v; i <= n - 1; ++i){
        cin >> u >> v;
        adj[u].push_back(v);
        adj[v].push_back(u);
    }
    for(int i = 1; i <= n; ++i){
        cin >> s[i];
        f[s[i]] += 1;
    }
    int root = 1;
    for(int i = 1; i <= n; ++i){
        if((int) adj[i].size() > 1) root = i;
    }
    dfs(root, -1);
    cout << (cost + exta + 1) / 2 << "\n";
    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...