Submission #48417

# Submission time Handle Problem Language Result Execution time Memory
48417 2018-05-13T00:49:13 Z aleksami Dostavljač (COCI18_dostavljac) C++14
14 / 140
1025 ms 4768 KB
#include <bits/stdc++.h>

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

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

int main()
{
  ios_base::sync_with_stdio(false);
  cin.tie(NULL);
  cout.tie(NULL);
	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);
	int mx=0;
	for(int i = 0; i < 2; i++)for(int j = 0; j < 2; j++)mx=max(mx,dp[1][m][i][j]);
	cout << mx;
	return 0;
}
# Verdict Execution time Memory Grader output
1 Correct 2 ms 376 KB Output is correct
# Verdict Execution time Memory Grader output
1 Incorrect 3 ms 496 KB Output isn't correct
# Verdict Execution time Memory Grader output
1 Incorrect 6 ms 716 KB Output isn't correct
# Verdict Execution time Memory Grader output
1 Incorrect 3 ms 768 KB Output isn't correct
# Verdict Execution time Memory Grader output
1 Incorrect 29 ms 1284 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 11 ms 1396 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 263 ms 2172 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 92 ms 2684 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 545 ms 3868 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 1025 ms 4768 KB Output isn't correct
2 Halted 0 ms 0 KB -