五角数

五角数

题目描述
一个五角数被定义为n(3n-1)/2，其中n>0 且n=1,2,…。所以，开始的几个数字就是1，5，12，22，…，编写一个方法根据用户输入，返回一个五角数，n不符合要求输出"输入错误"
输入一个整数
输出该数对应的五角数

样例输入
1 2 -1
样例输出
1
5
输入错误

import java.util.Scanner;

public class Main {
 public static void main(String[] args) {
  Scanner input = new Scanner(System.in);
  int n , s;
  while(input.hasNext()) {
   n = input.nextInt();
   if(n > 0) {
    s = pentagon(n);
    System.out.println(s);
   }else {
    System.out.println("输入错误");
   }
  }
 }
 public static int pentagon(int a) {
   int s = a * (3 * a - 1) / 2;
   return s;
 }
}