PTA作业： 7-3 建国的数学难题 #C语言代码及Python代码解析#

题目：众所周知，建国是一个数学天才，但是今天他被下面这道题考到了，你能帮建国解决这个难题吗？

f(1) = k

f(2) = f(1) + 1

f(3) = f(2) + 1 + 2

......

f(n) = f(n-1) + (1 + 2 + ... + n-1)

输入格式:

第一行输出一个整数T，表示样例数。(1 <= T <= 100)

每个样例占一行，输入两个整数n，k。（0 < n, k <= 1000) 。

输出格式:

每个样例输出一个整数表示f(n)。

输入样例:

2
1 1
2 3

输出样例:

1
4

代码长度限制

16 KB

时间限制

400 ms

内存限制

C语言代码：

#include<stdio.h>
int main()
{
    int T,n,k,f[1000],i,j;
    scanf("%d",&T);
    for(i=1;i<=T;i++)
    {
        scanf("%d %d",&n,&k);
        f[0]=k;
        for(j=1;j<n;j++)
        {
            f[j]=f[j-1]+j*(j+1)/2;
        }
        printf("%d\n",f[n-1]);
    }
    return 0;
}

Python代码：

T=int(input())
for i in range(T):
    n,k=map(int,input().split())
    f=k
    for j in range(n):
        f=f+j*(j+1)/2
    print("%d"%f)