Submission #48861

#	Time	Username	Problem	Language	Result	Execution time	Memory
48861		mrtsima22	Gift (IZhO18_nicegift)	C++17	30 / 100	134 ms	6896 KiB

nicegift

#include <bits/stdc++.h>
using namespace std;
#define lo long
const long long lmax=1999999999;
const long long lmx=1999999999999999999;
long long n,k,p;
long long a[100003],sum;
set<pair<long long,long long> >s;
vector<vector<long long> > ans;
int main(){std::ios::sync_with_stdio(false);
cin>>n>>k;
for(long long i=1;i<=n;i++)
{
	cin>>a[i];
	sum=sum+a[i];
}
bool ok=1;
for(long long i=2;i<=n;i++)
{
	ok=ok&(a[i]==a[i-1]);
}
if(!(sum<=1000000||ok)){
cout<<-1<<endl;return 0;}
if(sum%k!=0){
cout<<-1<<endl;return 0;}

for(long long i=1;i<=n;i++)
{
	s.insert(make_pair(a[i],i));
}
p=1;
if(ok){
	p=sum/(n*(k/__gcd(n,k)));
}
while(s.size())
{
	long long t=0;
	vector<long long>v;
	while(s.size()&&t<k)
	{
		set<pair<long long,long long> >::iterator it=s.end();
		it--;
		long long u=(*(it)).second;
		s.erase(it);
		v.push_back(u);
		a[u]-=p;
		t++;
	}
	if(t!=k){cout<<-1<<endl;return 0;
	}
	for(long long 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(long long i=0;i<ans.size();i++)
{
	cout<<p<<" ";
	for(long long j=0;j<ans[i].size();j++)
	{
		cout<<ans[i][j]<<" ";
	}
	cout<<endl;
}
}
/*

                   *         *
                  * *       * *
                 *   *     *   *
                *     *   *     *
               *       * *       *
               *        *        *
                *               *
                 *             *
                  *           *
                   *         *
                    *       *
                     *     *
                      *   *
                       * *
                        *



*/

Compilation message (stderr)

nicegift.cpp: In function 'int main()':
nicegift.cpp:51:21: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(long long i=0;i<v.size();i++)
                    ~^~~~~~~~~
nicegift.cpp:61:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
 for(long long i=0;i<ans.size();i++)
                   ~^~~~~~~~~~~
nicegift.cpp:64:21: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(long long 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