【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;
}