結果

問題 No.768 Tapris and Noel play the game on Treeone
コンテスト
ユーザー lumc_
提出日時 2018-06-15 23:56:41
言語 C++11（廃止可能性あり）
(gcc 13.3.0 + boost 1.89.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 1,563 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,561 ms
コンパイル使用メモリ 164,556 KB
実行使用メモリ 31,716 KB
最終ジャッジ日時 2024-09-25 06:14:59
合計ジャッジ時間 5,180 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 21 WA * 1
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include<bits/stdc++.h>
using namespace std;

const int N = 1e6; 
vector<int> g[N];
int dp0[N];
int dp1[N];
int n;
bool t[N]; // 毎度確保するほどでもない一時変数
bool ans[N];

bool dfs0(int v, int p) {
  bool flag = 0;
  for(int u : g[v]) if(u != p) if(dfs0(u, v)) flag = 1;
  // dp0は頂点rを根とした率直な答え
  return dp0[v] = !flag;
}

void dfs1(int v, int p) {
  int cnt = 0;
  ans[v] = 1;
  for(int u : g[v]) if(u != p) {
    t[u] = dp0[u];
    if(t[u]) ans[v] = 0, cnt++;
  }
  if(p != -1) {
    t[v] = dp1[v];
    if(t[v]) ans[v] = 0, cnt++;
  }
  // dp1はuをキャンセルした, 頂点rを根とする, vでの答え
  // cnt >= 2 ならどの頂点uをキャンセルしても他に t[x] == 1 となる x (!= u) あり
  // cnt == 1 なら t[u] == 1 なる 頂点u をキャンセル時のみほかに 1 なし
  for(int u : g[v]) if(u != p) { dp1[u] = !(cnt >= 2 || (cnt == 1 && t[u] == 0)); dfs1(u, v); }
}

int main() {
  cin >> n;
  if(n == 1) return (cout << 1 << endl << 1 << endl, 0);
  if(n == 2) return (cout << 2 << endl << 1 << endl << 2 << endl, 0);
  for(int i = 0; i < n - 1; i++) {
    int a, b; cin >> a >> b; a--; b--;
    g[a].emplace_back(b);
    g[b].emplace_back(a);
  }

  int r = 0;
  for(int i = 1; i < n; i++) if(g[i].size() >= 2) r = i; // 次数が2以上であるような頂点rを選ぶ
  vector<int> res; res.reserve(n);
  dfs0(r, -1);
  dfs1(r, -1);
  for(int i = 0; i < n; i++) if(ans[i]) res.emplace_back(i);

  cout << res.size() << endl;
  for(int el : res) cout << el + 1 << endl;
}
0