bzoj 3601: 一个人的数论 莫比乌斯反演+高斯消元

题目大意:
给定n=p1α1∗p2α1∗...∗pwαwn=p_1^{\alpha_1}*p_2^{\alpha_1}*...*p_w^{\alpha_w}n=p1α1p2α1...pwαwddd,求
∑i=1nid[gcd(i,n)==1]\sum_{i=1}^{n}i^d[gcd(i,n)==1]i=1nid[gcd(i,n)==1]
其中,d≤100d≤100d100w≤1000w≤1000w1000pip_ipi为质数,pi,qi≤109p_i,q_i≤10^9pi,qi109

分析:
根据题意推式子得到
ans=∑p∣nμ(p)∗pd∗∑i=1n/pidans=\sum_{p|n}\mu(p)*p^d*\sum_{i=1}^{n/p}i^dans=pnμ(p)pdi=1n/pid
对于最后面的自然数幂和,可以通过高斯消元求出多项式,设多项式为ad+1∗xd+1+ad∗xd+...+a0a_{d+1}*x^{d+1}+a_d*x^d+...+a_0ad+1xd+1+adxd+...+a0.
显然a0=0a_0=0a0=0,这一位就不用消了(避免出错)。
原式变成
ans=∑p∣nμ(p)∗pd∗∑i=1d+1ai∗(n/p)ians=\sum_{p|n}\mu(p)*p^d*\sum_{i=1}^{d+1}a_i*(n/p)^ians=pnμ(p)pdi=1d+1ai(n/p)i
iii移到前面,得到
ans=∑i=1d+1ai∑p∣nμ(p)∗pd∗(n/p)ians=\sum_{i=1}^{d+1}a_i\sum_{p|n}\mu(p)*p^d*(n/p)^ians=i=1d+1aipnμ(p)pd(n/p)i
我们可以枚举iii,当iii一定时,后面这些是一个积性函数,我们可以分别计算出每一个piαip_i^{\alpha_i}piαi的贡献,乘起来即可。而根据μ\muμ的性质,只有p=1p=1p=1p=pip=p_ip=pi时有贡献,这个可以O(1)O(1)O(1)算出。总复杂度是O(wd)O(wd)O(wd)的,另外高斯消元是O(n3)O(n^3)O(n3)

代码:

/**************************************************************
    Problem: 3601
    User: ypxrain
    Language: C++
    Result: Accepted
    Time:812 ms
    Memory:9240 kb
****************************************************************/
 
#include <iostream>
#include <cstdio>
#include <cmath>
#define LL long long
 
const int maxn=1007;
const LL mod=1e9+7;
 
using namespace std;
 
int n,k;
int p[maxn],q[maxn];
LL a[maxn][maxn],b[maxn],r[maxn];
LL sum,ans,num;
 
LL power(LL x,LL y)
{
    if (y==1) return x;
    if (y==0) return 1;
    LL c=power(x,y/2);
    c=(c*c)%mod;
    if (y&1) c=(c*x)%mod;
    return c;
}
 
void guass(int n)
{
    for (int i=1;i<=n;i++)
    {
        for (int j=i+1;j<=n;j++)
        {
            if (a[j][i]>a[i][i])
            {
                for (int k=i;k<=n;k++) swap(a[i][k],a[j][k]);
                swap(b[i],b[j]);
            }
        }
        LL inv=power(a[i][i],mod-2);
        for (int j=1;j<=n;j++)
        {
            if (i==j) continue;
            LL rate=a[j][i]*inv%mod;
            for (int k=i;k<=n;k++) a[j][k]=(a[j][k]+mod-a[i][k]*rate%mod)%mod;
            b[j]=(b[j]+mod-b[i]*rate%mod)%mod;
        }
    }
    for (int i=1;i<=n;i++) r[i]=b[i]*power(a[i][i],mod-2)%mod;
}
 
int main()
{
    scanf("%d%d",&k,&n);
    for (int i=1;i<=n;i++)
    {
        scanf("%d%d",&p[i],&q[i]);
        q[i]=power(p[i],q[i]);
    }   
    for (int i=1;i<=k+1;i++) a[i][1]=i;
    for (int j=2;j<=k+1;j++)
    {
        for (int i=1;i<=k+1;i++) a[i][j]=(a[i][j-1]*(LL)i)%mod;
    }
    for (int i=1;i<=k+1;i++)
    {
        sum=(sum+power(i,k))%mod;
        b[i]=sum;
    }
    guass(k+1);
    for (int i=1;i<=k+1;i++)
    {
        num=1;
        for (int j=1;j<=n;j++)
        {
            LL e=q[j]*power(p[j],mod-2)%mod;
            num=num*((power(q[j],i)+mod-power(p[j],k)*power(e,i)%mod)%mod)%mod;
        }
        ans=(ans+r[i]*num%mod)%mod;
    }
    printf("%lld\n",ans);
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值