Submission #1327647

#TimeUsernameProblemLanguageResultExecution timeMemory
1327647aaaaaaaaMergers (JOI19_mergers)C++20
0 / 100
14 ms8248 KiB
#include <bits/stdc++.h>
using namespace std;
#define int long long
const int mxN = 1005;
int dp[mxN][mxN], s[mxN], f[mxN], cost = 0, ans = 0, st, n, k;
vector<int> adj[mxN];
void dfs(int u = 1, int par = 0){
    dp[u][s[u]] += 1;
    for(auto it : adj[u]){
        if(it ^ par){
            dfs(it, u);
            for(int j = 1; j <= k; ++j){
                dp[u][j] += dp[it][j];
            }
        }
    }
    bool ok = 1;
    for(int j = 1; j <= k; ++j){
        if(dp[u][j] && dp[u][j] != f[j]){
            ok = 0;
        }
    }
    if(ok && u != st) {
        //cout << "invalid node: " << u << "\n";
        dp[u][s[u]] = -1e9, cost += 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;
    }
    for(int i = 1; i <= n; ++i){
        st = i, cost = 0;
        memset(dp, 0, sizeof(dp));
        dfs(i, -1);
        ans = max(ans, cost - 1);
    }
    cout << ans << "\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...