【简单】HJ100 等差数列(Python两种写法)

【题目】
等差数列 2，5，8，11，14。。。。
（从 2 开始的 3 为公差的等差数列）

输出求等差数列前n项和

数据范围： 1≤n≤1000 1≤n≤1000
输入描述：

输入一个正整数n。
输出描述：

输出一个相加后的整数。
【示例1】
输入：2
输出：7
【代码】
【写法1】盲目累计

import sys
for line in sys.stdin:
    ans=0
    num=int(line)
    base=2
    while num:
        num-=1
        ans+=base
        base+=3
    print(ans)

【写法2】公式法，使用等差数列求和公式Sn=(a1+an)*n/2

import sys
for line in sys.stdin:
    ans=0
    num=int(line)
    print((2+(3*num-1))*num//2)