結果
問題 | No.2205 Lights Out on Christmas Tree |
ユーザー |
|
提出日時 | 2023-03-13 21:33:49 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 162 ms / 2,000 ms |
コード長 | 1,869 bytes |
コンパイル時間 | 947 ms |
コンパイル使用メモリ | 93,476 KB |
最終ジャッジ日時 | 2025-02-11 11:00:42 |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 37 |
コンパイルメッセージ
main.cpp: In function ‘int main()’: main.cpp:70:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result] 70 | scanf("%d", &n); | ~~~~~^~~~~~~~~~ main.cpp:72:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result] 72 | scanf("%d %d", &a, &b); | ~~~~~^~~~~~~~~~~~~~~~~ main.cpp:77:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result] 77 | scanf("%d", &ck[i]); | ~~~~~^~~~~~~~~~~~~~
ソースコード
#include<stdio.h>#include<string.h>#include<stdlib.h>#include <map>#include <vector>#include <queue>#include <deque>#include <set>#include <stack>#include <algorithm>#include <array>#include <unordered_set>#include <unordered_map>#include <string>using namespace std;bool rcmp(int a, int b) { return a>b; }typedef long long LL;set<int> nei[200004];int ck[200004];int dp[2][200004];int vv[200004];int dfs(int m, int r, int p) {if (dp[m][r]!=-2) return dp[m][r];int v=-1, ok=1, bb, k, xm, i;for (auto b: nei[r]) {if (b==p) continue;if (dfs(0, b, r)<0&&dfs(1, b, r)<0) { ok=0; break; }}if (ok) {xm=m^ck[r];bb=0;k=0;for (auto b: nei[r]) {if (b==p) continue;if (dp[0][b]<0) { xm^=1; bb+=dp[1][b]; bb++;}else if (dp[1][b]<0) bb+=dp[0][b];else {vv[k++]=dp[1][b]+1-dp[0][b];bb+=dp[0][b];}}sort(vv, vv+k);if (k==0) {if (xm) v=-1;else v=bb;} else {v=bb;if (xm) {v+=vv[0]; for (i=1; i+2<k; i+=2) {if ((vv[i]+vv[i+1])>=0) break;v+=(vv[i]+vv[i+1]);}} else {for (i=0; i+2<k; i+=2) {if ((vv[i]+vv[i+1])>=0) break;v+=(vv[i]+vv[i+1]);}}}}dp[m][r]=v;return v;}int main() {int n, i, a, b;scanf("%d", &n);for (i=1; i<n; i++) {scanf("%d %d", &a, &b);nei[a].insert(b);nei[b].insert(a);}for (i=1; i<=n; i++) {scanf("%d", &ck[i]);ck[i]^=1;dp[0][i]=dp[1][i]=-2;}printf("%d\n", dfs(0, 1, -1));return 0;}