結果

問題 No.1414 東大文系数学2021第2問改
ユーザー evimaevima
提出日時 2021-02-23 09:49:57
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 235 ms / 2,000 ms
コード長 4,971 bytes
コンパイル時間 4,932 ms
コンパイル使用メモリ 269,560 KB
実行使用メモリ 120,696 KB
最終ジャッジ日時 2024-04-10 06:09:21
合計ジャッジ時間 9,939 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 192 ms
120,388 KB
testcase_03 AC 2 ms
6,940 KB
testcase_04 AC 235 ms
120,380 KB
testcase_05 AC 141 ms
120,540 KB
testcase_06 AC 139 ms
120,388 KB
testcase_07 AC 2 ms
6,944 KB
testcase_08 AC 2 ms
6,940 KB
testcase_09 AC 233 ms
120,496 KB
testcase_10 AC 141 ms
120,404 KB
testcase_11 AC 141 ms
120,440 KB
testcase_12 AC 142 ms
120,476 KB
testcase_13 AC 141 ms
120,392 KB
testcase_14 AC 198 ms
120,448 KB
testcase_15 AC 186 ms
120,468 KB
testcase_16 AC 187 ms
120,436 KB
testcase_17 AC 188 ms
120,604 KB
testcase_18 AC 187 ms
120,460 KB
testcase_19 AC 197 ms
120,544 KB
testcase_20 AC 197 ms
120,696 KB
testcase_21 AC 221 ms
119,096 KB
testcase_22 AC 2 ms
6,940 KB
testcase_23 AC 167 ms
120,460 KB
testcase_24 AC 150 ms
120,412 KB
testcase_25 AC 234 ms
120,604 KB
testcase_26 AC 234 ms
120,496 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

// Enjoy your stay. Code by evima

#include <atcoder/convolution>
#include <atcoder/dsu>
#include <atcoder/fenwicktree>
#include <atcoder/lazysegtree>
#include <atcoder/math>
#include <atcoder/maxflow>
#include <atcoder/mincostflow>
#include <atcoder/modint>
#include <atcoder/scc>
#include <atcoder/segtree>
#include <atcoder/string>
#include <atcoder/twosat>
using namespace atcoder;
using mint = modint998244353;// modint1000000007;// modint;

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

using ld = long double;
using ll = long long;
using lll = __int128;
using vl = vector<ll>;
using vs = vector<string>;
using LOOPVAR_TYPE = ll;

#define all(x) (x).begin(), (x).end()
#define sq(x) ((x) * (x))
#define sz(x) ll((x).size())
#define GET_MACRO(_1, _2, _3, NAME, ...) NAME
#define rep1(i, n) rep2(i, 0, n)
#define rep2(i, a, b) for(LOOPVAR_TYPE i = LOOPVAR_TYPE(a); i < LOOPVAR_TYPE(b); i++)
#define rep(...) GET_MACRO(__VA_ARGS__, rep2, rep1)(__VA_ARGS__)

#define eb emplace_back
#define fir first
#define sec second
#define mp make_pair
#define mt make_tuple

const ld EPS = 1e-9;
const ld PI = 3.14159265358979323846L;
const ll INF = 1070000000LL;
const ll MOD = 998244353LL;// 1000000007LL;

