結果

問題 No.2205 Lights Out on Christmas Tree
ユーザー lddlinanlddlinan
提出日時 2023-03-13 21:33:49
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 187 ms / 2,000 ms
コード長 1,869 bytes
コンパイル時間 1,063 ms
コンパイル使用メモリ 94,120 KB
実行使用メモリ 52,416 KB
最終ジャッジ日時 2024-09-18 07:46:03
合計ジャッジ時間 7,454 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 6 ms
12,544 KB
testcase_01 AC 6 ms
12,544 KB
testcase_02 AC 6 ms
12,544 KB
testcase_03 AC 6 ms
12,672 KB
testcase_04 AC 6 ms
12,544 KB
testcase_05 AC 6 ms
12,672 KB
testcase_06 AC 6 ms
12,544 KB
testcase_07 AC 6 ms
12,544 KB
testcase_08 AC 7 ms
12,672 KB
testcase_09 AC 6 ms
12,544 KB
testcase_10 AC 5 ms
12,672 KB
testcase_11 AC 187 ms
33,664 KB
testcase_12 AC 183 ms
33,600 KB
testcase_13 AC 177 ms
33,732 KB
testcase_14 AC 157 ms
33,720 KB
testcase_15 AC 119 ms
52,416 KB
testcase_16 AC 74 ms
26,496 KB
testcase_17 AC 71 ms
26,948 KB
testcase_18 AC 5 ms
14,144 KB
testcase_19 AC 113 ms
33,604 KB
testcase_20 AC 110 ms
33,860 KB
testcase_21 AC 110 ms
33,600 KB
testcase_22 AC 123 ms
33,892 KB
testcase_23 AC 111 ms
33,600 KB
testcase_24 AC 106 ms
33,600 KB
testcase_25 AC 118 ms
33,856 KB
testcase_26 AC 134 ms
33,604 KB
testcase_27 AC 114 ms
33,844 KB
testcase_28 AC 120 ms
33,600 KB
testcase_29 AC 117 ms
33,856 KB
testcase_30 AC 133 ms
33,852 KB
testcase_31 AC 116 ms
33,604 KB
testcase_32 AC 111 ms
33,724 KB
testcase_33 AC 108 ms
33,732 KB
testcase_34 AC 106 ms
33,588 KB
testcase_35 AC 131 ms
33,848 KB
testcase_36 AC 132 ms
33,780 KB
testcase_37 AC 109 ms
33,732 KB
testcase_38 AC 138 ms
33,580 KB
testcase_39 AC 109 ms
33,600 KB
権限があれば一括ダウンロードができます

ソースコード

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