結果
問題 | 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";}}