結果

問題 No.3117 Reversible Tile
ユーザー okkuukenken
提出日時 2025-04-20 16:56:44
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 880 bytes
コンパイル時間 7,550 ms
コンパイル使用メモリ 356,920 KB
実行使用メモリ 101,540 KB
最終ジャッジ日時 2025-04-20 16:57:45
合計ジャッジ時間 61,128 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 12 TLE * 12
権限があれば一括ダウンロードができます

ソースコード

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]);
	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]*j*(j-1)/2;
			dp[i+1][j]+=dp[i][j]*j*(n+1-j);
			if(j+2<=n+1)
				dp[i+1][j+2]+=dp[i][j]*(n+1-j)*(n+1-j-1)/2;
		}
	}
	cout<<(dp[m][cnt]/comb).val()<<endl;
}
0