oj.uz

답안 #48421

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
48421 2018-05-13T01:00:27 Z aleksami Dostavljač (COCI18_dostavljac) C++14
14 / 140
 240 ms 2472 KB 
#include <bits/stdc++.h>

using namespace std;
#define MAXN 505
int a[MAXN];
int dp[MAXN][MAXN][2];
vector <int> g[MAXN];
int n,m;

void dfs(int v,int p)
{
	for(auto x:g[v])
	{
		if(x==p)continue;
		dfs(x,v);
		for(int i = m; i >= 0; i--)
		{
			for(int j = 0; j <= i; j++)
			{
				if(i-j-1>=0)dp[v][i][0]=max(dp[v][i][0],dp[v][j][1]+max(dp[x][i-j-1][0],dp[x][i-j-1][1]));
				if(i-j-2>=0)dp[v][i][1]=max(dp[v][i][1],dp[v][j][1]+dp[x][i-j-2][1]);
			}
		}
	}
	for(int i = m; i >= 1; i--)
	{
		dp[v][i][0]=max(dp[v][i][0],max(dp[v][i-1][0],dp[v][i-1][1])+a[v]);
		dp[v][i][1]=max(dp[v][i][1],dp[v][i-1][1]+a[v]);
	}
	for(int i = 1; i <= m; i++)
	{
		dp[v][i][0]=max(dp[v][i][0],dp[v][i-1][0]);
		dp[v][i][1]=max(dp[v][i][1],dp[v][i-1][1]);
	}
}

int main()
{
	cin >> n >> m;
	for(int i = 1; i <= n; i++)
	{
		cin >> a[i];
	}
	for(int i = 0; i < n-1; i++)
	{
		int u,v;
		cin >> u >> v;
		g[u].push_back(v);
		g[v].push_back(u);
	}
	dfs(1,0);
	cout << max(dp[1][m][0],dp[1][m][1]);
	return 0;
}

Subtask #1
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 376 KB Output is correct

Subtask #2
0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 2 ms 616 KB Output isn't correct

Subtask #3
0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 4 ms 672 KB Output isn't correct

Subtask #4
0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 2 ms 720 KB Output isn't correct

Subtask #5
0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 9 ms 904 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #6
0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 5 ms 1032 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #7
0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 68 ms 1316 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #8
0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 26 ms 1704 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #9
0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 126 ms 2028 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #10
0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 240 ms 2472 KB Output isn't correct
2 Halted 0 ms 0 KB -