Submission #994752

#	Time	Username	Problem	Language	Result	Execution time	Memory
994752		kym	Jobs (BOI24_jobs)	C++14	100 / 100	193 ms	72900 KiB

Main

#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];

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);
		while(c->size()){
			pq->push(c->top());c->pop();
		}
	}
	
	if(A[x]>=0){
		pq->push({0,A[x]});
	} else{
		int s = A[x];
		int lim = -A[x];
		
		while(s<0 && pq->size()){

			lim = max(lim, pq->top().first - s);// you actually need this amount of initial capital now.
			s+=pq->top().second;
			pq->pop();
			
		}
		
	
		int sum=0;
		while(pq->size() && pq->top().first < lim){
			sum += pq->top().second;
			pq->pop();
		}
		if(sum){
			pq->push({lim,sum});
		}
		if(s>0)pq->push({lim,s});
	}
	

	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<=init){
		ans += pq->top().second;
		init += pq->top().second;
		pq->pop();
	}
	cout<<ans;
}

• Subtask 1: 11 / 11
• Subtask 2: 14 / 14
• Subtask 3: 15 / 15
• Subtask 4: 29 / 29
• Subtask 5: 31 / 31