結果

問題 No.1752 Up-Down Tree
ユーザー 👑 Nachia
提出日時 2021-11-19 22:07:55
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
WA  
(最新)
CE  
(最初)
実行時間 -
コード長 1,273 bytes
コンパイル時間 3,044 ms
コンパイル使用メモリ 119,984 KB
最終ジャッジ日時 2025-01-25 20:24:06
ジャッジサーバーID
(参考情報)
judge4 / judge7
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1 WA * 1
other AC * 11 WA * 9
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <vector>
#include <algorithm>
#include <set>
#include <atcoder/modint>
using namespace std;
using i32 = int32_t;
using u32 = uint32_t;
using i64 = int64_t;
using u64 = uint64_t;
#define rep(i,n) for(int i=0; i<(n); i++)

using m32 = atcoder::static_modint<1'000'000'007>;

int main(){
  int N; cin >> N;
  vector<vector<int>> E(N);
  rep(i,N-1){
    int u,v; cin >> u >> v; u--; v--;
    E[u].push_back(v);
    E[v].push_back(u);
  }

  vector<int> I = {0};
  vector<int> P(N,-1);
  vector<int> D(N,0);
  rep(i,I.size()){
    int p = I[i];
    for(int e : E[p]) if(P[p] != e){
      P[e] = p;
      D[e] = D[p] + 1;
      I.push_back(e);
    }
  }

  vector<int> Z(N,1);
  for(int i=N-1; i>=1; i--) Z[P[I[i]]] += Z[I[i]];

  vector<int> dpW(N,0);
  for(int i=N-1; i>=0; i--){
    int p = I[i];
    int sumCollected = 0;
    for(int e : E[p]) if(P[p] != e) sumCollected += dpW[e];
    if(sumCollected * 2 < Z[p] - 1) sumCollected++;
    dpW[p] = sumCollected;
  }

  // rep(i,N) cout << maxBlack[i] << " "; cout << endl;

  int ans = min(N, dpW[0] * 2 + 1);
  cout << ans << endl;
  return 0;
}




struct ios_do_not_sync {
    ios_do_not_sync() {
        ios::sync_with_stdio(false);
        cin.tie(nullptr);
    }
} ios_do_not_sync_instance;
0