HDU 4003 Find Metal Mineral(分组背包+树形DP)

题目链接

很棒的一个树形DP。学的太渣了。

 #include <cstdio>
#include <string>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
int dp[][];
struct node
{
int u,v,w,next;
}edge[];
int first[],t,flag[];
int n,m;
void CL()
{
t = ;
memset(first,-,sizeof(first));
memset(flag,,sizeof(flag));
}
void add(int u,int v,int w)
{
edge[t].u = u;
edge[t].v = v;
edge[t].w = w;
edge[t].next = first[u];
first[u] = t ++;
}
void dfs(int u)
{
int i,v,j,k;
flag[u] = ;
for(i = first[u];i != -;i = edge[i].next)
{
v = edge[i].v;
if(flag[v]) continue;
dfs(v);
for(j = m;j >= ;j --)
{
dp[u][j] += dp[v][] + *edge[i].w;
for(k = ;k <= j;k ++)
{
dp[u][j] = min(dp[u][j],dp[u][j-k] + dp[v][k] + k*edge[i].w);
}
}
}
}
int main()
{
int s,u,v,w,i;
while(scanf("%d%d%d",&n,&s,&m)!=EOF)
{
CL();
memset(dp,,sizeof(dp));
for(i = ;i < n;i ++)
{
scanf("%d%d%d",&u,&v,&w);
add(u,v,w);
add(v,u,w);
}
dfs(s);
printf("%d\n",dp[s][m]);
}
return ;
}
上一篇:(笔记)Linux内核学习(四)之系统调用


下一篇:从零自学Hadoop(20):HBase数据模型相关操作上