题解 | #金字塔#

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param n int整型 金字塔的层数
     * @return int整型
     */
    public int getNums (int n) {
        // write code here
        long a = (long) n *(n+1);
        a = a % 1000000007 * (n+2) / 6;
        return (int) (a % 1000000007);
    }
}