結果

問題 No.3117 Reversible Tile
ユーザー okkuukenken
提出日時 2025-04-20 17:00:36
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 219 ms / 3,000 ms
コード長 1,043 bytes
コンパイル時間 8,007 ms
コンパイル使用メモリ 356,176 KB
実行使用メモリ 101,536 KB
最終ジャッジ日時 2025-04-20 17:00:49
合計ジャッジ時間 12,957 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include<bits/stdc++.h>
#include<atcoder/all>
using namespace std;
using namespace atcoder; 
using ll=long long;
using mint=modint998244353;
const ll inf=(1e9+7)*(1e9+7);
int dx[4]={1,0,0,-1},dy[4]={0,1,-1,0};
const int mod=998244353;
using P=pair<ll,int>;
mint dp[5001][5002],fact[5002];
int main(){
	int n,m;
	cin>>n>>m;
	vector<int>a(n);
	for(int i=0;i<n;i++)
		cin>>a[i];
	int cnt=a[0]+a[n-1];
	for(int i=1;i<n;i++)
		if(a[i]!=a[i-1])
			cnt++;
	fact[0]=1;
	for(int i=1;i<=n+1;i++)
		fact[i]=fact[i-1]*i;
	mint comb=fact[n+1]/(fact[cnt]*fact[n+1-cnt]);
	int memo1[5002],memo2[5002],memo3[5003];
	for(int j=0;j<=n+1;j+=2){
		if(j>=2)
			memo1[j]=j*(j-1)/2;
		memo2[j]=j*(n+1-j);
		if(j+2<=n+1)
			memo3[j]=(n+1-j)*(n+1-j-1)/2;
	}
	dp[0][0]=1;
	for(int i=0;i<m;i++){
		for(int j=0;j<=n+1;j+=2){
			if(j>=2)
				dp[i+1][j-2]+=dp[i][j]*memo1[j];
			dp[i+1][j]+=dp[i][j]*memo2[j];
			if(j+2<=n+1)
				dp[i+1][j+2]+=dp[i][j]*memo3[j];
		}
	}
	cout<<(dp[m][cnt]/comb).val()<<endl;
}
0