16分求调
查看原帖
16分求调
1098351
hxd2011楼主2024/12/28 15:19
#include<bits/stdc++.h>
using namespace std;
const int maxn=3e4+5,INF=0x3f3f3f3f;
int f[105][maxn],a[maxn];
vector<pair<int,int>>g[maxn];
int n,m,ans;
void dfs(int u,int p,int pw)
{
	for(int j=0;j<=m;j++)	f[u][j]=0;
	for(auto pii:g[u])
	{
		int v=pii.first;
		int w=pii.second;
		if(v==p)	continue;
		dfs(u,v,w);
		for(int j=m;j>=0;j--)
		{
			for(int k=0;k<=m;k++)
			{
				if(j-k-1<0)	break;
				f[u][j]=max(f[u][j],f[u][j-k-1]+f[v][k]+w);
			}
		}
	}
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)	cin>>a[i];
    for(int i=1,u,v,w;i<n;i++)
    {
    	cin>>u>>v>>w;
    	g[u].push_back({v,w});
    	g[v].push_back({u,w});
	}
	int rt=1;
	dfs(rt,0,0);
    cout<<f[1][m]<<endl;
    return 0;
}

我也不知道哪里错了(初学树形DP)

2024/12/28 15:19
加载中...