bzoj3257 树的难题

http://www.elijahqi.win/2018/01/14/bzoj3257-%e6%a0%91%e7%9a%84%e9%9a%be%e9%a2%98/
Description

给出一个无根树。树有N个点，边有权值。每个点都有颜色，是黑色、白色、
灰色这三种颜色之一，称为一棵三色树。
可爱的 Alice觉得，一个三色树为均衡的，当且仅当，树中不含有黑色结点
或者含有至多一个白色节点。然而，给出的三色树可能并不满足这个性质。
所以，Alice打算删去若干条边使得形成的森林中每棵树都是均衡的，花费
的代价等于删去的边的权值之和。请你计算需要花费的代价最小是多少。
注意，输入文件包含多组测试数据。
Input

第一行包含一个正整数 T，表示有 T组测试数据。接下来依次是 T组测试数
据。每组测试数据的第一行包含一个正整数 N。
第二行包含 N个 0、1、2之一的整数，依次表示点 1到点 N的颜色。其中，
0 表示黑色，1表示白色，2表示灰色。
接下来 N-1行，每行为三个整数ui、vi、c i，表示一条权值等于ci的边(ui, vi)。
Output

输出 T行，每行一个整数，依次表示每组测试数据的答案。
Sample Input

1
5
0 1 1 1 0
1 2 5
1 3 3
5 2 5
2 4 16

Sample Output
10
HINT

花费 10的代价删去边(1, 2)和边(2, 5)。

对于 100%的数据：1 ≤ N ≤ 300 000 ，1 ≤ T ≤ 5 ，0 ≤ ci<=10^9

觉得real非常非常妙的题啊 %了icefox以后才会做了 orz巨佬好强

题意：每个点有三种颜色 黑白灰 现在要求把这棵树变成一些满足以下要求的树 只有白点或者只有一个白点 黑点无所谓 求问将给定的图变成如上要求的最小代价 代价是每次可以断边权

那么设dp[x][i][j]表示当前节点在x号且x为根 有i个黑色j个白色的其他子树或者其他树都满足条件的最小代价是多少 那么不妨先以1当根 然后树形dp下 既然我的定义是以x为根 为什么树形dp也okay 因为最后我要求的是整张图满足条件即可 并没有说谁要做根 那么x节点上下几条边都可以断那不就okay 做转移的时候我枚举子树再枚举我当前节点的状态 （黑点数&白点数） 对于黑点>1的情况我直接累加在1上 对于白点>2的情况 我直接累加在2上 然后再枚举子树的状态（黑点数&白点数） 然后判断 转移到我现在来 要么我现在的黑点数&白点数是我+子树的 或者断开和子树的边 黑点数&白点数就是我自己的 比较下即可

#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 330000  
#define inf 1LL<<60
using namespace std;
inline char gc(){
    static char now[1<<16],*S,*T;
    if (T==S){T=(S=now)+fread(now,1,1<<16,stdin);if (T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0;char ch=gc();
    while(ch<'0'||ch>'9') ch=gc();
    while(ch<='9'&&ch>='0') x=x*10+ch-'0',ch=gc();
    return x;
}
struct node{
    int y,z,next;
}data[N<<1];
int fa[N],n,a[N],h[N],num,T;long long dp[N][2][3],tmp[2][3];
inline void insert1(int x,int y,int z){
    data[++num].y=y;data[num].z=z;data[num].next=h[x];h[x]=num;
    data[++num].y=x;data[num].z=z;data[num].next=h[y];h[y]=num;
}
inline void dfs(int x){
    for (int i=0;i<=1;++i) for (int j=0;j<=2;++j) dp[x][i][j]=inf;dp[x][a[x]==0][a[x]==1]=0;
    for (int i=h[x];i;i=data[i].next){
        int y=data[i].y,z=data[i].z;
        if (fa[x]==y) continue;fa[y]=x;dfs(y);
        for (int j=0;j<=1;++j) for (int z=0;z<=2;++z) tmp[j][z]=inf;
        for (int i1=0;i1<=1;++i1) for (int j1=0;j1<=2;++j1)
        if (dp[x][i1][j1]<inf) for (int ii=0;ii<=1;++ii)
        for (int jj=0;jj<=2;++jj) if (dp[y][ii][jj]<inf){
            int sum_white=j1+jj>2?2:jj+j1;
            tmp[i1|ii][sum_white]=min(tmp[i1|ii][sum_white],dp[x][i1][j1]+dp[y][ii][jj]);
            if (!ii||jj<2) tmp[i1][j1]=min(tmp[i1][j1],dp[x][i1][j1]+dp[y][ii][jj]+z);
        }memcpy(dp[x],tmp,sizeof(tmp));
    }
}
int main(){
    freopen("bzoj3257.in","r",stdin);
    T=read();
    while(T--){
        n=read();memset(h,0,sizeof(h));num=0;
        for (int i=1;i<=n;++i) a[i]=read();
        for (int i=1;i<n;++i){int x=read(),y=read(),z=read();insert1(x,y,z);}
        dfs(1);long long ans=inf;
        for (int i=0;i<=1;++i) for (int j=0;j<=2;++j) if (!i||j<2) ans=min(ans,dp[1][i][j]);
        printf("%lld\n",ans);
    }
    return 0;
}