第七届蓝桥杯Java A——煤球数目

有一堆煤球，堆成三角棱锥形。具体：
第一层放1个，
第二层3个（排列成三角形），
第三层6个（排列成三角形），
第四层10个（排列成三角形），
....
如果一共有100层，共有多少个煤球？

规律：第$i$层的煤球数等于第$i-1$层煤球数量+$i$

public class Main {
    public static void main(String[] args) {
        int[] arr = new int[101];
        arr[4] = 10;
        int sum = 0;
        for (int i = 5; i <= 100; i++) {
            arr[i] = arr[i - 1] + i;
            sum += arr[i];
        }
        System.out.println(sum + 10 + 6 + 3 + 1);
    }
}