結果

問題 No.2524 Stripes
コンテスト
ユーザー kotatsugame
提出日時 2023-10-27 23:14:37
言語 C++14
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 620 ms / 7,000 ms
コード長 1,984 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 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 #
raw source code 

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