Submission #52023

#	Time	Username	Problem	Language	Result	Execution time	Memory
52023		mrtsima22	Gift (IZhO18_nicegift)	C++17	30 / 100	2069 ms	167268 KiB

nicegift

#include <bits/stdc++.h>
using namespace std;
#define lo long
const int lmax=1999999999;
const long long lmx=1999999999999999999;
int n,k;
long long a[1000003],sum,p;
set<pair<long long,int> >s;
vector<vector<int> > ans;
int main(){std::ios::sync_with_stdio(false);
cin>>n>>k;
for(int i=1;i<=n;i++)
{
	cin>>a[i];
	sum=sum+a[i];
}
if(sum%k!=0){
cout<<-1<<endl;return 0;}
 
for(int i=1;i<=n;i++)
{
	s.insert(make_pair(a[i],i));
}
p=1;
while(s.size())
{
	int t=0;
	vector<int>v;
	while(s.size()&&t<k)
	{
		set<pair<long long,int> >::iterator it=s.end();
		it--;
		int u=(*(it)).second;
		s.erase(it);
		v.push_back(u);
		a[u]-=p;
		t++;
	}
	if(t!=k){cout<<-1<<endl;return 0;
	}
	for(int i=0;i<v.size();i++)
	{
		if(a[v[i]]>0)
		{
			s.insert(make_pair(a[v[i]],v[i]));
		}
	}
	ans.push_back(v);
}
cout<<ans.size()<<endl;
for(int i=0;i<ans.size();i++)
{
	cout<<p<<" ";
	for(int j=0;j<ans[i].size();j++)
	{
		cout<<ans[i][j]<<" ";
	}
	cout<<endl;
}
}
/*
70. 
71.                   *         *
72.                  * *       * *
73.                 *   *     *   *
74.                *     *   *     *
75.               *       * *       *
76.               *        *        *
77.                *               *
78.                 *             *
79.                  *           *
80.                   *         *
81.                    *       *
82.                     *     *
83.                      *   *
84.                       * *
85.                        *
86. 
87. 
88. 
89.*/

Compilation message (stderr)

nicegift.cpp: In function 'int main()':
nicegift.cpp:41:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i=0;i<v.size();i++)
              ~^~~~~~~~~
nicegift.cpp:51:14: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
 for(int i=0;i<ans.size();i++)
             ~^~~~~~~~~~~
nicegift.cpp:54:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int j=0;j<ans[i].size();j++)
              ~^~~~~~~~~~~~~~

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