結果

問題 No.2181 LRM Question 2
ユーザー kotatsugame
提出日時 2023-01-06 23:31:11
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 54 ms / 2,000 ms
コード長 1,430 bytes
コンパイル時間 974 ms
コンパイル使用メモリ 85,544 KB
実行使用メモリ 7,680 KB
最終ジャッジ日時 2024-11-30 20:25:56
合計ジャッジ時間 2,310 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include<iostream>
#include<vector>
#include<atcoder/math>
#include<atcoder/modint>
using namespace std;
using mint=atcoder::modint;
int P;
struct val{
    long e;
    mint v;
    val(long n)
    {
        e=0;
        while(n%P==0)n/=P,e++;
        v=n;
    }
};//v*p**e
mint F[1<<20];
val f(long N)
{
    val ret(1);
    int PP=mint::mod();
    while(N>=1)
    {
        long q=N/P;
        ret.v*=F[PP-1].pow(N/PP)*F[N%PP];
        ret.e+=N=q;
    }
    return ret;
}
val mul(val a,val b)
{
    a.e+=b.e;
    a.v*=b.v;
    return a;
}
int main()
{
    long L,R;int M;
    cin>>L>>R>>M;
    vector<pair<int,int> >fac;
    {
        int m=M;
        for(int i=2;i*i<=m;i++)if(m%i==0)
        {
            fac.push_back(make_pair(i,0));
            while(m%i==0)m/=i,fac.back().second++;
        }
        if(m>1)fac.push_back(make_pair(m,1));
    }
    vector<long long>r,m;
    for(pair<int,int>fp:fac)
    {
        P=fp.first;
        int mod=1;
        for(int i=0;i<fp.second;i++)mod*=P;
        mint::set_mod(mod);
        m.push_back(mod);
        F[0]=1;
        for(int i=1;i<mod;i++)
        {
            F[i]=F[i-1];
            if(i%P!=0)F[i]*=i;
        }
        int now=0;
        val x=f(2*L),y=f(L);
        for(long n=L;n<=R;n++)
        {
            if(n>L)
            {
                //x <- 2*n-1,2*n
                //y <- n
                x=mul(mul(x,2*n-1),2*n);
                y=mul(y,n);
            }
            assert(x.e-y.e*2>=0);
            if(x.e-y.e*2>=fp.second)continue;
            int tmp=(x.v/y.v.pow(2)).val();
            for(int i=0;i<x.e-y.e*2;i++)tmp=(long)tmp*P%mod;
            now=(now+tmp)%mod;
        }
        r.push_back(now);
    }
    pair<long long,long long>ret=atcoder::crt(r,m);
    assert(ret.second==M);
    cout<<(ret.first-(R-L+1)*2%M+M)%M<<endl;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0