#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define f first
#define s second
#define pb push_back
#define ep emplace
#define eb emplace_back
#define lb lower_bound
#define ub upper_bound
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
#define uniquev(v) sort(all(v)), (v).resize(unique(all(v)) - (v).begin())
#define mem(f,x) memset(f , x , sizeof(f))
#define sz(x) (ll)(x).size()
#define __lcm(a, b) (1ll * ((a) / __gcd((a), (b))) * (b))
#define mxx *max_element
#define mnn *min_element
#define cntbit(x) __builtin_popcountll(x)
#define len(x) (int)(x.length())
const int N = 2e5 + 10;
vector <int> ed[N];
int sz[N], a[N];
int f[N][501];
int n, m;
void dfs(int u) {
f[u][0] = 0;
sz[u] = 1;
for (auto x : ed[u]) {
dfs(x);
sz[u] += sz[x];
for (int i = min(sz[u], m); i >= max(1, sz[u] - sz[x] - m); i--) {
for (int j = 0; j <= sz[x] && j <= i; j++) {
if (f[x][j] == -1)
continue;
f[u][i] = max(f[u][i], f[x][j] + f[u][i - j]);
}
}
}
f[u][1] = max(f[u][1], a[u]);
}
int main() {
ios_base::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
mem(f, -1);
cin >> n >> m;
int root = 0;
for (int i = 1; i <= n; i++) {
int x;
cin >> x >> a[i];
if (x) {
ed[x].pb(i);
} else {
root = i;
}
}
dfs(root);
cout << f[root][m] << '\n';
return 0;
}
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
101 ms |
398672 KB |
Output is correct |
| 2 |
Incorrect |
100 ms |
398756 KB |
Output isn't correct |
| 3 |
Incorrect |
96 ms |
398676 KB |
Output isn't correct |
| 4 |
Incorrect |
99 ms |
399000 KB |
Output isn't correct |
| 5 |
Incorrect |
115 ms |
398896 KB |
Output isn't correct |
| 6 |
Incorrect |
108 ms |
398904 KB |
Output isn't correct |
| 7 |
Incorrect |
195 ms |
399856 KB |
Output isn't correct |
| 8 |
Incorrect |
192 ms |
399796 KB |
Output isn't correct |
| 9 |
Incorrect |
207 ms |
399864 KB |
Output isn't correct |
| 10 |
Incorrect |
264 ms |
399880 KB |
Output isn't correct |
