Pyramid 【Gym - 101981G 2018南京区域赛 G题】【规律+逆元】

题目链接

题目大意

问这样的一个n行的三角形塔中，有多少个正三角形，斜着的也算

以上两种三角形都算

解题思路

比赛的时候手画的三角形，然后画漏了，一直没推出来
记着：以后用电脑打表比较准确（把点丢进去，然后n³跑一遍判断）
然后得到前几项：1,5,15,35,70, 126,210… …
第一次做差：4,10,20,35 ,56, 84
第二次做差：6 10 15 21 28
第三次做差：4,5,6,7
第四次做差：1,1,1,1
所以这是一个四阶的方程式（相当于求导四次是一个常数），那么递推公式应该是：
f(n) = a *n⁴ + b *n³ + c *n² + d *n + e
已知前五项，得五个方程求出a，b，c，d，e
最后算出来是：a=1/24 , b=6/24 , c=11/24 , d=6/24,e=0
所以递推公式为：f(n) = (n⁴ + 6n³ + 11n² +6n) / 24
因为有取模，所以除24等于乘上24逆元，直接O(1)算答案就好了

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
using namespace std;

long long mod=1e9+7;

long long inv(long long x)
{
    int k=mod-2;
    long long ans=1;
    while(k)
    {
        if (k&1) ans=(long long)ans*x%mod;
        x=(long long)x*x%mod;
        k>>=1;
    }
    return ans;
}
int main()
{
    int T;
    long long n;
    long long a=inv(24);//24的逆元
    scanf("%d",&T);
    while(T--)
    {
        scanf("%lld",&n);
        long long ans=1;
        for(int i=0;i<4;i++)//n的4次方
            ans=(ans*n)%mod;
        long long tmp=(long long)6;//tmp=6*n*n*n
        for(int i=0;i<3;i++)
            tmp=(tmp*n)%mod;
        ans=(ans+tmp)%mod;
        tmp=(((long long)11*n)%mod*n)%mod;//tmp=11*n*n
        ans=(ans+tmp)%mod;
        tmp=((long long)6*n)%mod;//tmp=6*n
        ans=(ans+tmp)%mod;
        ans=(ans*a)%mod;
        printf("%lld\n",ans);
    }
    return 0;
}