先枚举物品组,再枚举体积,再枚举决策(每一组里选哪个物品)
先阶段,再状态,再决策
无向边要在遍历时定义father,防止来回遍历
解释f[u][j-k-1]+f[e[i]][k]+w[i]
意思为,在没用掉k+1条边之前的最大价值,+用掉k+1条边之后的最大价值
//先枚举物品组,再枚举体积,再枚举决策(每一组里选哪个物品)
//先阶段,再状态,再决策
#include <iostream>
#include <cstring>
using namespace std;
const int N=110;
int h[N],e[N*2],ne[N*2],idx;
int w[N];
int f[N][N];
int n,m;
void add(int a,int b,int c)
{
e[idx]=b;
ne[idx]=h[a];
w[idx]=c;
h[a]=idx++;
}
//无向边要设置fa,防止来回遍历!!
void dfs(int u,int fa)
{
for(int i=h[u];i!=-1;i=ne[i])
{
if(e[i]==fa) continue;
dfs(e[i],u);
for(int j=m;j>=0;j--)
{
for(int k=0;k<j;k++)
{
f[u][j]=max(f[u][j],f[u][j-k-1]+f[e[i]][k]+w[i]);
}
}
}
}
int main()
{
cin>>n>>m;
memset(h,-1,sizeof h);
for(int i=0;i<n-1;i++)
{
int a,b,c;
cin>>a>>b>>c;
add(a,b,c);
add(b,a,c);
}
dfs(1,-1);
cout<<f[1][m];
return 0;
}
二刷,要注意的地方如下
- 在dfs中,每次更新的是f[u][j]而不是f[i][j]!!
- 输入的边为i-1条
#include <iostream>
#include <cstring>
using namespace std;
const int N=210;
int h[N],e[N],ne[N],w[N];
int idx;
int n,m;
int f[N][N];
void add(int a,int b,int c)
{
e[idx]=b;
ne[idx]=h[a];
w[idx]=c;
h[a]=idx++;
}
void dfs(int u,int fa)
{
for(int i=h[u];i!=-1;i=ne[i])
{
if(e[i]==fa) continue;
dfs(e[i],u);
for(int j=m;j>=0;j--)
{
for(int k=0;k<j;k++)
{
f[u][j]=max(f[u][j],f[u][j-k-1]+f[e[i]][k]+w[i]);
}
}
}
}
int main()
{
cin>>n>>m;
memset(h,-1,sizeof h);
for(int i=1;i<=n-1;i++)
{
int a,b,c;
cin>>a>>b>>c;
add(a,b,c);
add(b,a,c);
}
dfs(1,-1);
cout<<f[1][m];
return 0;
}