Submission #386703

#	Time	Username	Problem	Language	Result	Execution time	Memory
386703		vanic	Gift (IZhO18_nicegift)	C++14	30 / 100	2094 ms	136812 KiB

nicegift

#include <cstdio>
#include <cmath>
#include <algorithm>
#include <set>
#include <vector>
#include <array>

using namespace std;
typedef long long ll;

const int maxn=1e6+5;

ll a[maxn];
ll sum;
array < ll, 3 > v[maxn*4];
set < int > cur;

void scan(ll &a){
	a=0;
	char c=getchar();
	while(c>='0' && c<='9'){
		a*=10;
		a+=c-'0';
		c=getchar();
	}
}

int main(){
	int n, k;
	scanf("%d%d", &n, &k);
	for(int i=0; i<n; i++){
		scanf("%lld", a+i);
		sum+=a[i];
	}
	if(sum%k){
		printf("-1\n");
		return 0;
	}
	sum/=k;
	for(int i=0; i<n; i++){
		if(a[i]>sum){
			printf("-1\n");
			return 0;
		}
	}
	int pos=0;
	int iduc;
	int br=0;
	for(int i=0; i<n; i++){
		iduc=pos+a[i];
		if(iduc>sum){
			v[br]={pos, i+1, 1};
			br++;
			v[br]={sum, i+1, 0};
			br++;
			iduc-=sum;
			pos=0;
		}
		v[br]={pos, i+1, 1};
		br++;
		v[br]={iduc, i+1, 0};
		br++;
		if(iduc==sum){
			iduc=0;
		}
		pos=iduc;
	}
	sort(v, v+br);
	int sol=0;
	for(int i=0; i<br-1; i++){
		if(v[i][0]!=v[i+1][0]){
			sol++;
		}
	}
	printf("%d\n", sol);
	for(int i=0; i<br; i++){
		if(v[i][2]){
			cur.insert(v[i][1]);
		}
		else{
			cur.erase(v[i][1]);
		}
		if(i==br-1 || v[i][0]==v[i+1][0]){
			continue;
		}
		printf("%lld ", v[i+1][0]-v[i][0]);
		for(set < int >::iterator it=cur.begin(); it!=cur.end(); it++){
			printf("%d ", *it);
		}
		printf("\n");
	}
	return 0;
}

Compilation message (stderr)

nicegift.cpp: In function 'int main()':
nicegift.cpp:30:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   30 |  scanf("%d%d", &n, &k);
      |  ~~~~~^~~~~~~~~~~~~~~~
nicegift.cpp:32:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   32 |   scanf("%lld", a+i);
      |   ~~~~~^~~~~~~~~~~~~

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