結果

問題 No.768 Tapris and Noel play the game on Treeone
ユーザー pekempey
提出日時 2018-12-16 02:55:29
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,044 bytes
コンパイル時間 877 ms
コンパイル使用メモリ 85,508 KB
実行使用メモリ 10,056 KB
最終ジャッジ日時 2024-09-25 06:22:23
合計ジャッジ時間 4,666 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 5 WA * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <vector>
#include <map>

using namespace std;

vector<int> g[100000];
int dp[100000];
int ans[100000];

int mex(const map<int, int> &mp) {
    int res = 0;
    while (mp.count(res)) res++;
    return res;
}

void dfs(int u, int p) {
    map<int, int> mp;
    for (int v : g[u]) if (v != p) {
        dfs(v, u);
        mp[dp[v]]++;
    }
    dp[u] = mex(mp);
}

void dfs2(int u, int p) {
    map<int, int> mp;
    for (int v : g[u]) {
        mp[dp[v]]++;
    }
    ans[u] = mex(mp);
    for (int v : g[u]) if (v != p) {
        if (mp[dp[v]] == 1 && dp[v] < ans[u]) {
            dp[u] = dp[v];
        }
        dfs2(v, u);
    }
}

int main() {
	int n;
	cin >> n;
	for (int i = 0; i < n - 1; i++) {
        int u, v;
        cin >> u >> v;
		u--;
		v--;
		g[u].push_back(v);
		g[v].push_back(u);
	}
    dfs(0, -1);
    dfs2(0, -1);
    cout << count(ans, ans + n, 0) << endl;
    for (int i = 0; i < n; i++) {
        if (ans[i] == 0) {
            cout << i + 1 << endl;
        }
    }
}
0