結果

問題 No.1970 ひよこ鑑定士
ユーザー 👑 potato167potato167
提出日時 2022-06-04 03:53:35
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,440 bytes
コンパイル時間 2,176 ms
コンパイル使用メモリ 200,772 KB
最終ジャッジ日時 2025-01-29 18:01:44
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 WA * 1
other AC * 17 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#pragma GCC optimize("Ofast")
#define _GLIBCXX_DEBUG
using namespace std;
using ll=long long;
const ll mod=998244353;
#define rep(i,a) for (ll i=0;i<a;i++)


namespace po167{
struct combination{
	int upper;
	long long MOD;
	std::vector<long long> fact;
	std::vector<long long> rev;
	std::vector<long long> fact_rev;
	combination(int max,long long mod):upper(max),MOD(mod),fact(max+1),rev(max+1),fact_rev(max+1){
		for(long long i=0;i<=max;i++){
			if(i<2){
				fact[i]=1;
				fact_rev[i]=1;
				rev[i]=1;
				continue;
			}
			fact[i]=(fact[i-1]*i)%mod;
			rev[i]=mod-((mod/i)*rev[mod%i])%mod;
			fact_rev[i]=(fact_rev[i-1]*rev[i])%mod;
		}
	}
	long long Comb(int x,int y){
		assert(upper>=x);
		if (x<y||y<0||x<0) return 0;
		return (((fact_rev[y]*fact_rev[x-y])%MOD)*fact[x])%MOD;
	}
};
}
using po167::combination;

void solve();
// oddloop
int main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	
	int t=1;
	//cin>>t;
	rep(i,t) solve();
}

void solve(){
	ll N,K;
	cin>>N>>K;
	combination table(N*3,mod);
	vector<ll> sum(2*N+1,1);
	rep(i,N*2){
		sum[i+1]=(sum[i]+table.Comb(N*2,i+1))%mod;
	}
	auto f=[&](ll L)->ll{
		if(L<=1) return 0;
		ll ans=(L*table.Comb(N*2,N))%mod;
		for(int i=1;i<=N;i++){
			if(i%L==0){
				ans+=(2ll*(L-1)*table.Comb(2*N,N-i))%mod;
			}else{
				ans-=2ll*table.Comb(2*N,N-i);
			}
		}
		ans=(ans+mod)%mod;
		return ans;
	};
	cout<<(table.Comb(N*2,N)-f(K+1)+f(K)+mod)%mod<<"\n";
}
0