ZZULIOJ 2453: 数字分组

题目描述

给你n个数字，不改变数字的顺序，将连续的数字分为大小相等的k组，输出各组数字的数量。
若无法分解，则输出-1.

输入

第一行为两个数字n和k，表示数字的数量和要分的组数。
第二行为n个数字，每个数字都在int范围内。
0 < n <= 10^6
k <= n

输出

输出有一行，为各组数字的数量。
若无结果，输出-1.

样例输入

5 3
1 4 5 2 3

样例输出

2 1 2

#include<stdio.h>
int a[1000050];
int main()
{
	int n,k,sum=0;
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		sum=sum+a[i];
	}
	if(sum%k==0)
	{
		int flag=0;
		int som=sum/k;
		sum=0;
		for(int j=1;j<=n;j++)
		{
			sum=sum+a[j];
			flag++;
			if(sum==som)
			{
				sum=0;
				printf("%d%c",flag,j==n?'\n':' ');
				flag=0;
			}
		}
	}
	else
	printf("-1");
	return 0;
}