oj.uz

Submission #245725

# Submission time Handle Problem Language Result Execution time Memory
245725 2020-07-07T09:25:57 Z aloo123 Biochips (IZhO12_biochips) C++14
100 / 100
 673 ms 16488 KB 
#include <algorithm>
#include <bitset>
#include <cassert>
#include <chrono>
#include <cmath>
#include <complex>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <ctime>
#include <deque>
#include <functional>
#include <iomanip>
#include <iostream>
#include <iterator>
#include <limits>
#include <list>
#include <map>
#include <numeric>
#include <queue>
#include <random>
#include <ratio>
#include <set>
#include <sstream>
#include <stack>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <utility>
#include <vector>
#include <climits>
#include <ext/pb_ds/assoc_container.hpp>
// #include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/trie_policy.hpp>
#define ll long long
#define ld long double
#define mp make_pair
#define pb push_back
#define in insert
#define vll vector<ll>
#define endl "\n"
#define pll pair<ll,ll>
#define f first
#define s second
#define int ll
#define sz(x) (ll)x.size()
#define all(x) (x.begin(),x.end())
using namespace std;
using namespace __gnu_pbds;
typedef tree<int,null_type,less<int>,rb_tree_tag,tree_order_statistics_node_update> os;
 
 
const ll INF = 1e12;
const ll N =(2e5+5); // TODO : change value as per problem
// const ll MOD = 1e9+9;
const ll SUM = 20005;
ll a[N];
vll adj[N];
ll k;
ll dp[N];
ll dp2[N];
ll inv[N];
ll jump[N]; 
ll counter = 0;
void dfs(ll u,ll p){
    // counter++;
    ll pre = counter;
    inv[pre] = u;
    counter++;
    for(auto v:adj[u]){
        if(v==p) continue;
        dfs(v,u);
    }
    jump[pre] = counter;
    // counter++;
}
void solve(){
    ll n;
    cin >> n >> k;    
    ll r = 0;
    for(int i =1;i<=n;i++){
        ll u;
        cin >> u >> a[i-1];
        if(u == 0) {r=i-1;continue;} 
        u--;
        adj[u].pb(i-1);
        // adj[i-1].pb(u);
    }
    dfs(r,-1);
    // cout << r << endl;
    // for(int i =0;i<n;i++){
    //     cout << inv[i] << " ";
    // }cout<<endl;
    for(int i =0;i<=n;i++)dp[i] = -INF;
    dp[0] = 0;
    for(int j =0;j<k;j++){
        for(int i = 0;i<=n;i++) dp2[i] = -INF;
        for(int i =0;i<n;i++){
            dp[i+1] = max(dp[i+1],dp[i]);
            dp2[jump[i]] = max(dp2[jump[i]],dp[i] + a[inv[i]]);
        }
        for(int i =0;i<=n;i++) dp[i] = dp2[i]; 
    }
    ll ans = 0;
    for(int i =0;i<=n;i++) ans = max(ans,dp[i]);
    cout << ans << endl;

}

signed main(){

    ios_base::sync_with_stdio(0);
    cin.tie(NULL);
    

 
     ll tt=1;   
     // cin >> tt;
    while(tt--){    
        solve();
    }    
}

Subtask #1
100.0 / 100.0

# Verdict Execution time Memory Grader output
1 Correct 9 ms 5120 KB Output is correct
2 Correct 7 ms 5120 KB Output is correct
3 Correct 7 ms 5120 KB Output is correct
4 Correct 13 ms 5632 KB Output is correct
5 Correct 16 ms 5632 KB Output is correct
6 Correct 18 ms 5760 KB Output is correct
7 Correct 302 ms 14456 KB Output is correct
8 Correct 298 ms 14532 KB Output is correct
9 Correct 477 ms 15588 KB Output is correct
10 Correct 673 ms 16488 KB Output is correct