Submission #546930

#	Time	Username	Problem	Language	Result	Execution time	Memory
546930		pure_mem	Magic Tree (CEOI19_magictree)	C++14	0 / 100	2085 ms	21392 KiB

magictree

#pragma GCC optimize("Ofast")

#include <bits/stdc++.h>

#define ll long long
#define ld long double
#define X first
#define Y second
#define MP make_pair

using namespace std;

const int N = 1e5 + 1;
const ll INF = 1e7;

typedef map<int, ll> DP;

int n, m, k;
vector< int > g[N];
DP dp[N];
pair<int, int> bomb[N];

ostream& operator<< (ostream& out, DP& a) {
	for(auto [pos, val]: a) {
		out << pos << " " << val << "\n";
	}
	return out;
}

void insert(ll cost, int pos, DP &v) {
	v[pos] += cost;
	DP::iterator it = v.upper_bound(pos);

	ll add = 0;
	while(it != v.end()) {
		if(it->second + add - cost <= 0) {
			add += it->second;
			v.erase(it++);
		}
		else {
			it->second += add - cost;
			break;
		}
	}
}

void merge(DP &child, DP &v) {
	while(!child.empty()) {
		ll cost = child.begin()->second;
		int pos = child.begin()->first;
		child.erase(child.begin());

		v[pos] += cost;
		DP::iterator it = v.upper_bound(pos);
		if(it != v.end())
			it->second -= cost;
	}
	
	/*
	ll cost = 0;
	int pos = 0;
	while(!child.empty()) {
		cost += child.begin()->second;
		pos = child.begin()->first;
		child.erase(child.begin());
		
		insert(cost, pos, v);
	}*/
}
 
void dfs(int v) {
	//cerr << v << "\n";
	for(int to: g[v]) {
		dfs(to);
		if(dp[v].size() > dp[to].size())
			dp[v].swap(dp[to]);
		merge(dp[to], dp[v]);		
	}
	
	insert(bomb[v].Y, bomb[v].X, dp[v]);	
	//cerr << "\n" << dp[v] << "\n";
	//cerr << v << "\n";
}

int main () {
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);

	cin >> n >> m >> k;
	for(int i = 2, v;i <= n;i++) {
		cin >> v;
		g[v].push_back(i);
	}

	for(int i = 0;i < m;i++) {
		int v, d, w;
		cin >> v >> d >> w;	
		bomb[v] = {d, w};
	}
	dfs(1);

	ll res = 0;
	for(auto [pos, val]: dp[1]) {
		res += val;
	}
	cout << res;
}

Compilation message (stderr)

magictree.cpp: In function 'std::ostream& operator<<(std::ostream&, DP&)':
magictree.cpp:24:11: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   24 |  for(auto [pos, val]: a) {
      |           ^
magictree.cpp: In function 'int main()':
magictree.cpp:104:11: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
  104 |  for(auto [pos, val]: dp[1]) {
      |           ^

• 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