逆元+公式推导 zoj3903 Ant

传送门：点击打开链接

题意：一个立方体，已知立方体最长边长为n，一只蚂蚁从一个顶点走到对角线的另一个顶点，且走的是最短的路线，记路线长度为L。那么所有满足题意的情况的L*L之和是多少

思路：再去推导一下就发现，其实就是L^2=n^2+(a+b)^2,且a<=b<=n，把所有的情况累加起来就做完了

由于n太大了，所以只能是数学的角度推导公式

(a+b)^2拆成a^2+b^2+2*a*b

剩下的大概就是把每一部分都累加起来，然后答案就出来了，首先需要知道2个累加的公式

1^2+2^2+3^2+....n^2=n*(n+1)*(2*n+1)/6

1^3+2^3+3^3+....n^3=(n*(n+1)/2)^2

然后最后式子里可能有除法，利用费马小定理求出逆元就解决了

#include<map>
#include<set>
#include<cmath>
#include<ctime>
#include<stack>
#include<queue>
#include<cstdio>
#include<cctype>
#include<string>
#include<vector>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<functional>
#define fuck printf("fuck")
#define FIN freopen("input.in","r",stdin)
#define FOUT freopen("output.txt","w+",stdout)
using namespace std;
typedef long long LL;

const int MX = 1e6 + 5;
const int mod = 1e9 + 7;

LL power(LL a, LL b) {
    LL ret = 1;
    while(b) {
        if(b & 1) ret = (ret * a) % mod;
        a = (a * a) % mod;
        b >>= 1;
    }
    return ret;
}


int main() {
    int T; //FIN;
    scanf("%d", &T);
    while(T--) {
        LL n;
        scanf("%lld", &n); n %= mod;
        LL ans = n * n % mod * n % mod * (n + 1) % mod * power(2, mod - 2) % mod;
        ans = (ans + n * (n + 1) % mod * (n + 2) % mod * (2 * n + 1) % mod * power(6, mod - 2) % mod) % mod;
        ans = (ans + n * n % mod * (n + 1) % mod * (n + 1) % mod * power(4, mod - 2) % mod) % mod;
        printf("%lld\n", ans);
    }
    return 0;
}