当前位置: 代码迷 >> 综合 >> 【树形DP】【模版题】【p2015】
  详细解决方案

【树形DP】【模版题】【p2015】

热度:25   发布时间:2024-01-19 22:25:12.0

题目:我谷真好看

状态转移方程也就显而易见了:
f[u][i]=max(f[u][i],f[u][i-j-1]+f[v][j]+e[i].w)
( 1≤i≤min(q,sz[u]),0≤j≤min(sz[v],i?1) )

u表示当前节点,v是u的一个子节点,sz[u]表示u的子树上的边数,q就是题目中要求的最多保留边数

代码:

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cctype>
#define ll long long
#define gc getchar
#define maxn 105
using namespace std;inline ll read(){ll a=0;int f=0;char p=gc();while(!isdigit(p)){f|=p=='-';p=gc();}while(isdigit(p)){a=(a<<3)+(a<<1)+(p^48);p=gc();}return f?-a:a;
}int n,m,f[maxn][maxn];struct ahaha{int w,to,next;
}e[maxn<<1];int tot,head[maxn];
inline void add(int u,int v,int w){e[tot]={w,v,head[u]};head[u]=tot++;
}int sz[maxn];
void dfs(int u,int fa){for(int i=head[u];~i;i=e[i].next){int v=e[i].to;if(v==fa)continue;dfs(v,u);sz[u]+=sz[v]+1;for(int j=min(sz[u],m);j;--j)for(int k=min(sz[v],j-1);k>=0;--k)f[u][j]=max(f[u][j],f[u][j-k-1]+f[v][k]+e[i].w);}
}int main(){memset(head,-1,sizeof head);n=read();m=read();for(int i=1;i<n;++i){int u=read(),v=read(),w=read();add(u,v,w);add(v,u,w);}dfs(1,-1);printf("%d\n",f[1][m]);return 0;
}

结束