[HAOI 2015] 나무 염색
Code:
#include
#define maxn 2010
#define int long long
using namespace std;
struct Edge{
int to, next, len;
}edge[maxn << 1];
int num, head[maxn], size[maxn], dp[maxn][maxn], n, m;
inline int read(){
int s = 0, w = 1;
char c = getchar();
for (; !isdigit(c); c = getchar()) if (c == '-') w = -1;
for (; isdigit(c); c = getchar()) s = (s << 1) + (s << 3) + (c ^ 48);
return s * w;
}
void addedge(int x, int y, int z){ edge[++num] = (Edge){y, head[x], z}, head[x] = num; }
void dfs(int u, int pre){
for (int i = 2; i <= m; ++i) dp[u][i] = -1;
size[u] = 1;
for (int i = head[u]; i; i = edge[i].next){
int v = edge[i].to;
if (v != pre) dfs(v, u), size[u] += size[v];
}
for (int i = head[u]; i; i = edge[i].next){
int v = edge[i].to;
if (v != pre){
int l = edge[i].len;
for (int j = min(m, size[u]); j >= 0; --j)
for (int k = 0; k <= min(j, size[v]); ++k)
if (dp[u][j - k] >= 0)
dp[u][j] = max(dp[u][j], dp[u][j - k] + dp[v][k] + (m - k) * k * l + (size[v] - k) * (n - size[v] - m + k) * l);
}
}
}
signed main(){
n = read(), m = read();
for (int i = 1; i < n; ++i){
int x = read(), y = read(), z = read();
addedge(x, y, z), addedge(y, x, z);
}
dfs(1, 0);
printf("%lld
", dp[1][m]);
return 0;
}
이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
49일차 - 2022.04.20Baekjoon에서 문제풀이 1) 문제 : 첫째 줄에는 별 1개, 둘째 줄에는 별 2개, N번째 줄에는 별 N개를 찍는 문제/ 첫째 줄에 N(1 ≤ N ≤ 100)이 주어진다. 첫째 줄부터 N번째 줄까지 차례대로 별...
텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.