Submission #378724

#	Time	Username	Problem	Language	Result	Execution time	Memory
378724		iris2617	Gift (IZhO18_nicegift)	C++14	30 / 100	2111 ms	353424 KiB

nicegift

#include<bits/stdc++.h>
#define int long long
#define matsuri pair<int,int>
#define iris 1000000007
using namespace std;

priority_queue<matsuri> pq;
int arr[1000010];
vector<vector<int> > ans;
vector<int> ouo;

signed main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	
	int n,k,i,a;
	
	cin>>n>>k;
	for(i=1;i<=n;i++)
	{
		cin>>arr[i];
		pq.push({arr[i],i});
	}
	while(pq.size()>=k)
	{
		ouo.clear();
		for(i=0;i<k;i++)
		{
			ouo.emplace_back(pq.top().second);
			pq.pop();
		}
		ans.emplace_back(ouo);
		for(int aoi:ouo)
		{
			arr[aoi]--;
			if(arr[aoi])
				pq.push({arr[aoi],aoi});
		}
	}
	if(pq.empty())
	{
		cout<<ans.size()<<'\n';
		for(auto sana:ans)
		{
			cout<<"1";
			for(int aoi:sana)
			{
				cout<<" "<<aoi;
			}
			cout<<'\n';
		}
	}
	else
	{
		cout<<"-1\n";
	}
	
	return 0;
}

Compilation message (stderr)

nicegift.cpp: In function 'int main()':
nicegift.cpp:25:17: warning: comparison of integer expressions of different signedness: 'std::priority_queue<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} and 'long long int' [-Wsign-compare]
   25 |  while(pq.size()>=k)
      |        ~~~~~~~~~^~~
nicegift.cpp:17:12: warning: unused variable 'a' [-Wunused-variable]
   17 |  int n,k,i,a;
      |            ^

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