17AHU排位赛2 E题（树上最大匹配，树形DP)

problem

有一个n个节点n-1条边组成的树。
每个点看成一个人，连接u和v的边看成是“中意关系”，即u和v两个人都想和对方组队。每个人希望组队的对象有可能有多个。
一支队伍由且仅由两个人组成，并且如果u和v组队了，那么u、v将不能和其他人再组成一支队。
现在问你，这n个人最多能组成多少支队伍。（允许某些人组不了队）

Input

第一行输入一个整数n,m(1<=n<=200000)
接下来n-1行，每行两个整数u,v，表示u和v两个人都想和对方组队。
数据保证是一个合法的树。

Output

输出一个整数，表示最多能组成多少支队伍。

Input

5
1 2
1 3
2 4
4 5

Output

2

Limitation

1s 256MB

Hint

一种可行的组队方案：
1与3组队
4与5组队
最多组成2支队

思路

基本的树形DP
dp[rt][0]表示rt这个点不与任何一个儿子连边，以k为根的子树的最大匹配
dp[rt][1]表示rt这个点与某一个儿子连边，以k为根的子树的最大匹配

---

核心代码

void dfs(int rt,int f)
{
    int mi=n;
    if(G[rt].size()==1&&G[rt][0]==f){
        return ;
    }
    for(int i=0;isize();++i){
        int tt=G[rt][i];
        if(tt==f) continue;
        dfs(tt,rt);
        dp[rt][0]+=dp[tt][1];
        dp[rt][1]+=dp[tt][1];
        if(dp[tt][1]-dp[tt][0]1]-dp[tt][0];
    }
    dp[rt][1]=dp[rt][1]-mi+1;
}

代码示例

#include
using namespace std;
const int maxn=200010;

int n;

vector<int> G[maxn];

int dp[maxn][2];

void dfs(int rt,int f)
{
    int mi=n;
    if(G[rt].size()==1&&G[rt][0]==f){
        return ;
    }
    for(int i=0;iint tt=G[rt][i];
        if(tt==f) continue;
        dfs(tt,rt);
        dp[rt][0]+=dp[tt][1];
        dp[rt][1]+=dp[tt][1];
        if(dp[tt][1]-dp[tt][0]1]-dp[tt][0];
    }
    dp[rt][1]=dp[rt][1]-mi+1;
}

int main()
{
    ios::sync_with_stdio(false);
    cin>>n;
    int u,v;
    for(int i=1;icin>>u>>v;
        G[u].push_back(v);
        G[v].push_back(u);
    }
    dfs(1,0);
    cout<1][0],dp[1][1])<return 0;
}