（逆元）序列求和·牛客(https://ac.nowcoder.com/acm/contest/26908/1060)

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 131072K，其他语言262144K
64bit IO Format: %lld

题目描述

定义S(n) = 12 + 22 + … + n2，输出S(n) % 1000000007。

注意：1 < n < 1e18。

输入描述:

多组输入，输入直到遇到EOF为止；

第一行输入一个正整数n。

输出描述:

输出S(n) % 1000000007的结果。

示例1

输入

复制1 2 1000

1
2
1000

输出

复制1 5 333833500

1
5
333833500

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll mod=1000000007;

ll inv(ll a,ll n)
{
    ll ans=1;
    while(n)
    {
        if(n&1)ans=ans*a%mod;
        a=a*a%mod;
        n>>=1;
    }
    return ans;
}

int main()
{
    ll n,sum,ny;
    ny=inv(6,mod-2);
    while(scanf("%lld",&n)!=EOF)
    {
        n%=mod;//将n的值变小，防止数据爆炸
        sum=n%mod*(n+1)%mod*(2*n+1)%mod*ny%mod;
        cout<<sum<<endl;
    }
}