codeforces 1029E Tree with Small Distances

简述题意：

给定一棵n个节点树（），要求往树中加入一些边使得从1到其他节点的距离至多是2 。 输出加入边的最小数量。(边全部都是无向的)

算法：玄学图论！

难度：NOIP+

题解：

因为本题中所要求的最短dis<=2，所以我们将所有结点u到1的dis通过%3来分类，as usual，我们自下向上进行更新（仿佛有一点像树型DP）。我们处理出一个结点x所有儿子y%3后的距离的最小值min_dis，

那么一个结点向1连一条边的充要条件就是: &&  && 

时间复杂度：O(n)

代码如下：

#include <cstdio>
#include <iostream>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <algorithm>
#include <queue>
#define ll long long
#define N 200005
using namespace std;
struct node
{
    int next;
    int to;
}edg[N<<1];
int hea[N];
int cnt=1;
void init()
{
    memset(hea,-1,sizeof(hea));
    cnt=1;
}
void add(int u,int v)
{
    edg[cnt].next=hea[u];
    edg[cnt].to=v;
    hea[u]=cnt++;
}
int ans;
int dfs(int x,int fa)
{
    int dis=2;//这里赋值成2的原因！！认真思考！
    /*多练习树型DP的思路！！！*/
    for(int i = hea[x];i != -1;i=edg[i].next)
    {
        int to=edg[i].to;
        if(to==fa) continue;
        dis=min(dis,dfs(to,x));
    }
    if(x!=1&&fa!=1&&dis==0) ans++;
    return (dis+1)%3;
}
int main()
{
    int n;
    scanf("%d",&n);
    init();
    for(int i = 1;i < n;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        add(u,v),add(v,u);
    }
    dfs(1,-1); 
    printf("%d\n",ans);
    return 0 ;
}