제출 #386865

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
386865		vanic	Gift (IZhO18_nicegift)	C++14	30 / 100	2088 ms	122528 KiB

nicegift

#include <cstdio>
#include <cmath>
#include <algorithm>
#include <set>
#include <cassert>
#include <vector>
 
using namespace std;
typedef long long ll;
 
const int maxn=1e6+5;
 
ll a[maxn];
ll sum;
vector < pair < ll, int > > v;
set < int > cur;
  
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;
	for(int i=0; i<n; ++i){
		iduc=pos+a[i];
		if(iduc>sum){
			v.push_back({pos, i+1});
			iduc-=sum;
			pos=0;
		}
		v.push_back({pos, i+1});
		if(iduc!=sum){
			v.push_back({iduc, -i-1});
		}
		else{
			iduc=0;
		}
		pos=iduc;
	}
	v.push_back({sum, 0});
	sort(v.begin(), v.end());
	int sol=0;
	for(int i=0; i<(int)v.size()-1; ++i){
		if(v[i].first!=v[i+1].first){
			++sol;
		}
	}
	printf("%d\n", sol);
	for(int i=0; i<(int)v.size()-1; ++i){
		if(v[i].second>0){
			cur.insert(v[i].second);
		}
		else{
			cur.erase(-v[i].second);
		}
		if(v[i].first==v[i+1].first){
			continue;
		}
		printf("%lld ", v[i+1].first-v[i].first);
		for(int x : cur){
			printf("%d ", x);
		}
		printf("\n");
	}
	return 0;
}

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

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

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