PTA--递归： 7-2 建国的数学难题

 7-2 建国的数学难题

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

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

内存限制

64 MB

#include <stdio.h>

int T, n, k;

int f2(int n){
    if(n==1)
        return 0;
    return f2(n-1)+n-1;
}

int f1(int n){
    if(n==1)
        return k;
    return f1(n-1)+f2(n-1)+n-1;//或者写成f1(n-1)+f2(n)
}

int main(){
    scanf("%d", &T);
    while(T--){
        scanf("%d%d", &n, &k);
        printf("%d\n", f1(n));
    }
    return 0;
}