이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#pragma GCC optimize("O3", "unroll-loops")
using namespace std;
using ll = long long;
const int N = 300'000 + 10;
const ll INF = 1'000'000'000'000'000'000;
int n;
ll s;
vector<int> g[N];
ll cost[N];
int parent[N];
namespace sub1 {
bool check() {
return (s == INF);
}
ll dp[N];
void dfs(int u) {
dp[u] = cost[u];
for (const auto& v : g[u]) {
dfs(v);
dp[u] += dp[v];
}
dp[u] = max(dp[u], 0ll);
}
void solve() {
dfs(0);
// for (int u = 1; u <= n; ++ u) cout << dp[u] << " \n"[u == n];
cout << dp[0];
}
}
namespace sub23 {
bool check() {
for (int u = 1; u <= n; ++ u) {
if (parent[u] != 0 && parent[u] != u - 1) return false;
}
return true;
}
ll sum[N], mi[N];
void solve() {
ll first = s;
vector<pair<ll, ll>> candidates;
mi[0] = INF;
for (int u = 1; u <= n; ++ u) {
int p = parent[u];
ll add = sum[p] + cost[u];
mi[u] = min(mi[p], add);
if (add > 0) {
candidates.push_back(make_pair(-mi[u], add));
sum[u] = 0;
}
else {
sum[u] = add;
}
}
sort(candidates.begin(), candidates.end());
for (const auto& [need, add] : candidates) {
if (s - need < 0) {
break;
}
s += add;
}
cout << s - first;
}
}
namespace sub4 {
bool check() {
return (n <= 2'000);
}
struct state {
int u;
ll cost;
state() : u(0), cost(0ll) {};
state(int _u, ll _cost) : u(_u), cost(_cost) {};
bool operator<(const state& other) const {
if (this->cost != other.cost) return this->cost > other.cost;
return this->u > other.u;
}
};
set<state> tree[N];
ll mi[N], sum[N];
void dfsUpdate(int u) {
int curU = -1;
if (cost[u] >= 0) {
curU = u;
}
else if (!tree[u].empty() && tree[u].begin()->cost >= 0) {
curU = tree[u].begin()->u;
tree[u].erase(tree[u].begin());
}
if (curU > 0) {
while (!tree[u].empty() && tree[u].begin()->cost >= 0) {
auto [v, curCost] = *tree[u].begin();
cost[v] = 0;
cost[curU] += curCost;
tree[u].erase(state(v, curCost));
tree[curU].insert(tree[v].begin(), tree[v].end());
tree[v].clear();
}
if (curU != u) {
tree[u].insert(state(curU, cost[curU]));
}
}
for (const auto& [v, c] : tree[u]) {
dfsUpdate(v);
}
}
void updateTree() {
for (int u = 0; u <= n; ++ u) {
for (const auto& v : g[u]) {
tree[u].insert(state(v, cost[v]));
}
g[u].clear();
}
dfsUpdate(0);
for (int u = 0; u <= n; ++ u) {
for (const auto& [v, c] : tree[u]) {
g[u].push_back(v);
}
}
}
void dfsRem(int u, int& best) {
int p = parent[u];
ll add = sum[p] + cost[u];
mi[u] = min(mi[p], add);
if (add > 0) {
if (best == -1 || mi[u] > mi[best]) {
best = u;
}
return;
}
for (const auto& v : g[u]) {
dfsRem(v, best);
}
}
void solve() {
updateTree();
ll first = s;
for (int cnt = 1; cnt <= 2'000; ++ cnt) {
int best = -1;
for (int u = 0; u <= n; ++ u) {
mi[u] = INF;
sum[u] = 0;
}
dfsRem(0, best);
if (best == -1 || -mi[best] > s) {
break;
}
ll ss = 0;
while (best != 0) {
ss += cost[best];
cost[best] = 0;
best = parent[best];
}
s += ss;
}
cout << s - first;
}
}
int32_t main() {
std::ios_base::sync_with_stdio(false);
std::cin.tie(nullptr); std::cout.tie(nullptr);
cin >> n >> s;
for (int u = 1; u <= n; ++ u) {
cin >> cost[u] >> parent[u];
g[parent[u]].push_back(u);
}
if (sub1::check()) {
sub1::solve();
return 0;
}
if (sub23::check()) {
sub23::solve();
return 0;
}
if (sub4::check()) {
sub4::solve();
return 0;
}
assert(false);
return (0 ^ 0);
}
/// Code by vuavisao
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |