LiberOJ #6210. 「美团 CodeM 决赛」tree 树形DP

题目链接：点这里

题解：

　　需要证明，所求的路径一定是全部权值都为1或者，路径上权值至多有一个为2其余为1且权值2在路径中央。

　　然后树形DP

　　设定dp[i][0/1] 以1为根的情况下，以i 节点下子树走分别全1和 走一次2和剩余全走1 的最长链

　　每遍历一次子树，统计一次答案

　　下面给出代码

#include<bits/stdc++.h>
using namespace std;
#pragma comment(linker, "/STACK:102400000,102400000")
#define ls i<<1
#define rs ls | 1
#define mid ((ll+rr)>>1)
#define pii pair<int,int>
#define MP make_pair
typedef long long LL;
const long long INF = 1e18+1LL;
const double pi = acos(-1.0);
const int N = 5e5+, M = 1e3+,inf = 2e9;
const LL mod = 1e9+7LL;

int n,a[N],ans;
int ans1,ans2;
vector<int > G[N];
int dp[N][];
void dfs(int u,int f) {
    if(a[u] == ) dp[u][] = ;
    else if(a[u] == ) dp[u][] = ;
    for(int i = ; i < G[u].size(); ++i) {
        int to = G[u][i];
        if(to == f) continue;
        dfs(to,u);
        ans1 = max(ans1,dp[to][] + dp[u][]);
        ans2 = max(ans2,dp[u][] + dp[to][]);
        ans2 = max(ans2,dp[u][] + dp[to][]);
        if(a[u] > )
            continue;

        if(a[u] == ) {
            dp[u][] = max(dp[u][],dp[to][] + );
            dp[u][] = max(dp[u][],dp[to][] + );
        }
        else {
            dp[u][] = max(dp[u][],dp[to][] + );
        }
    }
}
int main(){
    scanf("%d",&n);
    for(int i = ; i < n; ++i) {
        int x,y;
        scanf("%d%d",&x,&y);
        G[x].push_back(y);
        G[y].push_back(x);
    }
    int mi = inf;
    for(int i = ; i <= n; ++i) {
        scanf("%d",&a[i]);
        mi = min(mi,a[i]);
    }
    ans = ;
    dfs(,);
    if(mi >= ) {
        printf("%d/1\n",mi);
    }
    else {
        if(*ans1 > ans2) printf("1/%d\n",ans1);
        else if(ans2 %  == ) printf("1/%d\n",ans2/);
        else printf("2/%d\n",ans2);
    }
    return ;
}