loj10153二叉苹果树

有一棵二叉苹果树，如果数字有分叉，一定是分两叉，即没有只有一个儿子的节点。这棵树共 N 个节点，标号 1 至 N，树根编号一定为 1。

我们用一根树枝两端连接的节点编号描述一根树枝的位置。一棵有四根树枝的苹果树，因为树枝太多了，需要剪枝。但是一些树枝上长有苹果，给定需要保留的树枝数量，求最多能留住多少苹果。

输入格式

第一行两个数 N 和 Q ，N 表示树的节点数，Q 表示要保留的树枝数量。

接下来 N−1 行描述树枝信息，每行三个整数，前两个是它连接的节点的编号，第三个数是这根树枝上苹果数量。

输出格式

输出仅一行，表示最多能留住的苹果的数量。

样例

样例输入

5 2
1 3 1
1 4 10
2 3 20
3 5 20

样例输出

21

数据范围与提示

对于 100% 的数据，1≤Q≤N≤100,N≠1，每根树枝上苹果不超过 30000 个。

————————————————————————————————————————————————————

树形动归，

开始写的是首先把树变成二叉树，然后动归，记过严重超时，69分，直接在树上进行动归，就可以了，而且时间快很多！！

————————————————————————————————————————————————————

#include<bits/stdc++.h>
using namespace std;
const int maxn=105;
struct edge
{
    int u,v,w,nxt;
}e[maxn<<1];
int head[maxn],js;
int n,m;
int f[maxn][maxn];
struct node
{
    int w,lc,rc;
}tree[maxn];
int lc[maxn],rc[maxn],ww[maxn];
inline void addage(int u,int v,int w)
{
    e[++js].u=u;e[js].v=v;e[js].w=w;
    e[js].nxt=head[u];head[u]=js;
}
int dfs(int u,int fa,int s)
{
    if(s==1)return f[u][s]=ww[u];
    if(s==0)return 0;
    if(f[u][s]>0)return f[u][s];
    for(int i=head[u];i;i=e[i].nxt)
    {
        int v=e[i].v;
        
        if(v!=fa)
        {
            ww[v]=e[i].w;
            if(lc[u]==0)lc[u]=v;
            else rc[u]=v;
            for(int j=0;j<s;++j)
            {
                dfs(v,u,j);
            }
        }
    }
    for(int i=0;i<s;++i)
        f[u][s]=max(f[u][s],f[lc[u]][i]+f[rc[u]][s-i-1]+ww[u]);
    return f[u][s];
}
void readint(int &x)
{
    x=0;
    char c=getchar();
    for(;c>'9'||c<'0';c=getchar());
    for(;c>='0' && c<='9';c=getchar())x=x*10+c-'0';
}
int main()
{
    readint(n);readint(m);
    for(int u,v,w,i=1;i<n;++i)
    {
        readint(u);readint(v);readint(w);
        addage(u,v,w);
        addage(v,u,w);
    }
    dfs(1,0,m+1);
    cout<<f[1][m+1];
    return 0;
}