程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
 程式師世界 >> 編程語言 >> C語言 >> C++ >> 關於C++ >> hdu 5148 Cities(樹形dp)

hdu 5148 Cities(樹形dp)

編輯:關於C++

題目鏈接:hdu 5148 Cities

dp[i][j]表示以i為根節點,選j個最優值,每條邊被選中的時候就計算出被經過的次數,並乘上權值。

#include 
#include 
#include 
#include 

using namespace std;

typedef pair pii;
typedef long long ll;
const int maxn = 2005;

int N, K;
ll ans, dp[maxn][55];
vector g[maxn];

void cmin(ll& v, ll x) {
	if (v == -1 || v > x)
		v = x;
}

void init () {
	scanf("%d%d", &N, &K);

	ans = -1;
	memset(dp, -1, sizeof(dp));
	for (int i = 1; i <= N; i++)
		g[i].clear();

	int a, b, c;
	for (int i = 1; i < N; i++) {
		scanf("%d%d%d", &a, &b, &c);
		g[a].push_back(make_pair(b, c));
		g[b].push_back(make_pair(a, c));
	}
}

void dfs(int u, int f) {
	dp[u][1] = 0;

	for (int i = 0; i < g[u].size(); i++) {
		int v = g[u][i].first;
		int w = g[u][i].second * 2;
		if (v == f) continue;
		dfs(g[u][i].first, u);
		for (int k = K - 1; k; k--) {
			if (dp[u][k] == -1) continue;
			for (int j = 1; j + k <= K; j++) {
				if (dp[v][j] == -1) continue;
				cmin(dp[u][k+j], dp[u][k] + dp[v][j] + w * (K - j) * j);
			}
		}
	}
	if (dp[u][K] != -1)
		cmin(ans, dp[u][K]);
}

int main () {
	int cas;
	scanf("%d", &cas);
	while (cas--) {
		init();
		dfs(1, 0);
		printf("%I64d\n", ans);
	}
	return 0;
}


 

  1. 上一頁:
  2. 下一頁:
Copyright © 程式師世界 All Rights Reserved