제출 #92348

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
92348		Pajaraja	Gift (IZhO18_nicegift)	C++17	49 / 100	2077 ms	46076 KiB

nicegift

#include <bits/stdc++.h>
using namespace std;
long long gcd(long long a,long long b)
{
	if(a==0) return b;
	return gcd(b%a,a);
}
int main()
{
	long long n,k,cnt=0;
	long long t,sum=0,mi=1000000000000000000LL;
	scanf("%lld%lld",&n,&k);
	priority_queue<pair<long long,int> > pq,pq1;
	for(int i=0;i<n;i++) {scanf("%lld",&t); pq.push(make_pair(t,i+1)); pq1.push(make_pair(t,i+1)); mi=min(mi,t); sum+=t;}
	if(sum%k!=0) {printf("-1"); return 0;}
	while(pq.size()>=k && mi!=pq.top().first)
	{
		stack<pair<long long,int> > st;
		for(int i=0;i<k;i++) {st.push(pq.top()); pq.pop();}
		long long f=st.top().first;
		if(!pq.empty()) f-=pq.top().first;
		f=max(f,1LL);
		for(int i=0;i<k;i++) {st.top().first-=f; if(st.top().first!=0) pq.push(st.top()); mi=min(mi,st.top().first); st.pop();}
		cnt++;
	}
	if(pq.size()%n!=0) {printf("-1"); return 0;}
	if(!pq.empty() && pq.top().first%(k/gcd(n,k))!=0) {printf("-1"); return 0;}
	if(pq.empty()) printf("%lld\n",cnt);
	else printf("%lld\n",cnt+(n*mi)/(k*(mi/(k/gcd(n,k)))));
	while(cnt--)
	{
		stack<pair<long long,int> > st;
		for(int i=0;i<k;i++) {st.push(pq1.top()); pq1.pop();}
		long long f=st.top().first;
		if(!pq1.empty()) f-=pq1.top().first;
		f=max(f,1LL);
		printf("%lld ",f);
		for(int i=0;i<k;i++) {st.top().first-=f; if(st.top().first!=0) pq1.push(st.top()); printf("%d ",st.top().second); st.pop();}
		printf("\n");
	}
	long long f=pq.top().first/(k/gcd(n,k));
	while(pq.size()>=k)
	{
		stack<pair<long long,int> > st;
		for(int i=0;i<k;i++) {st.push(pq.top()); pq.pop();}
		printf("%lld ",f);
		for(int i=0;i<k;i++) {st.top().first-=f; if(st.top().first!=0) pq.push(st.top()); printf("%d ",st.top().second); st.pop();}
		printf("\n");
	}
}

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

nicegift.cpp: In function 'int main()':
nicegift.cpp:16:17: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  while(pq.size()>=k && mi!=pq.top().first)
        ~~~~~~~~~^~~
nicegift.cpp:42:17: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  while(pq.size()>=k)
        ~~~~~~~~~^~~
nicegift.cpp:12:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  scanf("%lld%lld",&n,&k);
  ~~~~~^~~~~~~~~~~~~~~~~~
nicegift.cpp:14:29: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  for(int i=0;i<n;i++) {scanf("%lld",&t); pq.push(make_pair(t,i+1)); pq1.push(make_pair(t,i+1)); mi=min(mi,t); sum+=t;}
                        ~~~~~^~~~~~~~~~~

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