Problem C: 分糖果

Description

分糖果，排排坐。今天LZY买了N颗糖果，打算分给排好队的创意班M个同学。给第一个同学 1 颗糖果，第二个同学 2 颗，依此类推，直到给最后一个同学 m 颗糖果。

然后，我们再回到队伍的起点，给第一个同学 m + 1 颗糖果，第二个同学 m + 2 颗，依此类推，直到给最后一个同学 2 * m 颗糖果。

重复上述过程（每次都比上一次多给出一颗糖果，当到达队伍终点后再次从队伍起点开始），直到我们分完所有的糖果。注意，就算我们手中的剩下糖果数不够（不比前一次发出的糖果多），这些糖果也会全部发给当前的小朋友。

输出M个同学最终得到的糖果数量

Input

测试样例由多组测试数据组成。每组测试数据第一行输入两个正整数 n ( 1 < = n <= 10^9 ）和 m ( 1 <= m <= 1000）

Output

每组测试数据输出所有同学得到的糖果数量，每个同学之间用空格隔开。

Sample Input

7 4
10 3

Sample Output

1 2 3 1
5 2 3

代码

#include<bits/stdc++.h>
using namespace std;
int main(){
 int n,m,a[1010],b,i;
 while(~scanf("%d %d",&n,&m)){
  memset(a,0,sizeof(a));
  b=0;
  for(i=0;;i++){
   b++;
   if(n>=b){
    a[i]+=b;
    n=n-b;
   }else{
    a[i]+=n;
    n=0;
   }
   if(n==0)
    break;
   if(i+1==m)
    i=-1;
  }
  for(int i=0;i<m;i++){
   if(i==0)
    printf("%d",a[i]);
   else
    printf(" %d", a[i]);
  }
  printf("\n");
 }
 return 0;
} 

思路
输入 7 4
第一次 1 0 0 0
第二次 1 2 0 0
第三次 1 2 3 0
第四次 1 2 3 1给第二个人的糖果比第一个糖果多一个以此累加