走楼梯算法

import java.util.Scanner;

//走楼梯：2种方法： 一次一级/一次两级。 楼梯有n级时有多少种走法。
public class Main{

public static int getResult(int count) {

int digit = 4;

if(count < digit) {
return count;
}

int result = digit;


for(int i = digit; i < count + 1; i++) {

result += (i - digit) * 2 + 1;
}

return result;
}

public static void main(String[] args) {

Scanner input = new Scanner(System.in);

int times = 0;

try {
times = input.nextInt();
} catch (Exception e) {

}


if (times > 0) {
System.out.println(getResult(times));
}else {
System.out.println(0);
}


}
}