結果
問題 | No.1513 simple 門松列 problem |
ユーザー | 👑 potato167 |
提出日時 | 2022-02-09 18:13:37 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 49 ms / 3,000 ms |
コード長 | 1,756 bytes |
コンパイル時間 | 2,258 ms |
コンパイル使用メモリ | 210,580 KB |
実行使用メモリ | 6,948 KB |
最終ジャッジ日時 | 2024-06-24 22:26:37 |
合計ジャッジ時間 | 3,441 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,816 KB |
testcase_01 | AC | 2 ms
6,944 KB |
testcase_02 | AC | 49 ms
6,944 KB |
testcase_03 | AC | 2 ms
6,948 KB |
testcase_04 | AC | 2 ms
6,940 KB |
testcase_05 | AC | 2 ms
6,940 KB |
testcase_06 | AC | 2 ms
6,940 KB |
testcase_07 | AC | 2 ms
6,940 KB |
testcase_08 | AC | 2 ms
6,944 KB |
testcase_09 | AC | 2 ms
6,944 KB |
testcase_10 | AC | 2 ms
6,940 KB |
testcase_11 | AC | 2 ms
6,940 KB |
testcase_12 | AC | 2 ms
6,944 KB |
testcase_13 | AC | 2 ms
6,940 KB |
testcase_14 | AC | 48 ms
6,944 KB |
testcase_15 | AC | 48 ms
6,944 KB |
testcase_16 | AC | 49 ms
6,940 KB |
testcase_17 | AC | 30 ms
6,940 KB |
testcase_18 | AC | 12 ms
6,940 KB |
testcase_19 | AC | 3 ms
6,944 KB |
testcase_20 | AC | 6 ms
6,940 KB |
ソースコード
#include <bits/stdc++.h> #pragma GCC optimize("Ofast") #define _GLIBCXX_DEBUG using namespace std; using std::cout; using std::cin; using std::endl; using ll=long long; using ld=long double; ll ILL=1167167167167167167; const int INF=2100000000; const ll mod=998244353; #define rep(i,a) for (ll i=0;i<a;i++) template<class T> using _pq = priority_queue<T, vector<T>, greater<T>>; template<class T> ll LB(vector<T> &v,T a){return lower_bound(v.begin(),v.end(),a)-v.begin();} template<class T> ll UB(vector<T> &v,T a){return upper_bound(v.begin(),v.end(),a)-v.begin();} template<class T> bool chmin(T &a,const T &b){if(a>b){a=b;return 1;}else return 0;} template<class T> bool chmax(T &a,const T &b){if(a<b){a=b;return 1;}else return 0;} template<class T> void So(vector<T> &v) {sort(v.begin(),v.end());} template<class T> void Sore(vector<T> &v) {sort(v.begin(),v.end(),[](T x,T y){return x>y;});} void yneos(bool a){if(a) cout<<"YES\n"; else cout<<"NO\n";} void solve(); // rainy ~ 雨に打たれて ~ int main() { ios::sync_with_stdio(false); cin.tie(nullptr); int t=1; //cin>>t; rep(i,t) solve(); } void solve(){ int N,K; cin>>N>>K; vector<vector<ll>> dp(K,vector<ll>(K)); rep(i,K) rep(j,i) dp[j][i]=1; rep(i,N-2){ vector<vector<ll>> n_dp(K,vector<ll>(K)); rep(j,K) rep(k,K){ if(i%2!=1&&k!=0) n_dp[k][k-1]+=dp[j][k]; else if(i%2==1&&k+1!=K) n_dp[k][k+1]+=dp[j][k]; } if(i%2){ rep(j,K) rep(k,K-1) n_dp[j][k+1]+=n_dp[j][k]; }else{ rep(j,K) rep(k,K-1) n_dp[j][K-k-2]+=n_dp[j][K-k-1]; } rep(j,K) rep(k,K) n_dp[j][k]=(n_dp[j][k]-dp[k][j])%mod; swap(n_dp,dp); } ll ans=0; rep(j,K) rep(k,K) ans+=dp[j][k]; ans%=mod; ll ans2=(ans*(K-1)*N)%mod; ans*=2ll; cout<<(ans%mod+mod)%mod<<" "<<(ans2+mod)%mod<<"\n"; }