楊輝三角的變形

題目描述

1
         1  1  1
      1  2  3  2  1
   1  3  6  7  6  3  1
1  4  10 16 19  16 10  4  1
以上三角形的數陣，第一行只有一個數1，如下每行的每一個數，是剛好是它上面的數，左上角數到右上角的數，3個數之和（若是不存在某個數，認爲該數就是0）。
求第n行第一個偶數出現的位置。若是沒有偶數，則輸出-1。例如輸入3,則輸出2，輸入4則輸出3。

輸入n(n <= 1000000000)

輸入描述

輸入一個int整數

輸出描述

輸出返回的int值

輸入例子

4

輸出例子

3

算法實現

import java.util.Arrays;
import java.util.Scanner;

/*
 * All Rights Reserved !!!
 */
public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
//        Scanner scanner = new Scanner(Main.class.getClassLoader().getResourceAsStream("data.txt"));
        while (scanner.hasNext()) {
            int input = scanner.nextInt();
            System.out.println(findIndex(input));
        }

        scanner.close();
    }

    private static int findIndex(int n) {
        if (n <= 2) {
            return -1;
        }
        int[][] arr = {new int[2 * n - 1], new int[2 * n - 1], new int[2 * n - 1]};

        arr[0][0] = 1;
        arr[1][0] = 1;
        arr[1][1] = 1;
        arr[1][2] = 1;

        for (int i = 2; i < n; i++) {
            int curr = i % 3;
            int prev = (i - 1) % 3;

            int last = 2 * i; // 最後一個下標

            arr[curr][0] = 1; // 設置第一個值
            arr[curr][1] = i; // 設置第二個值
            arr[curr][last] = 1; // 設置倒數第一個值
            arr[curr][last - 1] = i; // 設置倒數第二個值

            for (int j = 2; j <= last - 2; j++) {
                arr[curr][j] = arr[prev][j - 2] + arr[prev][j - 1] + arr[prev][j];
            }
        }

        int curr = (n - 1) % 3;

        for (int i = 0; i < arr[curr].length; i++) {
            if (arr[curr][i] % 2 == 0) {
                return i + 1;
            }
        }

        return -1;
    }
}