AtCoder Product of Arithmetic Progression

/**
链接:https://atcoder.jp/contests/m-solutions2019/tasks/m_solutions2019_e
题意:给出等差数列的首项x,公差d,项数n。求该等差数列的前n项积;
分析:考虑从模数为1e6+3出发,很显然算法复杂度是一个多log的或者更优的写法。
    tag1: d=0时,ans = ksm(x,n);
    tag2: d!=0 时,则:
    ans = x*(x+d)*(x+2*d)*(x+3*d)*...*(x+(n-1)*d)
        = (x/d)*(x/d+1)*(x/d+2)*(x/d+3)*...*(x/d+(n-1))*ksm(d,n)
        由于mod==1e6+3,因此当 x/d + (n-1) >= mod时,ans=0;
        其它情况可以直接预处理出逆元和阶乘直接询问即可;
*/

#include<bits/stdc++.h>
#define ll long long
using namespace std;

const ll mod=1000003;

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


const int maxn=2e6+70;
ll fac[maxn],inv[maxn],finv[maxn];//阶乘,逆元,阶乘逆元
void init(){
    inv[0]=inv[1]=fac[0]=fac[1]=finv[0]=finv[1]=1;
    for(int i=2;i<maxn;i++) {
        fac[i]=1ll*fac[i-1]*i%mod;
        inv[i]=mod-inv[mod%i]*(mod/i)%mod;
        finv[i]=finv[i-1]*inv[i]%mod;
    }
}

void solved(int x,int d,int n){
    if(d==0) {
        cout<<ksm(x*1ll,n*1ll)<<endl;
        return ;//公差为0的情况
    }
    ll zx=1ll*x*inv[d]%mod;
    if(zx+n-1>=mod) {
        cout<<0<<endl;//x/d+(n-1) >=mod 的情况 
        return ;
    }
    ll ret=ksm(d*1ll,n);
    ret=ret*fac[zx+n-1]%mod;
    ret=ret*finv[zx-1]%mod;
    cout<<ret<<endl;
    return ;//正常的情况.
}

int main (){
    init();
    int Q;scanf("%d",&Q);
    int x,d,n;
    while(Q--) {
        scanf("%d %d %d",&x,&d,&n);
        solved(x,d,n);
    }
    return 0;
}

 

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值