bzoj1907: 树的路径覆盖（树形DP）

Posted 2020-10-17 Sakits

　　一眼题...

　　f[i][0]表示在i连接一个子树的最小值，f[i][1]表示在i连接两个子树的最小值，随便转移...

　　样例挺强的1A了美滋滋...

#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=500010, inf=1e9;
struct poi{int too, pre;}e[maxn];
int n, T, tot, x, y;
int f[maxn][2], last[maxn];
void read(int &k)
{
    int f=1; k=0; char c=getchar();
    while(c<‘0‘ || c>‘9‘) c==‘-‘ && (f=-1), c=getchar();
    while(c<=‘9‘ && c>=‘0‘) k=k*10+c-‘0‘, c=getchar();
    k*=f;
}
inline void add(int x, int y){e[++tot]=(poi){y, last[x]}; last[x]=tot;}
inline int min(int a, int b){return a<b?a:b;}
void dfs(int x, int fa) 
{
    int mn1=inf, mn2=inf, mni1=0, mni2=0, tmp=1; f[x][0]=0; f[x][1]=-1; 
    for(int i=last[x], too;i;i=e[i].pre)
    if((too=e[i].too)!=fa) 
    {
        dfs(too, x);
        f[x][0]+=min(f[too][1], f[too][0]); 
        if(f[too][0]==min(f[too][1], f[too][0])) tmp=0;
        if(f[too][0]-min(f[too][1], f[too][0])<mn1) mn1=f[too][0]-min(f[too][1], f[too][0]), mni1=too;
        else if(f[too][0]-min(f[too][1], f[too][0])<mn2) mn2=f[too][0]-min(f[too][1], f[too][0]), mni2=too;
    }
    f[x][0]+=tmp; 
    if(!(f[x][0]-tmp)) {f[x][0]=1; f[x][1]=inf; return;}
    if(mn2==inf) {f[x][1]=inf; return;}
    for(int i=last[x], too;i;i=e[i].pre)
    if((too=e[i].too)!=fa) 
    {
        if(too==mni1 || too==mni2) f[x][1]+=f[too][0]; 
            else f[x][1]+=min(f[too][1], f[too][0]);
    }
}
int main()
{
    read(T);
    while(T--)
    {
        read(n); memset(last, 0, (n+1)<<2); tot=0; 
        for(int i=1;i<n;i++) read(x), read(y), add(x, y), add(y, x);
        dfs(1, 0); printf("%d\n", min(f[1][1], f[1][0]));
    }
}

View Code