PTA 算法1-5 计算1~ n平方的和加上1~n的和

算法1-5 计算1~ n平方的和加上1~n的和

分数 10

作者 陈越

单位 浙江大学

给定正整数 n，请编写程序，求 (1+2+⋯+n2)+(1+2+⋯+n) 的值。

输入格式：

输入在一行中给出正整数 n（≤100）。

输出格式：

在一行中输出题面中要求的值。

输入样例：

1
5

输出样例：

1
340

答案

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
#include<iostream>
using namespace std;

// 计算m到n的和
int getSum(int m, int n) {
    int sum = 0;
    for (int i = m; i <= n; i++) {
        sum += i;
    }
    return sum;
}
int main() {
    int n;
    cin >> n;
    int sum = 0;
    sum += getSum(1, n);
    sum += getSum(1, n*n);
    cout << sum;
}