题目描述
在一个 N×N 的点阵上,取其中 4 个点恰好组成一个正方形的 4 个顶点,一共有多少种不同的取法?
由于结果可能非常大,你只需要输出模 1e9+7 的余数。
证明过程是有点复杂的,这里就贴个链接吧
证明过程
注意一下long long就好了
#pragma GCC optimize(2)
#include<bits/stdc++.h>
using namespace std;
const int mod = 1e9+7;
typedef long long ll;
int main(){
int n;
cin>>n;
ll ans = 0;
for(int i=1;i<=n;i++){
ans=(ans+(ll)i*i*i-(ll)2*n*i*i+(ll)n*n*i)%mod;
}
cout<<ans<<endl;
return 0;
}