oj.uz

Submission #338558

# Submission time Handle Problem Language Result Execution time Memory
338558 2020-12-23T11:54:15 Z beksultan04 Biochips (IZhO12_biochips) C++14
100 / 100
 447 ms 419328 KB 
#include <bits/stdc++.h>
using namespace std;
//#define int long long
#define pii pair<int,int>
#define OK puts("OK");
#define NO puts("NO");
#define YES puts("YES");
#define fr first
#define sc second
#define ret return
#define scan1(a) scanf("%lld",&a);
#define scan2(a,b) scanf("%lld %lld",&a, &b);
#define scan3(a,b,c) scanf("%lld %lld %lld",&a,&b,&c);
#define all(s) s.begin(),s.end()
#define allr(s) s.rbegin(),s.rend()
#define pb push_back
#define sz(v) (int)v.size()
#define endi puts("");
const int N = 1e6+12,INF=1e9+7;
vector <int> g[N];
int l[N],cnt=1,n,m,w[N],ves[N],dp[200001][501];
void dfs(int x){
    int t = cnt,i;
    for (i=0;i<g[x].size();++i){
        dfs(g[x][i]);
    }
    ves[cnt]=w[x];
    l[cnt++]=t;
}



main(){
    int s=0,i,j;
    cin>>n>>m;
    for (i=1;i<=n;++i){
        int a,b;
        cin>>a>>b;
        w[i]=b;
        if (a)
            g[a].pb(i);
        else s = i;
    }
    dfs(s);

    for(int i = 0; i <= n; ++i)
        for(int j = 1; j <= m; ++j)
            dp[i][j] = -INF;
    for (i=1;i<=n;++i){
        for (j=1;j<=m;++j){
            dp[i][j] = max(dp[i-1][j],dp[l[i]-1][j-1]+ves[i]);
        }
    }
    cout <<dp[n][m];
}

Compilation message

biochips.cpp: In function 'void dfs(int)':
biochips.cpp:24:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   24 |     for (i=0;i<g[x].size();++i){
      |              ~^~~~~~~~~~~~
biochips.cpp: At global scope:
biochips.cpp:33:6: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   33 | main(){
      |      ^

Subtask #1
100.0 / 100.0

# Verdict Execution time Memory Grader output
1 Correct 17 ms 23788 KB Output is correct
2 Correct 17 ms 23788 KB Output is correct
3 Correct 18 ms 24044 KB Output is correct
4 Correct 32 ms 41452 KB Output is correct
5 Correct 34 ms 43628 KB Output is correct
6 Correct 34 ms 43628 KB Output is correct
7 Correct 326 ms 317420 KB Output is correct
8 Correct 310 ms 317420 KB Output is correct
9 Correct 447 ms 381420 KB Output is correct
10 Correct 442 ms 419328 KB Output is correct