結果

問題 No.2514 Twelvefold Way Returns
ユーザー 👑 NachiaNachia
提出日時 2023-10-20 23:34:03
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 66 ms / 3,000 ms
コード長 3,283 bytes
コンパイル時間 949 ms
コンパイル使用メモリ 85,500 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-09-20 23:32:52
合計ジャッジ時間 3,170 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 3 ms
6,944 KB
testcase_02 AC 13 ms
6,940 KB
testcase_03 AC 4 ms
6,940 KB
testcase_04 AC 52 ms
6,940 KB
testcase_05 AC 47 ms
6,944 KB
testcase_06 AC 38 ms
6,940 KB
testcase_07 AC 2 ms
6,944 KB
testcase_08 AC 28 ms
6,944 KB
testcase_09 AC 29 ms
6,944 KB
testcase_10 AC 59 ms
6,940 KB
testcase_11 AC 5 ms
6,944 KB
testcase_12 AC 15 ms
6,944 KB
testcase_13 AC 2 ms
6,940 KB
testcase_14 AC 12 ms
6,940 KB
testcase_15 AC 25 ms
6,940 KB
testcase_16 AC 60 ms
6,940 KB
testcase_17 AC 35 ms
6,940 KB
testcase_18 AC 59 ms
6,940 KB
testcase_19 AC 62 ms
6,944 KB
testcase_20 AC 66 ms
6,944 KB
testcase_21 AC 62 ms
6,944 KB
testcase_22 AC 62 ms
6,944 KB
testcase_23 AC 21 ms
6,940 KB
testcase_24 AC 63 ms
6,940 KB
testcase_25 AC 30 ms
6,940 KB
testcase_26 AC 2 ms
6,948 KB
testcase_27 AC 63 ms
6,940 KB
testcase_28 AC 31 ms
6,940 KB
testcase_29 AC 26 ms
6,940 KB
testcase_30 AC 27 ms
6,944 KB
testcase_31 AC 32 ms
6,944 KB
testcase_32 AC 16 ms
6,940 KB
testcase_33 AC 1 ms
6,940 KB
testcase_34 AC 11 ms
6,940 KB
testcase_35 AC 50 ms
6,944 KB
testcase_36 AC 1 ms
6,944 KB
testcase_37 AC 2 ms
6,944 KB
testcase_38 AC 60 ms
6,944 KB
testcase_39 AC 66 ms
6,944 KB
testcase_40 AC 63 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#line 1 "..\\Main.cpp"
#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <atcoder/modint>
#line 3 "D:\\Programming\\VSCode\\competitive-cpp\\nachia\\math\\combination.hpp"

namespace nachia{

template<class Modint>
class Comb{
private:
    static constexpr int MOD = Modint::mod();
    std::vector<Modint> F;
    std::vector<Modint> iF;
public:
    void extend(int newN){
        int prevN = (int)F.size() - 1;
        if(newN >= MOD) newN = MOD - 1;
        if(prevN >= newN) return;
        F.resize(newN+1);
        iF.resize(newN+1);
        for(int i=prevN+1; i<=newN; i++) F[i] = F[i-1] * Modint::raw(i);
        iF[newN] = F[newN].inv();
        for(int i=newN; i>prevN; i--) iF[i-1] = iF[i] * Modint::raw(i);
    }
    Comb(int n = 1){
        F.assign(2, Modint(1));
        iF.assign(2, Modint(1));
        extend(n);
    }
    Modint factorial(int n) const { return F[n]; }
    Modint invFactorial(int n) const { return iF[n]; }
    Modint invOf(int n) const { return iF[n] * F[n-1]; }
    Modint comb(int n, int r) const {
        if(n < 0 || n < r || r < 0) return Modint(0);
        return F[n] * iF[r] * iF[n-r];
    }
    Modint invComb(int n, int r) const {
        if(n < 0 || n < r || r < 0) return Modint(0);
        return iF[n] * F[r] * F[n-r];
    }
    Modint perm(int n, int r) const {
        if(n < 0 || n < r || r < 0) return Modint(0);
        return F[n] * iF[n-r];
    }
    Modint invPerm(int n, int r) const {
        if(n < 0 || n < r || r < 0) return Modint(0);
        return iF[n] * F[n-r];
    }
    Modint operator()(int n, int r) const { return comb(n,r); }
};

} // namespace nachia
#line 7 "..\\Main.cpp"
using Modint = atcoder::static_modint<998244353>;
using namespace std;
using i64 = long long;
using u64 = unsigned long long;
#define rep(i,n) for(int i=0; i<(int)(n); i++)
const i64 INF = 1001001001001001001;

struct F{
    Modint a,b,c;
};
F operator+(F l, F r){ return { l.a+r.a, l.b+r.b, l.c+r.c }; }
F operator*(F l, F r){ return { l.a*r.a + l.b*r.c + l.c*r.b, l.a*r.b + l.b*r.a + l.c*r.c, l.a*r.c + l.b*r.b + l.c*r.a }; }
F Omega(i64 t){
    t %= 3; t += 3; t %= 3;
    if(t == 0) return { 1, 0, 0 };
    if(t == 1) return { 0, 1, 0 };
    return { 0, 0, 1 };
}

F powMod(F a, i64 i){
    if(i == 0) return {1,0,0};
    F f = powMod(a*a, i/2);
    if(i%2 == 1) f = f * a;
    return f;
}

ostream& operator<<(ostream& ostr, F f){ return ostr << "(" << f.a.val() << ' ' << f.b.val() << ' ' << f.c.val() << ")"; }

void testcase(){
    auto comb = nachia::Comb<Modint>(1000);
    i64 N, M; cin >> N >> M;
    F Q = {0,0,0};
    for(i64 a=0; a<=M; a++) for(i64 b=0; a+b<=M; b++){
        i64 c = M - a - b;
        F ex = { a, b, c };
        //cout << "ex = " << ex << endl;
        ex = powMod(ex, N);
        //cout << "pow ex = " << ex << endl;
        F t = F{ comb.factorial(M) / comb.factorial(a) / comb.factorial(b) / comb.factorial(c), 0, 0 };
        Q = Q + ex * t * Omega(b+b+c);
    }
    //cout << Q.a.val() << " " << Q.b.val() << " " << Q.c.val() << endl;
    Modint ans = (Q.a - (Q.b + Q.c) / 2) / Modint(3).pow(M);
    cout << ans.val() << endl;
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    //int T; cin >> T; rep(t,T)
    testcase();
    return 0;
}
0