Submission #378786

#	Time	Username	Problem	Language	Result	Execution time	Memory
378786		iris2617	Gift (IZhO18_nicegift)	C++14	100 / 100	1189 ms	89212 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<pair<int, vector<int> > > ans;
vector<int> ouo;

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

Compilation message (stderr)

nicegift.cpp: In function 'int main()':
nicegift.cpp:26: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]
   26 |  while(pq.size()>=k)
      |        ~~~~~~~~~^~~
nicegift.cpp:17:12: warning: unused variable 'a' [-Wunused-variable]
   17 |  int n,k,i,a,x,noel,nonoka,sum=0;
      |            ^
nicegift.cpp:17:21: warning: 'nonoka' may be used uninitialized in this function [-Wmaybe-uninitialized]
   17 |  int n,k,i,a,x,noel,nonoka,sum=0;
      |                     ^~~~~~

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