#include <bits/stdc++.h>
#define fr first
#define sc second
#define mk make_pair
#define pb emplace_back
#define all(s) s.begin(), s.end()
using namespace std;
const int N = 2e5 + 5;
int n, m, x, y, root, a[N];
int dp[N][505], tin[N], tiktak, inf = 1e9 + 7, tout[N], ans, last;
vector < vector <int> > g;
vector < pair <int, int> > vec;
void dfs (int v)
{
tin[v] = ++tiktak;
for (auto to : g[v])
dfs(to);
tout[v] = tiktak;
vec.pb( mk( tout[v], v ) );
}
main()
{
cin >> n >> m;
g.resize(n + 1);
for (int i = 1; i <= n; i++){
scanf("%d%d", &x, &a[i]);
if (x == 0)
root = i;
else
g[x].pb(i);
}
dfs(root);
sort( all(vec) );
for (int i = 0; i <= n; i++)
for (int j = 1; j <= m; j++)
dp[i][j] = -inf;
for (auto it : vec)
{
for (int j = 1; j <= m; j++)
dp[it.fr][j] = max( dp[ tin[it.sc] - 1 ][j - 1] + a[it.sc], dp[it.fr][j] );
for (int l = last + 1; l <= it.fr; l++)
for (int j = 0; j <= m; j++)
dp[l][j] = max( dp[l][j], dp[l - 1][j] );
last = it.fr;
}
for (int j = 1; j <= n; j++)
ans = max(dp[j][m], ans);
cout << ans << endl;
}
Compilation message
biochips.cpp:33:6: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
main()
^
biochips.cpp: In function 'int main()':
biochips.cpp:39:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%d%d", &x, &a[i]);
~~~~~^~~~~~~~~~~~~~~~~~~
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Incorrect |
2 ms |
376 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