void fast_io() { cin.tie(NULL); ios_base::sync_with_stdio(false); }
ll lin() { ll x; cin >> x; return x; }
string input() { string s; cin >> s; return s; }
vl vin(int n) { vector<ll> v(n); rep(i, n) cin >> v[i]; return v; }
template<typename T> bool chmin(T& a, const T& b) { return (b < a) ? (a = b, true) : false; }
template<typename T> bool chmax(T& a, const T& b) { return (a < b) ? (a = b, true) : false; }
template<typename T> map<T, ll> freq(const vector<T>& v){ map<T, ll> ret; for(T x: v) ++ret[x]; return ret; }
template<typename T> vector<T> reversed(vector<T> v) { reverse(all(v)); return v; }
template<typename T> vector<T> sorted(vector<T> v) { sort(all(v)); return v; }
template<typename T> vector<T> sub(const vector<T>& v, int from, int to) { vector<T> ret; copy(&v[from], &v[to], back_inserter(ret)); return ret; }
template<typename T> string str(const T& x) { stringstream ss; ss << x; return ss.str(); }
template<typename Tuple, size_t... I> array<int, sizeof...(I)> str_tuple_impl(stringstream& ss, Tuple&& t, index_sequence<I...>){ return {{ (void( ss << str(get<I>(t)) << " " ), 0)... }}; }
template<typename Tuple> string str_tuple(Tuple&& t){ stringstream ss; str_tuple_impl(ss, forward<Tuple>(t), make_index_sequence<tuple_size<decay_t<Tuple>>::value>{}); string s = ss.str(); if(s.size() > 0) s.pop_back(); return s; }
template<typename T, typename U> string str(const pair<T, U>& p) { return str_tuple(p); }
template<typename... T> string str(const tuple<T...>& t) { return str_tuple(t); }
template<typename T> string str(const vector<T>& v) { stringstream ss; rep(i, sz(v)) ss << v[i] << (i < sz(v) - 1 ? " " : ""); return ss.str(); }
template<typename T> void print1(T&& x, const string& end) { cout << str(x) << end; }
void print() { print1("", "\n"); }
template<typename T, typename... U> void print(T&& head, U&&... tail) { print1(head, " "); print(forward<U>(tail)...); }
template<typename T> void eprint1(T&& x, const string& end) { cerr << str(x) << end; }
void eprint() { eprint1("", "\n"); }
template<typename T, typename... U> void eprint(T&& head, U&&... tail) { eprint1(head, " "); eprint(forward<U>(tail)...); }
template<typename... T> void quit(T&&... x) { print(forward<T>(x)...); exit(0); }
template<typename T, typename U> void yn(bool cnd, T&& yes = "Yes", U&& no = "No") { if(cnd) print(yes); else print(no); }
void zip(vl& v) { vl w = v; sort(all(w)); int n = unique(all(w)) - w.begin(); for(ll& x: v) x = lower_bound(&w[0], &w[n], x) - &w[0]; }
vl zipped(vl v) { zip(v); return v; }
map<ll, ll> zipmap(vl v) { map<ll, ll> ret; sort(all(v)); v.erase(unique(all(v)), v.end()); rep(i, sz(v)) ret[v[i]] = i; return ret; }

vector<mint> fact, ifact, inv;

void init_fact_ifact_inv(ll n){
    fact.resize(n + 1);
    ifact.resize(n + 1);
    inv.resize(n + 1);
    fact[0] = 1;
    rep(i, n){
        fact[i + 1] = fact[i] * (i + 1);
    }
    ifact[n] = fact[n].inv();
    for(int i = n - 1; i >= 0; --i){
        ifact[i] = ifact[i + 1] * (i + 1);
    }
    rep(i, n){
        inv[i + 1] = fact[i] * ifact[i + 1];
    }
}

const mint ncr(ll n, ll r){
    return n < 0 || r < 0 || r > n ? 0 : fact[n] * ifact[r] * ifact[n - r];
}

void solveOne(){
    ll N, M, K, L;
    cin >> N >> M >> K;
    L = N - M;
    init_fact_ifact_inv(N);
    mint ans = 0;
    rep(i, 1, L + 2){
        ans += (i % 2 ? 1 : -1) * ncr(L + 1, i) * ncr(N - i * K, L);
    }
    print(ans.val());
}

int main(){
    fast_io();
    cout << setprecision(20);
    
    int num_tc = 1;
    // cin >> num_tc;
    
    
    
    rep(tc, 1, num_tc + 1){
        // cout << "Case #" << tc << ": " ;// << endl;
        solveOne();
    }
}
0