結果

問題 No.2524 Stripes
ユーザー kotatsugamekotatsugame
提出日時 2023-10-27 23:14:37
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 654 ms / 7,000 ms
コード長 1,984 bytes
コンパイル時間 2,342 ms
コンパイル使用メモリ 115,700 KB
実行使用メモリ 10,948 KB
最終ジャッジ日時 2023-10-27 23:14:45
合計ジャッジ時間 7,144 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge10
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,348 KB
testcase_01 AC 2 ms
4,348 KB
testcase_02 AC 2 ms
4,348 KB
testcase_03 AC 196 ms
5,564 KB
testcase_04 AC 9 ms
4,348 KB
testcase_05 AC 133 ms
5,064 KB
testcase_06 AC 194 ms
5,576 KB
testcase_07 AC 361 ms
7,292 KB
testcase_08 AC 388 ms
7,264 KB
testcase_09 AC 67 ms
4,440 KB
testcase_10 AC 179 ms
5,480 KB
testcase_11 AC 9 ms
4,348 KB
testcase_12 AC 44 ms
4,348 KB
testcase_13 AC 378 ms
7,484 KB
testcase_14 AC 401 ms
7,588 KB
testcase_15 AC 378 ms
7,372 KB
testcase_16 AC 654 ms
10,948 KB
testcase_17 AC 155 ms
4,348 KB
testcase_18 AC 2 ms
4,348 KB
testcase_19 AC 2 ms
4,348 KB
testcase_20 AC 2 ms
4,348 KB
testcase_21 AC 2 ms
4,348 KB
testcase_22 AC 2 ms
4,348 KB
testcase_23 AC 2 ms
4,348 KB
testcase_24 AC 2 ms
4,348 KB
testcase_25 AC 2 ms
4,348 KB
testcase_26 AC 2 ms
4,348 KB
testcase_27 AC 2 ms
4,348 KB
権限があれば一括ダウンロードができます

ソースコード

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