oj.uz

Submission #320958

# Submission time Handle Problem Language Result Execution time Memory
320958 2020-11-10T10:51:14 Z prarie Biochips (IZhO12_biochips) C++14
100 / 100
 430 ms 412868 KB 
#include <bits/stdc++.h>
 
#define fastio ios_base::sync_with_stdio(0), cin.tie(0)
#define fi first
#define se second
#define endl '\n'
#define all(v) (v).begin(), (v).end()
#define compress(v) sort(all(v)), (v).erase(unique(all((v))), v.end())

using namespace std;
 
using ll = long long;
using pii = pair<int, int>;
using pll = pair<ll, ll>;
using ld = long double;

const int N = 2e5 + 5;
const int M = 500 + 5;

// INFO
int n, m;
int root;
int a[N];
vector<int> adj[N];

// DFS
int out[N], cnt1 = 1;
int invout[N];
int arr[N << 1], cnt2 = 1;
int freq[N], match[N];

// DP
int dp[N][M];

void dfs(int cur, int pv) {
	arr[cnt2++] = cur;
	for (auto &next : adj[cur]) {
		if (next == pv) continue;
		dfs(next, cur);
	}
	arr[cnt2++] = cur;
	out[cnt1] = cur;
	invout[cur] = cnt1;
	cnt1++;
}

int main() {
	fastio;
	cin >> n >> m;
	for (int i = 1; i <= n; i++) {
		int u;
		cin >> u >> a[i];
		if (u == 0) root = i;
		adj[i].push_back(u);
		adj[u].push_back(i);
	}
	dfs(root, 0);
	for (int i = 1, p = 0; i <= 2 * n; i++) {
		freq[arr[i]]++;
		if (freq[arr[i]] == 2) p = arr[i];
		else match[arr[i]] = p;
	}
	for (int i = 0; i <= n; i++)
		for (int j = 1; j <= m; j++) dp[i][j] = -2e9;
	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= m; j++) {
			dp[i][j] = dp[i - 1][j];
			int node = match[out[i]];
			int idx = invout[node];
			dp[i][j] = max(dp[i][j], dp[idx][j - 1] + a[out[i]]);
		}
	}
	cout << dp[n][m] << endl;
}

Subtask #1
100.0 / 100.0

# Verdict Execution time Memory Grader output
1 Correct 4 ms 5100 KB Output is correct
2 Correct 4 ms 5100 KB Output is correct
3 Correct 4 ms 5328 KB Output is correct
4 Correct 20 ms 23276 KB Output is correct
5 Correct 19 ms 25572 KB Output is correct
6 Correct 20 ms 25580 KB Output is correct
7 Correct 312 ms 307808 KB Output is correct
8 Correct 315 ms 307780 KB Output is correct
9 Correct 354 ms 373988 KB Output is correct
10 Correct 430 ms 412868 KB Output is correct