【HDU6051】if the starlight never fade

题意：

给出m，P，求$sumlimits_{i=1}^{P-1}if(i)$，其中$f(i)=sumlimits_{x=1}^{P-1}sumlimits_{y=1}^{m}[(x+y)^i=(x^i)(mod P)]$

$1leq mleq P-1leq 10^9+6$

题解：

神仙数学题。。。

由于$P$是奇质数，所以肯定存在原根$g$，那么肯定唯一存在$a$，$b$满足$1leq a,bleq P-1$使得$x=g^a$，$y=g^b$，所以有：

$(x+y)^i≡x^i(mod P)$

$(g^{a}+g^{b})^i≡g^{ai}(mod P)$

$(1+g^{b-a})^i≡1(mod P)$

由于$1+g^{b-a}geq 2$，所以必定存在唯一的$k$满足$1leq kleq P-1$使得$g^k≡1+g^{b-a}(mod P)$；

显然$g^{ki}≡1(mod P)$，所以$ki≡0(mod P-1)$；

那么$k=a imesfrac{P-1}{gcd(P-1,i)}$，其中$a=1,2,3,...,gcd(P-1,i)-1$；

所以满足条件的$k$有$gcd(P-1,i)-1$个，固定$y$，则有$x≡y(g^{k}-1)^{-1}(mod P)$；

易知若$y(g^{k_1}-1)^{-1}≡y(g^{k_2}-1)^{-1}(mod P)$，则$g^{k_1}≡g^{k_2}(mod P)$，即$k_1=k_2$；

所以对于确定的$y$，$x$会有$gcd(P-1,i)-1$种不同取值。

至此式子化为$f(i)=m(gcd(P-1,i))$

然后大力推一波式子：

$sumlimits_{i=1}^{P-1}if(i)=msumlimits_{i=1}^{P-1}i((gcd(P-1),i)-1)=msumlimits_{i=1}^{P-1}i(gcd(P-1),i)-mfrac{P imes(P-1)}{2}$

把前半部分拿出来：

$msumlimits_{i=1}^{P-1}i(gcd(P-1),i)$

$=sumlimits_{d|(p-1)}dsumlimits_{d|i,1leq ileq P-1}i[gcd(P-1,i)=d]$

$=sumlimits_{d|(p-1)}d^2sumlimits_{i=1}^{frac{P-1}{d}}i[gcd(frac{P-1}{d},i)=1]$

$=sumlimits_{d|(p-1)}d^2frac{frac{P-1}{d}varphi(frac{P-1}{d})+[frac{P-1}{d}=1]}{2}$（引理）

这样复杂度就是$O(sqrt{n})$的，可以直接做。

引理：$sumlimits_{i=1}^{n}i[gcd(i,n)=1]=frac{1}{2}sumlimits_{i=1}^{n}(i+(n-i))[gcd(i,n)=1]=frac{1}{2}nsumlimits_{i=1}^{n}[gcd(i,n)=1]$

$=frac{nvarphi(n)+[n=1]}{2}$

证明：若$gcd(i,n)=1$，那么$gcd(n-i,n)=1$。

代码：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#define eps 1e-4
#define mod 1000000007
#define S(n) (ll)(n*(n+1)/2)
using namespace std;
typedef long long ll;
ll phi(int x){
    ll ret=x;
    for(ll i=2;i*i<=x;i++){
        if(!(x%i)){
            ret=ret/i*(i-1);
            while(!(x%i))x/=i;
        }
    }
    if(x>1)ret=ret/x*(x-1);
    return ret;
}
ll f(ll x,ll d){
    return d*d%mod*((x*phi(x)+(x==1))/2)%mod;
}
ll calc(ll x){
    ll ret=0;
    for(ll i=1;i*i<=x;i++){
        if(!(x%i)){
            ret=(ret+f(x/i,i))%mod;
            if(i*i!=x)ret=(ret+f(i,x/i))%mod;
        }
    }
    ret=(ret-x*(x+1)/2%mod+mod)%mod;
    return ret;
}
ll m,p;
int main(){
    scanf("%lld%lld",&m,&p);
    printf("%lld",calc(p-1)*m%mod);
    return 0;
}