結果

問題 No.2524 Stripes
ユーザー kotatsugame
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#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";
}
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0