3-8 C: 分糖果

题目描述

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

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

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

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

输入

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

输出

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

样例输入 Copy

7 4
10 3

样例输出 Copy

1 2 3 1
5 2 3

水题++

AC代码：

#include<bits/stdc++.h>
 
using namespace std;
long long a[1005];
long long n,m,s,d; 
int main(){
    while(cin>>n>>m){
        s=1;
        d=0;
        memset(a,0,sizeof a);
        while(s<n){
            a[d]+=s;
            n-=s;
            s++;
            d++;
            if(d==m){
                d=0;
            }
        }
        a[d]+=n;
        for(int i=0;i<m;i++){
            if(i!=0)cout<<" ";
            cout<<a[i];
        }
        cout<<endl;
    }
     
    return 0;
}