結果

問題 No.2205 Lights Out on Christmas Tree
ユーザー lddlinan
提出日時 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]);
      |         ~~~~~^~~~~~~~~~~~~~

ソースコード

diff #
プレゼンテーションモードにする

#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;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0