【牛客网华为机试】HJ100 等差数列

题目

描述

功能:等差数列 2，5，8，11，14。。。。

输入:正整数N >0

输出:求等差数列前N项和

本题为多组输入，请使用while(cin>>)等形式读取数据

输入描述：

输入一个正整数。

输出描述：

输出一个相加后的整数。

示例1

输入：

2

输出：

7

解题思路

（1）计算公差为3的等差数列，输入前N项

（2）首项为2，末项为2+3*（n-1）

（3）输出str(int((3*num+1)*num/2))

代码

while True:
    try:
        num = int(input())
        print(str(int((3*num+1)*num/2)))
    except Exception as e:
        break

<