結果

問題 No.1414 東大文系数学2021第2問改
ユーザー evimaevima
提出日時 2021-02-23 09:52:04
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 5,231 bytes
コンパイル時間 5,001 ms
コンパイル使用メモリ 278,536 KB
実行使用メモリ 429,544 KB
最終ジャッジ日時 2024-10-02 07:48:46
合計ジャッジ時間 10,527 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 301 ms
429,544 KB
testcase_01 AC 319 ms
429,440 KB
testcase_02 TLE -
testcase_03 -- -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
権限があれば一括ダウンロードができます

ソースコード

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(int 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(int n, int r){
    return n < 0 || r < 0 || r > n ? 0 : fact[n] * ifact[r] * ifact[n - r];
}

ll N, M, K, dp[301][301][301][2];

ll f(int cur, int num, int k, int flag){
    ll& ret = dp[cur][num][k][flag];
    if(ret != -1) return ret;
    ret = 0;
    if(cur == N) ret = num == M && flag;
    else{
        ret += f(cur + 1, num, 0, flag);
        if(num < M){
            ret += f(cur + 1, num + 1, min(K, k + 1LL), flag || k + 1 == K);
        }
    }
    return ret %= MOD;
}

void solveOne(){
    cin >> N >> M >> K;
    memset(dp, -1, sizeof(dp));
    print(f(0, 0, 0, 0));
}

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