結果

問題 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;
}
0