Submission #245721

#	Time	Username	Problem	Language	Result	Execution time	Memory
245721		aloo123	Biochips (IZhO12_biochips)	C++14	0 / 100	7 ms	5120 KiB

biochips

#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 pre = counter;
    inv[pre] = u;
    for(auto v:adj[u]){
        dfs(v);
    }
    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;continue;} 
        u--;
        adj[u].pb(i-1);
    }
    dfs(r);
    // cout << r << 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: 0 / 100