結果

問題 No.2205 Lights Out on Christmas Tree
ユーザー tofu_dra2tofu_dra2
提出日時 2023-02-04 00:49:36
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 96 ms / 2,000 ms
コード長 2,449 bytes
コンパイル時間 4,477 ms
コンパイル使用メモリ 232,992 KB
実行使用メモリ 31,256 KB
最終ジャッジ日時 2023-09-15 21:15:11
合計ジャッジ時間 12,482 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,376 KB
testcase_01 AC 2 ms
4,376 KB
testcase_02 AC 1 ms
4,380 KB
testcase_03 AC 1 ms
4,376 KB
testcase_04 AC 2 ms
4,376 KB
testcase_05 AC 1 ms
4,376 KB
testcase_06 AC 2 ms
4,376 KB
testcase_07 AC 2 ms
4,376 KB
testcase_08 AC 2 ms
4,380 KB
testcase_09 AC 1 ms
4,380 KB
testcase_10 AC 1 ms
4,380 KB
testcase_11 AC 73 ms
20,304 KB
testcase_12 AC 74 ms
20,280 KB
testcase_13 AC 72 ms
20,284 KB
testcase_14 AC 67 ms
20,232 KB
testcase_15 AC 88 ms
31,256 KB
testcase_16 AC 53 ms
14,744 KB
testcase_17 AC 53 ms
14,816 KB
testcase_18 AC 2 ms
4,380 KB
testcase_19 AC 79 ms
20,496 KB
testcase_20 AC 84 ms
19,876 KB
testcase_21 AC 82 ms
19,844 KB
testcase_22 AC 72 ms
20,896 KB
testcase_23 AC 79 ms
19,836 KB
testcase_24 AC 80 ms
20,072 KB
testcase_25 AC 88 ms
19,240 KB
testcase_26 AC 95 ms
18,816 KB
testcase_27 AC 80 ms
20,548 KB
testcase_28 AC 69 ms
20,556 KB
testcase_29 AC 73 ms
20,852 KB
testcase_30 AC 95 ms
18,708 KB
testcase_31 AC 77 ms
20,288 KB
testcase_32 AC 79 ms
19,488 KB
testcase_33 AC 80 ms
19,488 KB
testcase_34 AC 78 ms
19,964 KB
testcase_35 AC 96 ms
19,076 KB
testcase_36 AC 94 ms
18,972 KB
testcase_37 AC 79 ms
19,816 KB
testcase_38 AC 94 ms
18,780 KB
testcase_39 AC 78 ms
20,296 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>

typedef long long int ll;
typedef long double ld;

using namespace std;
using namespace atcoder;

#define inf 1010000000
#define llinf 1001000000000000000ll
#define pi 3.14159265358979
#define rep(i, n) for(ll i = 0; i < (n); i++)
#define rep1(i, n) for(ll i = 1; i <= (n); i++)
#define rep2(i,l,r) for(ll i = (l); i < (r); i++)
#define rng(a) a.begin(),a.end()
#define fi first
#define se second
#define pb push_back
#define eb emplace_back
#define pob pop_back
#define mp make_pair
#define st string
#define sz(x) (int)(x).size()
#define mems(x) memset(x, -1, sizeof(x));
#define pcnt __builtin_popcountll
#define _GLIBCXX_DEBUG

#define dame { puts("-1"); return 0;}
#define yes { puts("Yes"); return 0;}
#define no { puts("No"); return 0;}
#define ret(x) { cout<<(x)<<endl;}
#define yn {puts("Yes");}else{puts("No");}

template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return true; } return false; }
template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return true; } return false; }

// srand((unsigned)time(NULL)); rand()を用いる際にmainの頭に置く
// clock()/CLOCKS_PER_SEC 秒数を知りたいときに用いる
#define mod 998244353
//問題に応じて 1000000007 に変える
using mint = modint998244353; // .val()で出力

vector<ll> dx={1,0,-1,0};
vector<ll> dy={0,1,0,-1};

using pl = pair<ll,ll>;
using ppl = pair<pl,ll>;
using V = vector<ll>;
using Graph = vector<vector<ll>>; //llをEdgeにする等,適宜書き換える
// G.assign(n, vector<pl>()); グローバル変数にGを置く時に置く

// 関数を置くのはここ以下
ll n;
pl p[200200];
Graph G;
ll c[200200];

pl dp(int now, int par){
  ll ans = 0;
  ll check = c[now];
  for(auto x : G[now]){
    if(x==par) continue;
    // 返り値の一つ目は操作数、二つ目は白と黒のどちらを返すか(白ならば1)
    pl tmp = dp(x,now);
    ans += tmp.fi + tmp.se;
    check += tmp.se;
  }
  if(now==0){
    if(check%2==0){
      ret(-1);
    }
    else{
      ret(ans);
    }
  }
  return mp(ans,(check+1)%2);
}

int main() {
  cin.tie(nullptr);
  ios::sync_with_stdio(false);
  cout << fixed << setprecision(20);
  // ここから編集
  cin >> n;
  G.assign(n, vector<ll>());

  rep(i,n-1){
    cin >> p[i].fi >> p[i].se;
    p[i].fi--;
    p[i].se--;
    G[p[i].fi].eb(p[i].se);
    G[p[i].se].eb(p[i].fi);
  }
  rep(i,n) cin >> c[i];

  dp(0,-1);
}
0