【HDU】2015 偶數求和

Problem Description

有一個長度爲n(n<=100)的數列，該數列定義爲從2開始的遞增有序偶數，現在要求你按照順序每m個數求出一個平均值，如果最後不足m個，則以實際數量求平均值。編程輸出該平均值序列。

 

Input

輸入數據有多組，每組佔一行，包含兩個正整數n和m，n和m的含義如上所述。

Output

對於每組輸入數據，輸出一個平均值序列，每組輸出佔一行。 

Sample Input

3 2 4 2

 

Sample Output

3 6 3 7 

AC代碼：

//
//  main.cpp
//  2015
//
//  Created by showlo on 2018/4/14.
//  Copyright © 2018年 showlo. All rights reserved.
//

#include <stdio.h>
#include <algorithm>
#include <math.h>
using namespace std;

int main() {
    int n,m,num;
    int a[110],b;
    int ans[110];
    while (scanf("%d %d",&n,&m)!=EOF) {
        memset(ans, 0, sizeof(ans));
        memset(a, 0, sizeof(a));
        num=0;
        b=0;
        a[0]=2;
        for (int i=1; i<n; i++) {
            a[i]=a[i-1]+2;
        }
        
        for (int i=0; i<n; i+=m) {
            for (int j=0; j<m; j++) {
                b=b+a[i+j];
            }
            if(n-i<m)
                ans[num]=b/(n-i);
            else
                ans[num]=b/m;
            num++;
            b=0;
        }
        printf("%d",ans[0]);
        for (int i=1; i<num; i++)
            printf(" %d",ans[i]);
        printf("\n");
    }
    return 0;
}