oj.uz

Submission #319385

# Submission time Handle Problem Language Result Execution time Memory
319385 2020-11-05T05:35:25 Z CodeTiger927 Dostavljač (COCI18_dostavljac) C++14
140 / 140
 7 ms 4336 KB 
using namespace std;

#include <iostream>
#include <vector>
#include <cstring>

#define MAXN 505

long long dp[MAXN][2][MAXN],arr[MAXN],tmp[MAXN],tmp2[MAXN];
int N,M,sz[MAXN],parent[MAXN];
vector<int> edges[MAXN];

void dfs(int cur) {
	// cout << cur << endl;
	sz[cur] = 1;
	dp[cur][0][1] = arr[cur];
	dp[cur][1][1] = arr[cur];
	for(int i = 0;i < edges[cur].size();++i) {
		int next = edges[cur][i];
		if(next == parent[cur]) continue;
		parent[next] = cur;
		dfs(next);
		for(int j = 0;j <= M;++j) {
			tmp[j] = dp[cur][0][j];
			tmp2[j] = dp[cur][1][j];
		}

		for(int j = 0;j <= 3 * sz[cur];++j) {
			for(int k = 0;k <= 3 * sz[next];++k) {
				if(cur == 2 && j + k == 5) {
					// cout << "hi " << j << " " << k << " " << dp[cur][0][j] << " " << dp[next][0][k] << endl;
				}
				if(j + k + 2 <= M) tmp[j + k + 2] = max(tmp[j + k + 2],dp[cur][0][j] + dp[next][0][k]);
				if(j + k + 1 <= M) tmp2[j + k + 1] = max(tmp2[j + k + 1],dp[cur][0][j] + dp[next][1][k]);
				if(j + k + 2 <= M) tmp2[j + k + 2] = max(tmp2[j + k + 2],dp[cur][1][j] + dp[next][0][k]);
			}
		}
		memcpy(dp[cur][0],tmp,sizeof(tmp));
		memcpy(dp[cur][1],tmp2,sizeof(tmp2));

		sz[cur] += sz[next];

	}

}

int main() {
	cin >> N >> M;
	for(int i = 1;i <= N;++i) {
		cin >> arr[i];
	}
	for(int i = 0;i < N - 1;++i) {
		int a,b;
		cin >> a >> b;
		edges[a].push_back(b);
		edges[b].push_back(a);
	}
	parent[1] = 1;
	dfs(1);
	long long ans = 0;
	for(int i = 0;i <= M;++i) {
		ans = max(ans,max(dp[1][0][i],dp[1][1][i]));
	}
	cout << ans << endl;
	return 0;
}

Compilation message

dostavljac.cpp: In function 'void dfs(int)':
dostavljac.cpp:18:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   18 |  for(int i = 0;i < edges[cur].size();++i) {
      |                ~~^~~~~~~~~~~~~~~~~~~

Subtask #1
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 492 KB Output is correct

Subtask #2
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 620 KB Output is correct

Subtask #3
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 748 KB Output is correct

Subtask #4
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 876 KB Output is correct

Subtask #5
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 1132 KB Output is correct
2 Correct 1 ms 1132 KB Output is correct

Subtask #6
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 1516 KB Output is correct
2 Correct 2 ms 1536 KB Output is correct

Subtask #7
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 1900 KB Output is correct
2 Correct 2 ms 1900 KB Output is correct

Subtask #8
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 4 ms 2668 KB Output is correct
2 Correct 4 ms 2668 KB Output is correct

Subtask #9
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 3564 KB Output is correct
2 Correct 4 ms 3564 KB Output is correct

Subtask #10
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 6 ms 4332 KB Output is correct
2 Correct 7 ms 4336 KB Output is correct