結果

問題 No.3117 Reversible Tile
コンテスト
ユーザー okkuukenken
提出日時 2025-04-20 16:53:45
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 919 bytes
コンパイル時間 11,010 ms
コンパイル使用メモリ 355,740 KB
実行使用メモリ 117,192 KB
最終ジャッジ日時 2025-04-20 16:54:05
合計ジャッジ時間 17,535 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 3
other TLE * 1 -- * 23
権限があれば一括ダウンロードができます

ソースコード

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>;
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++;
	vector<vector<mint>>dp(m+1,vector<mint>(n+2));
	vector<mint>fact(n+2);
	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++){
			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