【PTA数据结构 | C语言版】计算1~n平方的和加上1~n的和

本专栏持续输出数据结构题目集，欢迎订阅。

文章目录

• • 题目
  • 代码

题目

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

输入格式：

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

输出格式：

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

输入样例：

5

输出样例：

340

代码

#include <stdio.h>

int main() {
    int n;
    scanf("%d", &n);
    
    // 计算1到n的和
    int sum_n = n * (n + 1) / 2;
    
    // 计算1到n²的和
    int sum_n_sq = n * n * (n * n + 1) / 2;
    
    // 输出结果
    printf("%d\n", sum_n_sq + sum_n);
    
    return 0;
}