【P1303】苹果二叉树

树归入门题

原题：

 有一棵苹果树，如果树枝有分叉，一定是分2叉（就是说没有只有1个儿子的结点）。这棵树共有N个结点（叶子点或者树枝分叉点），编号为1-N,树根编号一定是1。
我们用一根树枝两端连接的结点的编号来描述一根树枝的位置。下面是一颗有4个树枝的树： 
2 5 
/ 
3   4 
  / 
  1 
现在这颗树枝条太多了，需要剪枝。但是一些树枝上长有苹果。
给定需要保留的树枝数量，求出最多能留住多少苹果。

1<=Q<= N,1<N<=100

每根树枝上的苹果不超过30000个。

核心思想是用f[x][y]表示以x为根的子树保留y条根的最大值

每次枚举i，f[x][y]=max(f[x][y],f[lchild][i]+f[rchild][y-i-1])，如果f[lchild][i]==0或f[rchild][y-i-1]==0，进去递归，求出f[lchild][i]或f[rchild][y-i-1]再更新

当size==1或y==0时要特判

刚开始写的时候写成先分别把lchild和rchild从0-size[lchild]或size[rchild]的最优值求出来，在枚举i更新x的从0-size[x]的最优值，不过这么做好想错了，看了oy学长的程序才改过来

不知道像上面这么写↑能不能写对，还需要做更多的题总结

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
int read(){int z=0,mark=1;  char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')mark=-1;  ch=getchar();}
    while(ch>='0'&&ch<='9'){z=(z<<3)+(z<<1)+ch-'0';  ch=getchar();}
    return z*mark;
}
struct ddd{int next,y,value;}e[210];int LINK[110],ltop=0;
inline void insert(int x,int y,int z){e[++ltop].next=LINK[x];LINK[x]=ltop;e[ltop].y=y;e[ltop].value=z;}
struct dcd{int lchild,rchild,value,size;}tree[210];
int n,m;
bool visited[110];
int f[110][110];
void get_tree(int x,int y){
    visited[x]=true;
    tree[x].value=y;
    for(int i=LINK[x];i;i=e[i].next)if(!visited[e[i].y]){
        if(!tree[x].lchild)  tree[x].lchild=e[i].y;
        else  tree[x].rchild=e[i].y;
        get_tree(e[i].y,e[i].value);
    }
    tree[x].size=tree[tree[x].lchild].size+tree[tree[x].rchild].size+1;
}
void tree_DP(int x,int y){
    if(y==0)  f[x][y]=0;
    else if(!tree[x].lchild && !tree[x].rchild)  f[x][y]=tree[x].value;
    else{
        for(int i=0;i<y;i++){
            if(f[tree[x].lchild][i]==0)  tree_DP(tree[x].lchild,i);
            if(f[tree[x].rchild][y-i-1]==0)  tree_DP(tree[x].rchild,y-i-1);
            f[x][y]=max(f[x][y],f[tree[x].lchild][i]+f[tree[x].rchild][y-i-1]);
        }
        f[x][y]+=tree[x].value;
    }
}
int main(){//freopen("ddd.in","r",stdin);
    memset(visited,0,sizeof(visited));
    memset(tree,0,sizeof(tree));
    memset(f,0,sizeof(f));
    cin>>n>>m;
    int _left,_right,_value;
    for(int i=1;i<n;i++){
        _left=read();  _right=read();  _value=read();
        insert(_left,_right,_value);  insert(_right,_left,_value);
    }
    get_tree(1,0);
    tree_DP(1,m+1);
    cout<<f[1][m+1]<<endl;
    return 0;
}