Submission #847518

#	Time	Username	Problem	Language	Result	Execution time	Memory
847518		Lib	Closing Time (IOI23_closing)	C++17	9 / 100	67 ms	16204 KiB

closing

#include "closing.h"
#include <bits/stdc++.h>
using namespace std;
long long pos[200003];
long long cost1[200003];
long long cost2[200003];
long long pfx1[200003];
long long pfx2[200003];
long long check[200003];
long long maxroll[200003]; //maxroll[i]=a <==> de "tra no" 1 khoang toi da =a, can rollback it nhat i vi tri tu loai 1 xuong loai 0
long long minupd[200003]; //minupt[i]=k <==> voi 1 budget<=i; upgrade duoc toi da k vi tri tu loai 1 len loai 2
struct cus{
bool operator()(const long long &x, const long long &y){
	return x>y;
}
};
int max_score(int N, int X, int Y, long long K, std::vector<int> U, std::vector<int> V, std::vector<int> W)
{
	int n=N;
	int x=X;
	int y=Y;
	long long k=K;
	cin>>x>>y>>k;
	for(int i=0;i<n-1;i++){
		cin>>U[i]>>V[i]>>W[i];
		pos[i+1]=pos[i]+W[i];
	}
	for(int i=0;i<n;i++){
		cost1[i]=min(abs(pos[i]-pos[x]),abs(pos[i]-pos[y]));
		cost2[i]=max(abs(pos[i]-pos[x]),abs(pos[i]-pos[y]))-cost1[i];
		pfx1[i+1]=pfx1[i]+cost1[i];
	}
	priority_queue <pair <long long,int> > pq;
	long long mcost=k;
	long long ans=0,tans=0;
	int c;
	int ccnt=0;
	int tpos;
	//phase 1: thi nang cap cac diem len toi da lv1
	for(int i=0;i<n;i++){
		pq.push({-cost1[i],i});
	}
	ans=0;
	while(mcost>0&&!pq.empty()){
		mcost+=pq.top().first;
		ans++;
		pq.pop();
	}
	if(mcost<0){
		ans--;
	}
	tans=max(tans,ans);
	while(!pq.empty()){
		pq.pop();
	}
	//phase 2: co ca loai 1 lan 2. Khi do, thang loai 2 chi duoc accept khi o day da co 1 roi
	mcost=k;
	mcost-=(pfx1[y+1]-pfx1[x]);
		if(mcost>0){
		ans=0;
		for(int i=0;i<n;i++){
			if(i<x||i>y){
			pq.push({-cost1[i],i});
			}else{
				pq.push({0,i});
			}
		}
		while(!pq.empty()&&mcost>0){
			mcost+=pq.top().first;
			if(mcost>=0){
			ans++;
			tpos=pq.top().second;
			pq.pop();
			check[tpos]++;
			//cout<<tpos<<" "<<check[tpos]<<" "<<-pq.top().first<<"\n";
			if(check[tpos]==1){
				pq.push({-cost2[tpos],tpos});
			}
			}
		}
		tans=max(tans,ans);
		}
	return tans;
}

Compilation message (stderr)

closing.cpp: In function 'int max_score(int, int, int, long long int, std::vector<int>, std::vector<int>, std::vector<int>)':
closing.cpp:36:6: warning: unused variable 'c' [-Wunused-variable]
   36 |  int c;
      |      ^
closing.cpp:37:6: warning: unused variable 'ccnt' [-Wunused-variable]
   37 |  int ccnt=0;
      |      ^~~~

• Subtask 0: 0 / 0
• Subtask 1: 0 / 8
• Subtask 2: 9 / 9
• Subtask 3: 0 / 12
• Subtask 4: 0 / 14
• Subtask 5: 0 / 9
• Subtask 6: 0 / 11
• Subtask 7: 0 / 10
• Subtask 8: 0 / 10
• Subtask 9: 0 / 17