Submission #204605

#	Time	Username	Problem	Language	Result	Execution time	Memory
204605		MohamedAhmed04	Schools (IZhO13_school)	C++14	100 / 100	146 ms	16088 KiB

school

#include <bits/stdc++.h>

using namespace std ;

const int MAX = 3e5 + 10 ;

long long music[MAX] , sport[MAX] ;
long long pref[MAX] , suff[MAX] ;
int n , m , s ;

int main()
{
	ios_base::sync_with_stdio(0) ;
	cin.tie(0) ;
	cin>>n>>m>>s ;
	vector< pair<int , pair<int , int> > >vp ;
	for(int i = 1 ; i <= n ; ++i)
	{
		cin>>music[i]>>sport[i] ;
		vp.push_back({music[i] - sport[i] , {music[i] , sport[i]}}) ;
	}
	sort(vp.rbegin() , vp.rend()) ;
	for(int i = 1 ; i <= n ; ++i)
		music[i] = vp[i-1].second.first , sport[i] = vp[i-1].second.second ;
	priority_queue<int , vector<int> , greater<int> >q ;
	long long now = 0ll ;
	for(int i = 1 ; i <= n ; ++i)
	{
		if(q.size() >= m && q.size() > 0 && q.top() < music[i])
		{
			now -= q.top() ;
			q.pop() ;
			now += music[i] * 1ll ;
			q.push(music[i]) ;
		}
		else if(q.size() < m)
		{
			now += music[i] * 1ll ;
			q.push(music[i]) ;
		}
		if(q.size() == m)
			pref[i] = now ;
	}
	while(!q.empty())
		q.pop() ;
	now = 0ll ;
	for(int i = n ; i >= 1 ; --i)
	{
		if(q.size() >= s && q.size() > 0 && q.top() < sport[i])
		{
			now -= q.top() ;
			q.pop() ;
			now += sport[i] * 1ll ;
			q.push(sport[i]) ;
		}
		else if(q.size() < s)
		{
			now += sport[i] * 1ll ;
			q.push(sport[i]) ;
		}
		if(q.size() == s)
			suff[i] = now ;
	}
	long long ans = 0ll ;
	for(int i = 0 ; i <= n ; ++i)
		ans = max(ans , pref[i] + suff[i+1]) ;
	return cout<<ans<<"\n" , 0 ;
}		

Compilation message (stderr)

school.cpp: In function 'int main()':
school.cpp:29:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   if(q.size() >= m && q.size() > 0 && q.top() < music[i])
      ~~~~~~~~~^~~~
school.cpp:36:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   else if(q.size() < m)
           ~~~~~~~~~^~~
school.cpp:41:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   if(q.size() == m)
      ~~~~~~~~~^~~~
school.cpp:49:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   if(q.size() >= s && q.size() > 0 && q.top() < sport[i])
      ~~~~~~~~~^~~~
school.cpp:56:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   else if(q.size() < s)
           ~~~~~~~~~^~~
school.cpp:61:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   if(q.size() == s)
      ~~~~~~~~~^~~~

• Subtask 1: 100 / 100