結果

提出番号	1203
提出者	square1001
言語	C++
提出日時	2018-05-18 21:11:29
問題名	(52)fireworks(花火)
結果	MLE
点数	55%

---

テストケース

ソースコード

#include <set>
#include <vector>
#include <iostream>
#include <algorithm>
using namespace std;
int N, M, p[300009], c[300009], id[300009]; vector<int> g[300009]; multiset<long long> v[300009];
int main() {
	cin.tie(0);
	ios_base::sync_with_stdio(false);
	cin >> N >> M;
	long long sum = 0;
	for (int i = 1; i < N + M; i++) {
		cin >> p[i] >> c[i];
		g[--p[i]].push_back(i);
		sum += c[i];
		if (i >= N) {
			v[i - N].insert(c[i]);
			v[i - N].insert(c[i]);
			id[i] = i - N;
		}
	}
	for (int i = N - 1; i >= 0; i--) {
		int mx = 0;
		for (int j : g[i]) {
			if (mx < v[id[j]].size()) {
				mx = v[id[j]].size();
				id[i] = id[j];
			}
		}
		for (int j : g[i]) {
			if (id[j] != id[i]) {
				int sz = v[id[i]].size();
				v[id[i]].insert(v[id[j]].begin(), v[id[j]].end());
			}
		}
		multiset<long long>::iterator it = --v[id[i]].end();
		for (int j = 1; j < g[i].size(); j++) {
			it = --v[id[i]].erase(it);
		}
		long long va = *it; it = --v[id[i]].erase(it);
		long long vb = *it; it = --v[id[i]].erase(it);
		v[id[i]].insert(va + c[i]);
		v[id[i]].insert(vb + c[i]);
	}
	vector<long long> res(v[id[0]].begin(), v[id[0]].end());
	for (int i = 0; i < res.size(); i++) {
		sum -= (res[i] - (i == 0 ? 0 : res[i - 1])) * (res.size() - i - 1);
	}
	cout << sum << '\n';
	return 0;
}