bzoj1785 [Usaco2010 Jan]telephone

明显的贪心。。先跑树内，树内不行匹配树外。。

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
#define inf 0x3f3f3f3f3f3f3f3fll
using namespace std;
const int N=3e5+5;
typedef long long ll;
int n,m,k;
int head[N],next[N],go[N];
int val[N],f[N],ans;
bool g[N];
int tot;
inline void add(int x,int y)
{
    go[++tot]=y;
    next[tot]=head[x];
    head[x]=tot;
}
inline void dfs(int x,int fa)
{
    if (next[head[x]]==0)
    {
        g[x]=1;
        if (x!=1)return;
    }
    for(int i=head[x];i;i=next[i])
    {
        int v=go[i];
        if (v!=fa)
        {
            dfs(v,x);
            if (g[v])
            {
                if (g[x])
                {
                    f[x]++;
                    g[x]=0;
                }
                else 
                if (f[x]!=k)g[x]=1;
            }
        }
    }
    ans+=f[x];
}
int main()
{
    scanf("%d%d",&n,&k);
    fo(i,1,n-1)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y),add(y,x);
    }
    dfs(1,0);
    printf("%d\n",ans);
}