結果
問題 | No.2514 Twelvefold Way Returns |
ユーザー | chineristAC |
提出日時 | 2023-10-20 22:32:29 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 666 ms / 3,000 ms |
コード長 | 3,324 bytes |
コンパイル時間 | 2,226 ms |
コンパイル使用メモリ | 148,340 KB |
実行使用メモリ | 8,992 KB |
最終ジャッジ日時 | 2023-10-20 22:32:47 |
合計ジャッジ時間 | 16,244 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge13 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
4,348 KB |
testcase_01 | AC | 4 ms
4,348 KB |
testcase_02 | AC | 62 ms
4,348 KB |
testcase_03 | AC | 12 ms
4,348 KB |
testcase_04 | AC | 481 ms
7,936 KB |
testcase_05 | AC | 403 ms
7,316 KB |
testcase_06 | AC | 319 ms
6,616 KB |
testcase_07 | AC | 2 ms
4,348 KB |
testcase_08 | AC | 198 ms
5,824 KB |
testcase_09 | AC | 203 ms
5,736 KB |
testcase_10 | AC | 578 ms
8,464 KB |
testcase_11 | AC | 10 ms
4,348 KB |
testcase_12 | AC | 75 ms
4,504 KB |
testcase_13 | AC | 3 ms
4,348 KB |
testcase_14 | AC | 56 ms
4,348 KB |
testcase_15 | AC | 159 ms
5,296 KB |
testcase_16 | AC | 546 ms
8,200 KB |
testcase_17 | AC | 272 ms
6,352 KB |
testcase_18 | AC | 619 ms
8,512 KB |
testcase_19 | AC | 648 ms
8,888 KB |
testcase_20 | AC | 636 ms
8,728 KB |
testcase_21 | AC | 659 ms
8,904 KB |
testcase_22 | AC | 633 ms
8,888 KB |
testcase_23 | AC | 176 ms
5,560 KB |
testcase_24 | AC | 616 ms
8,464 KB |
testcase_25 | AC | 229 ms
5,836 KB |
testcase_26 | AC | 2 ms
4,348 KB |
testcase_27 | AC | 640 ms
8,924 KB |
testcase_28 | AC | 557 ms
8,728 KB |
testcase_29 | AC | 368 ms
7,464 KB |
testcase_30 | AC | 485 ms
8,464 KB |
testcase_31 | AC | 582 ms
8,992 KB |
testcase_32 | AC | 186 ms
5,836 KB |
testcase_33 | AC | 1 ms
4,348 KB |
testcase_34 | AC | 52 ms
4,348 KB |
testcase_35 | AC | 541 ms
8,368 KB |
testcase_36 | AC | 2 ms
4,348 KB |
testcase_37 | AC | 1 ms
4,348 KB |
testcase_38 | AC | 660 ms
8,904 KB |
testcase_39 | AC | 666 ms
8,992 KB |
testcase_40 | AC | 644 ms
8,992 KB |
ソースコード
#pragma GCC target("avx2") #pragma GCC optimize("O3") #pragma GCC optimize("unroll-loops") #include <iostream> #include <vector> #include <string> #include <map> #include <set> #include <queue> #include <algorithm> #include <cmath> #include <iomanip> #include <random> #include <stdio.h> #include <fstream> #include <functional> #include <cassert> #include <atcoder/modint> using namespace std; using namespace atcoder; using mint = modint998244353; #define rep(i,n) for (int i=0;i<n;i+=1) #define append push_back #define all(x) (x).begin(), (x).end() template<class T> using vec = vector<T>; template<class T> using vvec = vec<vec<T>>; template<class T> using vvvec = vec<vvec<T>>; using ll = long long; using pii = pair<int,int>; using pll = pair<ll,ll>; template<class T> bool chmin(T &a, T b){ if (a>b){ a = b; return true; } return false; } template<class T> bool chmax(T &a, T b){ if (a<b){ a = b; return true; } return false; } template<class T> T sum(vec<T> x){ T res=0; for (auto e:x){ res += e; } return res; } template<class T> void printv(vec<T> x){ for (auto e:x){ cout<<e<<" "; } cout<<"\n"; } template<class T> ostream& operator<<(ostream& os, const vec<T>& A){ os << "{"; rep(i,A.size()){ os << A[i]; if (i!=A.size()-1){ os << ", "; } } os << "}" ; return os; } template<class T> ostream& operator<<(ostream& os, const deque<T>& A){ os << "deque{["; rep(i,A.size()){ os << A[i]; if (i!=A.size()-1){ os << ", "; } } os << "]}" ; return os; } template<class T> ostream& operator<<(ostream& os, const pair<T,T>& A){ os << "("; os << A.first ; os << ", "; os << A.second; os << ")"; return os; } pair<mint,mint> omega_prod(pair<mint,mint> X,pair<mint,mint> Y){ auto [a,b] = X; auto [c,d] = Y; return {(a*d+b*c-a*c),b*d-a*c}; }; pair<mint,mint> omega_pow(pair<mint,mint> X,int N){ pair<mint,mint> res = {0,1}; while (N){ if ((N&1)){ res = omega_prod(res,X); } X = omega_prod(X,X); N >>= 1; } return res; } ostream& operator<<(ostream& os, const mint a){ os << a.val(); return os; } int main() { ios::sync_with_stdio(false); cin.tie(nullptr); int N,M; cin>>N>>M; vec<vec<pair<mint,mint>>> dp(2*M+1,vec<pair<mint,mint>>(2*M+1,{0,0})); pair<mint,mint> zero = {0,0}; dp[M][M] = {0,1}; mint i3 = mint(3).inv(); for (int _=0;_<M;_++){ vec<vec<pair<mint,mint>>> ndp(2*M+1,vec<pair<mint,mint>>(2*M+1,{0,0})); for (int i=0;i<=2*M;i++){ for (int j=0;j<=2*M;j++){ if (dp[i][j] == zero) continue; auto e = dp[i][j]; ndp[i][j+1].first += e.first * i3; ndp[i][j+1].second += e.second * i3; e = omega_prod(e,{1,0}); ndp[i-1][j-1].first += e.first * i3; ndp[i-1][j-1].second += e.second * i3; e = omega_prod(e,{1,0}); ndp[i+1][j].first += e.first * i3; ndp[i+1][j].second += e.second * i3; } } swap(dp,ndp); } mint res = 0; for (int i=0;i<=2*M;i++){ for (int j=0;j<=2*M;j++){ auto e = omega_pow({i-M,j-M},N); auto f = omega_prod(dp[i][j],e); res += f.second; } } cout << res.val() << endl; }