이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
#define int long long
#ifdef LOCAL
void debug_out() {cerr<<endl;}
template <typename Head, typename... Tail>
void debug_out(Head _H, Tail... _T) {cerr<<" "<<to_string(_H);debug_out(_T...);}
#define debug(...) cerr<<"["<<#__VA_ARGS__<<"]:",debug_out(__VA_ARGS__)
#else
#define debug(...)
#endif
const int MAXN = 300005;
const int inf=1000000500ll;
const long long oo =1000000000000000500ll;
const int MOD = (int)1e9+7;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
typedef pair<int,int> pi;
int n,init;
int A[MAXN], P[MAXN];
vector<int>child[MAXN];
int lazy[MAXN];
priority_queue<pi,vector<pi>,greater<pi>> * dfs(int x){
priority_queue<pi,vector<pi>,greater<pi>> * pq = new priority_queue<pi,vector<pi>,greater<pi>>();
for(auto v:child[x]){
priority_queue<pi,vector<pi>,greater<pi>> * c = dfs(v);
if(c->size() > pq->size()){
swap(c,pq);
swap(lazy[v],lazy[x]);
}
while(c->size()){
pi cur = c->top();
pq->push({cur.first+lazy[v] - lazy[x],cur.second});
c->pop();
}
}
if(A[x]>=0){
pq->push({-lazy[x],A[x]});
} else{
int s=A[x];
while(pq->size()){
if(pq->top().second + s < 0){
s+=pq->top().second;
pq->pop();
} else{
pi c=pq->top();
int rem=pq->top().second + s;
pq->pop();
pq->push({c.first,rem});
break;
}
}
lazy[x]+=-A[x];
}
return pq;
}
int32_t main()
{
ios_base::sync_with_stdio(0); cin.tie(0);
cin >> n >> init;
for(int i=1;i<=n;i++){
cin>>A[i]>>P[i];
child[P[i]].push_back(i);
}
priority_queue<pi,vector<pi>,greater<pi>> * pq =dfs(0);
int ans=0;
while(pq->size() && pq->top().first+lazy[0]<=init){
ans += pq->top().second;
init += pq->top().second;
pq->pop();
}
cout<<ans;
}
# | 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... |