HDU5834 문제 보고서

트리 dp 세트 문제지만 옮기기가 좀 번거롭습니다.한 가지 분명한 생각은 먼저 dfs가 서브 트리를 한 번에 완성한 다음에 dfs에 부모 노드가 현재 노드에 미치는 영향을 더하는 것이다.약간 매운 눈으로 써서 오랫동안 조정하였다.

dp[0][u], dp[1][u]는 각각 u를 근본으로 하고 돌아오는 것과 돌아오지 않는 최대치를 나타낸다. 첫 번째 dfs는 매우 간단하고 일반적인 방법이다. 두 번째 때 우리는 아버지 노드가 방문할 아들 노드에 대한 영향을 배제하고 다시 dfs를 한 다음에 거슬러 올라갈 때 복원하면 된다. dp[1][u]를 주의할 때 다음에 크고 이동할 때를 기록해야 한다.방문할 아들 노드 v가 u에서 돌아오지 않는 길에 있으면 차대치로 영향을 줄인다.

//
//  Created by Running Photon
//  Copyright (c) 2015 Running Photon. All rights reserved.
//
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#define ALL(x) x.begin(), x.end()
#define INS(x) inserter(x, x,begin())
#define ll long long
#define CLR(x) memset(x, 0, sizeof x)
using namespace std;
const int inf = 0x3f3f3f3f;
const int MOD = 1e9 + 7;
const int maxn = 2e5 + 10;
const int maxv = 1e5 + 10;
const double eps = 1e-9;

int pnt[maxn], nxt[maxn], val[maxv], head[maxv], cost[maxn], cnt;
void add_edge(int u, int v, int value) {
    pnt[cnt] = v;
    cost[cnt] = value;
    nxt[cnt] = head[u];
    head[u] = cnt++;
}
int dp[2][maxv];
void dfs1(int u, int fa) {
    dp[0][u] = val[u];
    dp[1][u] = val[u];
    for(int i = head[u]; ~i; i = nxt[i]) {
        int v = pnt[i];
        if(v == fa) continue;
        dfs1(v, u);
        if(dp[0][v] - cost[i] * 2 > 0) {
            dp[0][u] += dp[0][v] - cost[i] * 2;
        }
    }
    for(int i = head[u]; ~i; i = nxt[i]) {
        int v = pnt[i];
        if(v == fa) continue;
        if(dp[0][u] - max(0, dp[0][v] - cost[i] * 2) + (dp[1][v] - cost[i]) >= dp[1][u]) {
            dp[1][u] = dp[0][u] - max(0, dp[0][v] - cost[i] * 2) + (dp[1][v] - cost[i]);
        }
    }
}
void dfs2(int u, int fa, int Cost) {
    int dir = u, tmp;
    if(fa != -1 && dp[0][fa] - Cost * 2 > 0) {
        dp[0][u] += dp[0][fa] - Cost * 2;
    }
    tmp = dp[1][u] = dp[0][u];
    for(int i = head[u]; ~i; i = nxt[i]) {
        int v = pnt[i];

        if(dp[0][u] - max(0, dp[0][v] - cost[i] * 2) + (dp[1][v] - cost[i]) >= dp[1][u]) {
            dir = v;
            swap(tmp, dp[1][u]);
            dp[1][u] = dp[0][u] - max(0, dp[0][v] - cost[i] * 2) + (dp[1][v] - cost[i]);
        }
        else if(dp[0][u] - max(0, dp[0][v] - cost[i] * 2) + (dp[1][v] - cost[i]) >= tmp) {
            tmp = dp[0][u] - max(0, dp[0][v] - cost[i] * 2) + (dp[1][v] - cost[i]);
        }
    }
    // printf("dp[0][%d] = %d dp[1][%d] = %d  tmp = %d
", u, dp[0][u], u, dp[1][u], tmp);
    int L = dp[0][u], R = dp[1][u];
    for(int i = head[u]; ~i; i = nxt[i]) {
        int v = pnt[i];
        if(v == fa) continue;
        if(dp[0][v] - cost[i] * 2 > 0) {
            dp[0][u] = L - (dp[0][v] - cost[i] * 2);
        }
        if(dir == v) {
            dp[1][u] = tmp;
            if(dp[0][v] - cost[i] * 2 > 0) dp[1][u] -= (dp[0][v] - cost[i] * 2);
        } else if(dp[0][v] - cost[i] * 2 > 0) {
            dp[1][u] = R - (dp[0][v] - cost[i] * 2);
        }
        // printf("dir = %d
", dir);
        // printf("nxt%d = %d
", v, dp[1][u]);
        dfs2(v, u, cost[i]);
        dp[0][u] = L, dp[1][u] = R;
    }
}
int main() {
#ifdef LOCAL
    freopen("C:\\Users\\Administrator\\Desktop\\in.txt", "r", stdin);
    freopen("C:\\Users\\Administrator\\Desktop\\out.txt","w",stdout);
#endif
//  ios_base::sync_with_stdio(0);
    int T;
    int cas = 0;
    scanf("%d", &T);
    while(T--) {
        int n;
        printf("Case #%d:
", ++cas);
        scanf("%d", &n);
        cnt = 0;
        memset(head, -1, sizeof (int) * (n + 1));
        for(int i = 1; i <= n; i++) scanf("%d", val + i);
        for(int i = 1; i < n; i++) {
            int u, v, value;
            scanf("%d%d%d", &u, &v, &value);
            add_edge(u, v, value);
            add_edge(v, u, value);
        }
        dfs1(1, -1);
        dfs2(1, -1, 0);
        for(int i = 1; i <= n; i++) {
            printf("%d
", max(dp[0][i], dp[1][i]));
        }
    }

    return 0;
}