結果

問題 No.2733 Just K-times TSP
ユーザー kaichou243kaichou243
提出日時 2024-04-19 21:52:12
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 906 ms / 2,000 ms
コード長 1,315 bytes
コンパイル時間 2,446 ms
コンパイル使用メモリ 211,388 KB
実行使用メモリ 32,256 KB
最終ジャッジ日時 2024-10-11 14:50:00
合計ジャッジ時間 6,754 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
void mpl(ll& x,ll y){
	x+=y;
	if(x>998244353) x-=998244353;
}
int main(){
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int n,m,k;
	cin>>n>>m>>k;
	vector<vector<int>> g(n);
	for(int i=0;i<m;i++){
	    int u,v;
	    cin>>u>>v;
	    g[--u].push_back(--v);
	    g[v].push_back(u);
	}
	int cnts=1;
	for(int i=0;i<n;i++) cnts*=(k+1);
	vector<vector<ll>> dp(n,vector<ll>(cnts));
	auto turn_vec=[&](int state)->vector<int>{
	    vector<int> ret;
	    for(int i=0;i<n;i++){
	        ret.push_back(state%(k+1));
	        state/=k+1;
	    }
	    return ret;
	};
	auto turn_int=[&](vector<int> state)->int{
	    int ret=0,pw=1;
	    for(int i=0;i<n;i++){
	        ret+=pw*state[i];
	        pw*=k+1;
	    }
	    return ret;
	};
	for(int i=0;i<n;i++){
	    vector<int> state(n);
	    state[i]=1;
	    dp[i][turn_int(state)]=1;
	}
	for(int s=1;s<cnts;s++){
	    vector<int> state=turn_vec(s);
	    for(int i=0;i<n;i++){
	        for(auto& nv:g[i]){
	            if(state[nv]<k){
	                vector<int> new_state=state;
	                new_state[nv]++;
	                int ns=turn_int(new_state);
	                mpl(dp[nv][ns],dp[i][s]);
	            }
	        }
	    }
	}
	ll ans=0;
	for(int i=0;i<n;i++) mpl(ans,dp[i][cnts-1]);
	cout<<ans<<endl;
}
0