제출 #76851

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
76851		zetapi	Gift (IZhO18_nicegift)	C++14	7 / 100	2059 ms	314412 KiB

nicegift

#include "bits/stdc++.h"
using namespace std;

#define pb  push_back
#define mp  make_pair
#define	int long long
#define itr iterator

typedef pair<int,int> pii;

const int MAX=5e6;

set<pii> st;

set<pii>::itr it_1,it_2;

vector<int> res,looper;

int N,K,sum,arr[MAX];

void print_1()
{
	assert(!(st.size()%K));
	for(int A=0;A<st.size()/K;A++)
		looper.pb(-st.begin()->first);
	while(!st.empty())
	{	
		res.pb(st.begin()->second);
		st.erase(st.begin());
	}
	cout<<looper.size()<<"\n";
	for(int A=0;A<looper.size();A++)
	{
		cout<<looper[A]<<" ";
		for(int B=K*A;B<K*(A+1);B++)
			cout<<res[B]<<" ";
		cout<<"\n";
	}
	return ;
}

signed main()
{
	ios_base::sync_with_stdio(false);

	cin>>N>>K;
	for(int A=1;A<=N;A++)
	{
		cin>>arr[A];
		sum+=arr[A];
		if(arr[A])
			st.insert(mp(-arr[A],A));
	}
	if(sum%K)
		return cout<<-1,0;
	vector<pii> lol(K);
	int dif;
	while(true)
	{
		if(st.size()<K)
			return cout<<-1,0;
		dif=0;
		it_1=st.begin();
		while(it_1!=st.end())
		{	
			dif=(-st.begin()->first)-(-it_1->first);
			if(dif)
				break;
			++it_1;
		}
		if(!dif)
		{
			/*for(auto A:st)
				cout<<-A.first<<"\n";
			cout<<"\n";*/
			print_1();
			break;
		}
		looper.pb(dif);
		for(int A=0;A<K;A++)
		{
			lol[A]=*st.begin();
			st.erase(st.begin());
		}
		for(int A=0;A<K;A++)
		{
			lol[A].first+=dif;
			if(lol[A].first!=0)
				st.insert(lol[A]);
			res.pb(lol[A].second);
		}
		/*for(auto A:st)
			cout<<-A.first<<" ";
		cout<<"\n";*/
	}
	return 0;
}

컴파일 시 표준 에러 (stderr) 메시지

nicegift.cpp: In function 'void print_1()':
nicegift.cpp:24:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int A=0;A<st.size()/K;A++)
              ~^~~~~~~~~~~~
nicegift.cpp:32:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int A=0;A<looper.size();A++)
              ~^~~~~~~~~~~~~~
nicegift.cpp: In function 'int main()':
nicegift.cpp:60:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   if(st.size()<K)
      ~~~~~~~~~^~

• Subtask 1: 7 / 7
• Subtask 2: 0 / 11
• Subtask 3: 0 / 12
• Subtask 4: 0 / 19
• Subtask 5: 0 / 51