結果
問題 |
No.2524 Stripes
|
ユーザー |
|
提出日時 | 2023-10-27 23:14:37 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 620 ms / 7,000 ms |
コード長 | 1,984 bytes |
コンパイル時間 | 2,406 ms |
コンパイル使用メモリ | 117,016 KB |
実行使用メモリ | 10,976 KB |
最終ジャッジ日時 | 2024-09-25 15:12:51 |
合計ジャッジ時間 | 7,016 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 25 |
ソースコード
#include<iostream> #include<cassert> #include<atcoder/modint> #include<atcoder/convolution> using namespace std; using mint=atcoder::modint998244353; struct dat{ vector<mint>RR,RB,BR,BB; }; void add(vector<mint>&lhs,const vector<mint>&rhs) { if(lhs.size()<rhs.size())lhs.resize(rhs.size()); for(int i=0;i<rhs.size();i++)lhs[i]+=rhs[i]; } dat mul(dat a,dat b) { dat ret; ret.RR=atcoder::convolution(a.RR,b.RR); add(ret.RR,atcoder::convolution(a.RB,b.BR)); ret.RB=atcoder::convolution(a.RR,b.RB); add(ret.RB,atcoder::convolution(a.RB,b.BB)); ret.BR=atcoder::convolution(a.BR,b.RR); add(ret.BR,atcoder::convolution(a.BB,b.BR)); ret.BB=atcoder::convolution(a.BR,b.RB); add(ret.BB,atcoder::convolution(a.BB,b.BB)); return ret; } const int LIM=300; mint dp[2][2][2][LIM+1]; void upd(vector<mint>&A,mint dp[LIM+1]) { int sz=LIM+1; while(sz>0&&dp[sz-1]==mint::raw(0))sz--; A=vector<mint>(dp,dp+sz); } dat solve(string S) { if(S.size()<=LIM) { int now=0; for(int j=0;j<2;j++)for(int k=0;k<2;k++) for(int i=0;i<=LIM;i++) dp[now][j][k][i]=mint::raw(0); dp[now][0][0][0]=dp[now][1][1][0]=mint::raw(1); for(char c:S) { int nxt=1-now; for(int j=0;j<2;j++)for(int k=0;k<2;k++) for(int i=0;i<=LIM;i++) dp[nxt][j][k][i]=dp[now][j][k][i]; int k=c=='R'?0:1; for(int j=0;j<2;j++) for(int i=0;i<LIM;i++) dp[nxt][j][1-k][i+1]+=dp[now][j][k][i]; now=nxt; } dat ret; upd(ret.RR,dp[now][0][0]); upd(ret.RB,dp[now][0][1]); upd(ret.BR,dp[now][1][0]); upd(ret.BB,dp[now][1][1]); return ret; } int mid=S.size()/2; dat L=solve(S.substr(0,mid)),R=solve(S.substr(mid,S.size()-mid)); return mul(L,R); } int main() { ios::sync_with_stdio(false); cin.tie(nullptr); int N;string S; cin>>N>>S; dat ret=solve(S); for(int i=1;i<=N;i++) { mint ans=0; if(ret.RR.size()>i)ans+=ret.RR[i]; if(ret.RB.size()>i)ans+=ret.RB[i]; if(ret.BR.size()>i)ans+=ret.BR[i]; if(ret.BB.size()>i)ans+=ret.BB[i]; cout<<ans.val()<<"\n"; } }