Submission #847459

#	Time	Username	Problem	Language	Result	Execution time	Memory
847459		Lib	Closing Time (IOI23_closing)	C++17	0 / 100	59 ms	17200 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 tpos;
		for(int i=0;i<n;i++){
			pq.push({-cost1[i],i});
		}
		while(!pq.empty()&&mcost>0){
			mcost+=pq.top().first;
			if(mcost>=0){
			ans++;
			check[pq.top().second]++;
			//cout<<pq.top().second<<" "<<check[pq.top().second]<<" "<<-pq.top().first<<"\n";
			tpos=pq.top().second;
			pq.pop();
			if(tpos==0&&check[tpos]+check[tpos+1]==2||tpos==n-1&&check[tpos]+check[tpos-1]==2||0<tpos&&tpos<n-1&&check[tpos-1]+check[tpos+1]+check[tpos]==3){
				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:48:14: warning: suggest parentheses around '&&' within '||' [-Wparentheses]
   48 |    if(tpos==0&&check[tpos]+check[tpos+1]==2||tpos==n-1&&check[tpos]+check[tpos-1]==2||0<tpos&&tpos<n-1&&check[tpos-1]+check[tpos+1]+check[tpos]==3){
      |       ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
closing.cpp:48:103: warning: suggest parentheses around '&&' within '||' [-Wparentheses]
   48 |    if(tpos==0&&check[tpos]+check[tpos+1]==2||tpos==n-1&&check[tpos]+check[tpos-1]==2||0<tpos&&tpos<n-1&&check[tpos-1]+check[tpos+1]+check[tpos]==3){
      |                                                                                       ~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 0: 0 / 0
• Subtask 1: 0 / 8
• Subtask 2: 0 / 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