牛客练习数学题序列求和

最新推荐文章于 2021-06-21 15:54:00 发布

二货RK

最新推荐文章于 2021-06-21 15:54:00 发布

阅读量288

点赞数

分类专栏： C++ 算法数学牛客文章标签：序列求和牛客数学题

本文链接：https://blog.csdn.net/qq_33437973/article/details/99890689

版权

C++ 同时被 3 个专栏收录

40 篇文章 1 订阅

订阅专栏

算法

26 篇文章 0 订阅

订阅专栏

牛客

8 篇文章 0 订阅

订阅专栏

牛客练习数学题序列求和

题目链接：https://ac.nowcoder.com/acm/problem/15950?tdsourcetag=s_pctim_aiomsg&headNav=acm

我是真的好久没写数学题了。。。智商已下线。。

今天必须把这道题拿出来好好补点智商税

这题在高中的时候我估计还能写得出来，因为知道这道题怎么做之后，我脑海里清晰的记得自己将好多数列的公式都写在笔记本的一个位置。。。并且还有这道题的答案。。。是我太菜了

既然把公式忘了，那我就再来重新推一遍

题面可以得出我们所求的，是数列 1²⁺²2+3^2+……+n2之和

我们知道 a³-b³=(a-b)(a²+a*b+b²)

由此推得 n³-(n-1)³ = n²+n(n-1)+(n-1)² = 3n²+3n+1

进而可知 n³=(n³-(n-1)³)+

((n-1)³-(n-2)³)+

……+(2³-1³)+1

最终可得 n³=3×(1²+2²+3²+……+n²)-3×(1+2+3+……+n)+n

经转化后即为 S(n)=(2n³+3n²+n)/6 == n(n+1)(2n+1)/6

那么程序为

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int mod=1e9+7;
ll n;
ll power(ll x,ll y){
    ll ans=1;
    while(y){
        if(y&1)
            ans=(ans*x)%mod;
            y>>=1;
        x=(x*x)%mod;
    }
return ans;
}
int main(){
    while(~scanf("%lld",&n)){
        n%=mod;
        ll zans=n*(n+1)%mod*(2*n+1)%mod*power(6,mod-2)%mod;
        printf("%lld\n",zans);
    }
return 0;
}