結果

問題 No.2791 Beginner Contest
ユーザー aogeraaogera
提出日時 2024-07-07 16:32:42
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 2,533 bytes
コンパイル時間 6,959 ms
コンパイル使用メモリ 350,212 KB
実行使用メモリ 9,572 KB
最終ジャッジ日時 2024-07-07 16:32:52
合計ジャッジ時間 7,885 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 AC 15 ms
9,408 KB
testcase_02 AC 16 ms
9,520 KB
testcase_03 WA -
testcase_04 AC 15 ms
9,364 KB
testcase_05 AC 16 ms
9,416 KB
testcase_06 AC 15 ms
9,572 KB
testcase_07 AC 15 ms
9,404 KB
testcase_08 AC 14 ms
9,476 KB
testcase_09 AC 18 ms
9,344 KB
testcase_10 AC 15 ms
9,460 KB
testcase_11 AC 15 ms
9,424 KB
testcase_12 AC 14 ms
9,536 KB
testcase_13 AC 14 ms
9,460 KB
testcase_14 AC 14 ms
9,492 KB
testcase_15 AC 15 ms
9,532 KB
testcase_16 AC 16 ms
9,396 KB
testcase_17 AC 15 ms
9,520 KB
testcase_18 AC 15 ms
9,412 KB
testcase_19 AC 15 ms
9,512 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include<bits/stdc++.h>
#include <atcoder/all>
using ull = unsigned long long;
using namespace std;
using namespace atcoder;
using vst = vector<string>;
using ll = long long;
using ld = long double;
using P = pair<ll,ll>;
using vl = vector<ll>;
using vvl = vector<vl>;
using vP = vector<P>;
#define rep(i, n) for (ll i = 0; i < n; i++)
#define repp(i,k,n) for (ll i = k; i < n; i++)
#define per(i,s,e) for(ll i = s; i >= e; i--)
#define all(v) v.begin(),v.end()
#define yesno(a) a ? cout << "Yes" << '\n' : cout << "No" << '\n'
#define YESNO(a) a ? cout << "YES" << '\n' : cout << "NO" << '\n'
#define UNOmap unordered_map
#define UNOset unordered_set
#define chmax(a,b) a=max(a,b)
#define chmin(a,b) a=min(a,b)
#define debug(x) cerr << #x << " = " << x << '\n'

template<class... T>void in(T&... a){(cin >> ... >> a);}
template<class T, class... Ts>void out(const T& a, const Ts&... b){cout << a;((cout << ' ' <<  b), ...);cout << '\n';}
template<class T> void vin2(vector<T> &u,vector<T> &v){for(ll i = 0; i < (ll)v.size(); i++) in(u[i],v[i]);}
template<class T> void vin(vector<T> &v){for(ll i = 0; i < (ll)v.size(); i++)in(v[i]);}
template<class T> void vout(vector<T> &v){for(ll i = 0; i < (ll)v.size(); i++) cout << v[i] << ' ';cout << "\n";}

ll INF = 1152921504606846976;ll MOD =998244353; ll MOD1 =1000000007;
/* INF = 1 << 60 */

#define sl(...) ll __VA_ARGS__; in(__VA_ARGS__)
 
using mint = modint998244353;
using mint1 = modint1000000007;
using mintn = modint;
using vm = vector<mint>;
using vvm = vector<vm>;

vl dx = {0,0,1,-1}, dy = {1,-1,0,0};
//----------------------------------------------

const int MAX = 510000;
mint fac[MAX], finv[MAX], inv[MAX];

// テーブルを作る前処理
void COMinit() {
    fac[0] = fac[1] = 1;
    finv[0] = finv[1] = 1;
    inv[1] = 1;
    for (int i = 2; i < MAX; i++){
        fac[i] = fac[i - 1] * i;
        inv[i] = MOD - inv[MOD%i] * (MOD / i);
        finv[i] = finv[i - 1] * inv[i];
    }
}

// 二項係数計算
mint COM(int n, int k){
    if (n < k) return 0;
    if (n < 0 || k < 0) return 0;
    return fac[n] * finv[k] * finv[n - k];
}

int main(){
  ios::sync_with_stdio(false);cin.tie(nullptr);cout<<fixed<<setprecision(15);
//==============================================
  
  COMinit();
  // out(COM(30,2).val());
  
  ll N,K;
  in(N,K);
  mint ans = 0;
  rep(i,100009){
    if(N-K*i <= 0) break;
    ans += COM(N-K*i+i,i);
    // out(N-K*i,i);
  }
  out(ans.val());
}
0