牛客算法题 HJ100 等差数列 golang语言实现

算法题目

HJ100 等差数列

描述
等差数列 2，5，8，11，14。。。。
（从 2 开始的 3 为公差的等差数列）
输出求等差数列前n项和


数据范围： 
1
≤
�
≤
1000
 
1≤n≤1000 
输入描述：
输入一个正整数n。

输出描述：
输出一个相加后的整数。

示例1
输入：
2
复制
输出：
7
复制
说明：
2+5=7  
示例2
输入：
275
复制
输出：
113575
复制
说明：
2+5+...+821+824=113575  

golang语言实现

package main

import (
	"bufio"
	"fmt"
	"os"
	"strconv"
)

func main() {
    n:=0
    sum:=0
    scan:=bufio.NewScanner(os.Stdin)
    for scan.Scan(){
        line:=scan.Text()
        n,_=strconv.Atoi(line)
    }
    // fmt.Printf("%d", n)
    n_val:=2
    if n>0 {
        for i:=1;i<=n;i++{
            if i==1{
                n_val = 2
            }else{
                n_val=n_val+3
            }
            sum = sum+n_val
        }
    }
    fmt.Printf("%d", sum)

}

##运行结果