Submission #990335

#	Time	Username	Problem	Language	Result	Execution time	Memory
990335		imarn	Magic Tree (CEOI19_magictree)	C++14	0 / 100	34 ms	21844 KiB

magictree

#include<bits/stdc++.h>
#define ll long long
#define pii pair<int,ll>
#define f first
#define s second
#define pb push_back
#define all(x) x.begin(),x.end()
using namespace std;
const int mxn=1e5+5;
int d[mxn]{0},w[mxn];
map<int,int>dp[mxn];
vector<int>g[mxn];
void dfs(int u){
    for(auto v:g[u]){
        dfs(v);
        if(dp[v].size()>dp[u].size())swap(dp[u],dp[v]);
        for(auto it:dp[v]){
            dp[u][it.f]+=it.s;
        }
    }
    if(!d[u])return;
    dp[u][d[u]]+=w[u];int cur=w[u];
    auto it = dp[u].upper_bound(d[u]);
    while(it!=dp[u].end()){
        if(cur>=it->s){
            cur-=it->s;
            dp[u].erase(it->f);
        }
        else {
            dp[u][it->f]-=cur;
        }it++;
    }
}
int main(){
    ios_base::sync_with_stdio(0);cin.tie(0);
    int n,m,k;cin>>n>>m>>k;
    for(int i=2;i<=n;i++){
        int p;cin>>p;g[p].pb(i);
    }
    for(int i=1;i<=m;i++){
        int v;cin>>v;cin>>d[v]>>w[v];
    }dfs(1);ll ans=0;
    for(auto it : dp[1])ans+=it.s;cout<<ans;
}

Compilation message (stderr)

magictree.cpp: In function 'int main()':
magictree.cpp:43:5: warning: this 'for' clause does not guard... [-Wmisleading-indentation]
   43 |     for(auto it : dp[1])ans+=it.s;cout<<ans;
      |     ^~~
magictree.cpp:43:35: note: ...this statement, but the latter is misleadingly indented as if it were guarded by the 'for'
   43 |     for(auto it : dp[1])ans+=it.s;cout<<ans;
      |                                   ^~~~

• Subtask 1: 0 / 6
• Subtask 2: 0 / 3
• Subtask 3: 0 / 11
• Subtask 4: 0 / 12
• Subtask 5: 0 / 16
• Subtask 6: 0 / 13
• Subtask 7: 0 / 22
• Subtask 8: 0 / 17