数学思维1-金字塔_pyramids数学-CSDN博客

本文链接：https://blog.csdn.net/weixin_43442778/article/details/123301317

题目描述

众所周知，金字塔是一层一层堆砌而成。

如下图，金字塔的最顶层有一个点，第二层有三个点，第三层有六个点，以此类推……

有一个无限大的金字塔，前 nn 层一共有多少个点？答案请对 10^9+7取模。

示例1

输入：
4
复制返回值：
20
复制说明：
前四层的点数量为 1 + 3 + 6 +10 = 20 

果果念

又是调了很长时间bug的一道题目，一开始看错题目了，第四层怎么也不知道怎么是10，原来是1+2+3+4=10。

一开始写的是循环模拟求解，emmm，超时了。看题解，直接就是一个公式，1+3+6+...+(n*(n+1)/2)=(n*(n+1)*(n+2)/6).

还有一点，就是取模的时候要注意。（a*b）%maxInt=(a%maxInt)*(b%maxInt).

代码

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