結果

問題 No.2181 LRM Question 2
コンテスト
ユーザー kotatsugame
提出日時 2023-01-06 22:59:51
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,449 bytes
コンパイル時間 1,019 ms
コンパイル使用メモリ 85,508 KB
実行使用メモリ 7,680 KB
最終ジャッジ日時 2024-11-30 19:38:02
合計ジャッジ時間 2,347 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18 WA * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<vector>
#include<atcoder/math>
#include<atcoder/modint>
using namespace std;
using mint=atcoder::modint;
mint F[1<<20];
pair<long,mint>f(long N,int p)
{
	pair<long,mint>ret=make_pair(0L,mint(1));
	while(N>=1)
	{
		long q=N/p,r=N%p;
		ret.second*=F[p-1].pow(q)*F[r];
		ret.first+=N=q;
	}
	return ret;
}
pair<long,mint>mul(pair<long,mint>ret,long v,int p)
{
	while(v%p==0)v/=p,ret.first++;
	ret.second*=v;
	return ret;
}
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)
	{
		int p=fp.first,mod=1;
		mint::set_mod(p);
		for(int i=0;i<fp.second;i++)mod*=p;
		m.push_back(mod);
		F[0]=1;
		for(int i=1;i<p;i++)F[i]=F[i-1]*i;
		int now=0;
		pair<long,mint>x=f(2*L,p),y=f(L,p);
		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,p),2*n,p);
				y=mul(y,n,p);
			}
			int tmp=(x.second/y.second.pow(2)).val();
			assert(x.first-y.first*2>=0);
			if(x.first-y.first*2>=fp.second)tmp=0;
			else
			{
				for(int i=0;i<x.first-y.first*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;
}
0