【JZOJ 4887】最大匹配

Description

mhy12345学习了二分图匹配，二分图是一种特殊的图，其中的点可以分到两个集合中，使得相同的集合中的点两两没有连边。
图的“匹配”是指这个图的一个边集，里面的边两两不存在公共端点。
匹配的大小是指该匹配有多少条边。
二分图匹配我们可以通过匈牙利算法得以在O(VE)时间复杂度内解决。
mhy12345觉得单纯的二分图匹配算法毫无难度，因此提出新的问题：
现在给你一个N个点N-1条边的连通图，希望你能够求出这个图的最大匹配以及最大匹配的数量。
两个匹配不同当且仅当存在一条边在第一个匹配中存在而在第二个匹配中不存在。

Analysis

抽象题意：一棵树，一条边所连的两个点不能同时选，问最多选几个点以及在点最多的情况下的选择方案数。
这是经典树形dp模型，设f[i][0]表示i选，f[i][1]表示i不选，且在以i为根的子树内选取的最多点数。
转移显然，但是主要是要求方案数。那我们再设g[i][0/1]，状态含义与f相同，表示方案数。
打起来不难，但是要细心。
我比赛的时候没时间打拍一次过掉了，这是运气

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define efo(i,v) for(int i=last[v];i;i=next[i])
using namespace std;
typedef long long ll;
const int N=100010,M=N*2,mo=1e9+7;
int n,tot,to[M],next[M],last[N],f[N][2];
ll c[N],g[N][2];
void link(int u,int v)
{
    to[++tot]=v,next[tot]=last[u],last[u]=tot;
}
ll qmi(ll x,ll n)
{
    ll t=1;
    for(;n;n>>=1)
    {
        if(n&1) t=t*x%mo;
        x=x*x%mo;
    }
    return t;
}
ll ny(ll x)
{
    return qmi(x,mo-2);
}
void dfs(int v,int fr)
{
    g[v][0]=g[v][1]=1;
    bool bz=0;
    efo(i,v)
    {
        int u=to[i];
        if(u==fr) continue;
        bz=1;
        dfs(u,v);
        f[v][0]+=max(f[u][0],f[u][1]);
        (g[v][0]*=c[u])%=mo;
    }
    efo(i,v)
    {
        int u=to[i];
        if(u==fr) continue;
        int x=max(f[u][0],f[u][1]);
        int t=f[v][0]-x+f[u][0]+1;
        ll y=g[v][0]*ny(c[u])%mo*g[u][0]%mo;
        if(t==f[v][1]) (g[v][1]+=y)%=mo;
        if(t>f[v][1]) f[v][1]=t,g[v][1]=y;
    }
    if(!bz) c[v]=1;
    else
    if(f[v][0]==f[v][1]) c[v]=(g[v][0]+g[v][1])%mo;
    else
    if(f[v][0]>f[v][1]) c[v]=g[v][0];
    else c[v]=g[v][1];
}
int main()
{
    freopen("hungary.in","r",stdin);
    freopen("hungary.out","w",stdout);
    int _,p,u,v;
    scanf("%d %d",&_,&p);
    while(_--)
    {
        scanf("%d",&n);
        memset(last,0,sizeof(last));
        tot=0;
        fo(i,1,n-1)
        {
            scanf("%d %d",&u,&v);
            link(u,v),link(v,u);
        }
        memset(f,0,sizeof(f));
        dfs(1,1);
        printf("%d ",max(f[1][0],f[1][1]));
        if(p==2) printf("%lld",c[1]);
        printf("\n");
    }
    return 0;
}