Submission #922884

#	Time	Username	Problem	Language	Result	Execution time	Memory
922884		ttamx	Magic Tree (CEOI19_magictree)	C++14	100 / 100	97 ms	32848 KiB

magictree

#include<bits/stdc++.h>

using namespace std;

using ll = long long;

const int N=1e5+5;

int n,m,k;
vector<int> adj[N];
int d[N],w[N];
map<int,ll> dp[N];

void dfs(int u){
    for(auto v:adj[u]){
        dfs(v);
        if(dp[v].size()>dp[u].size())swap(dp[u],dp[v]);
        for(auto [x,y]:dp[v])dp[u][x]+=y;
        dp[v].clear();
    }
    if(w[u]){
        dp[u][d[u]]+=w[u];
        int tmp=w[u];
        for(auto it=next(dp[u].find(d[u]));it!=dp[u].end();){
            if(tmp>=it->second){
                tmp-=it->second;
                it=dp[u].erase(it);
            }else{
                it->second-=tmp;
                break;
            }
        }
    }
}

int main(){
    cin.tie(nullptr)->sync_with_stdio(false);
    cin >> n >> m >> k;
    for(int i=2;i<=n;i++){
        int p;
        cin >> p;
        adj[p].emplace_back(i);
    }
    for(int i=1;i<=m;i++){
        int u;
        cin >> u >> d[u] >> w[u];
    }
    dfs(1);
    ll ans=0;
    for(auto [x,y]:dp[1])ans+=y;
    cout << ans;
}

Compilation message (stderr)

magictree.cpp: In function 'void dfs(int)':
magictree.cpp:18:18: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   18 |         for(auto [x,y]:dp[v])dp[u][x]+=y;
      |                  ^
magictree.cpp: In function 'int main()':
magictree.cpp:50:14: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   50 |     for(auto [x,y]:dp[1])ans+=y;
      |              ^

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