组队赛的时候的一道题,那个时候想了一下感觉dp不怎么好写呀,现在写了出来,交上去过了,但是我觉得我还是应该WA的呀,因为总感觉dp的不对。

#pragma warning(disable:4996)
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#include<vector>
#define maxn 150
using namespace std;

struct Edge
{
	int v, w;
	Edge(int vi, int wi) :v(vi), w(wi){}
	Edge(){}
};

vector<Edge> G[maxn];
int n, q;
int dp[maxn][maxn];
int tmp[maxn];

void dfs(int u, int fa)
{
	for (int i = 0; i < G[u].size(); i++){
		int v = G[u][i].v, w = G[u][i].w;
		if (v == fa) continue;
		dfs(v, u);
		memcpy(tmp, dp[u], sizeof(dp[u]));
		for (int k = q; k >= 0; k--){
			for (int j = k - 1; j >= 0; j--){
				dp[u][k] = max(dp[u][k], dp[v][j] + tmp[k-1- j] + w);
			}
		}
	}
}

int main()
{
	while (cin >> n >> q)
	{
		for (int i = 0; i <= n; i++) G[i].clear();
		int ui, vi, wi;
		for (int i = 0; i < n - 1; i++){
			scanf("%d%d%d", &ui, &vi, &wi);
			G[ui].push_back(Edge(vi, wi));
			G[vi].push_back(Edge(ui, wi));
		}
		memset(dp, 0, sizeof(dp));
		dfs(1, -1);
		cout << dp[1][q] << endl;
	}
	return 0;
}

 

相关文章:

  • 2021-08-13
  • 2021-09-18
  • 2021-07-11
  • 2022-12-23
  • 2021-10-26
  • 2021-09-25
  • 2021-12-31
  • 2021-05-21
猜你喜欢
  • 2022-02-04
  • 2021-07-07
  • 2022-01-17
  • 2022-12-23
  • 2022-01-03
  • 2022-03-01
  • 2022-01-18
相关资源
相似解决方案