Program Mencari Jumlah Garis Kotak

Pada kesempatan ini saya akan membagikan program sederhana untuk mencari jumlah baris dari kota.

Jumlah garis = 4

Jumlah garis = 27





Source code program :
import java.util.ArrayList;
import java.util.Scanner;
public class MencariJumlahGarisKotak {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        ArrayList<Integer> N = new ArrayList<>();
        boolean a = true;
        while (a) {
            System.out.print("Masukkan jumlah kotak : ");
            String X = in.next();
            if (!X.matches("[0-9]*")) {
                a = false;
            } else {
                N.add(Integer.parseInt(X));
            }
        }
        int hasil = 0;
        int selisih = 0;
        for (int i = 0; i < N.size(); i++) {
            if (i == 0) {
                if (N.get(i) != 0) {
                    hasil += 4;
                    hasil += (N.get(i) - 1) * 3;
                }
            } else {
                if (selisih < 0) {
                    hasil += (3 * (-selisih + 1));
                    hasil += (N.get(i) - (-selisih) - 1) * 2;
                } else {
                    hasil += 3;
                    hasil += 2 * (N.get(i) - 1);
                }
            }
            if (i < N.size() && i != N.size() - 1) {
                selisih = N.get(i) - N.get(i + 1);
            }
        }
        System.out.println("Jumlah garis : " + hasil);
    }
}