結果
問題 | No.2205 Lights Out on Christmas Tree |
ユーザー |
![]() |
提出日時 | 2023-04-22 23:13:01 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 107 ms / 2,000 ms |
コード長 | 1,014 bytes |
コンパイル時間 | 411 ms |
コンパイル使用メモリ | 52,608 KB |
実行使用メモリ | 17,408 KB |
最終ジャッジ日時 | 2024-11-07 10:19:52 |
合計ジャッジ時間 | 5,792 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 37 |
ソースコード
/* -*- coding: utf-8 -*-** 2205.cc: No.2205 Lights Out on Christmas Tree - yukicoder*/#include<cstdio>#include<vector>#include<algorithm>using namespace std;/* constant */const int MAX_N = 200000;/* typedef */typedef vector<int> vi;/* global variables */vi nbrs[MAX_N];int cs[MAX_N], ps[MAX_N], ss[MAX_N], cis[MAX_N];/* subroutines *//* main */int main() {int n;scanf("%d", &n);for (int i = 1; i < n; i++) {int u, v;scanf("%d%d", &u, &v);u--, v--;nbrs[u].push_back(v);nbrs[v].push_back(u);}for (int i = 0; i < n; i++) scanf("%d", cs + i);ps[0] = -1;for (int u = 0; u >= 0;) {vi &nbru = nbrs[u];int up = ps[u];if (cis[u] < nbru.size()) {int v = nbru[cis[u]++];if (v != up) {ps[v] = u;u = v;}}else {if (up >= 0) {ss[up] += ss[u];if (! cs[u]) {ss[up]++;cs[u] ^= 1, cs[up] ^= 1;}}u = up;}}printf("%d\n", cs[0] ? ss[0] : -1);return 0;}