結果

問題 No.2733 Just K-times TSP
ユーザー umezoumezo
提出日時 2024-04-19 23:15:44
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 348 ms / 2,000 ms
コード長 905 bytes
コンパイル時間 2,980 ms
コンパイル使用メモリ 249,824 KB
実行使用メモリ 28,272 KB
最終ジャッジ日時 2024-10-11 17:43:31
合計ジャッジ時間 5,097 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;

#define rep(i,n) for(int i=0;i<(int)(n);i++)
#define ALL(v) v.begin(),v.end()
typedef long long ll;
template <class T> using V=vector<T>;
template <class T> using VV=V<V<T>>; //B(n,V<int>(n))
 
ll dp[600000][6];
V<int> G[6];
const int MOD=998244353;
 
int main(){
  ios::sync_with_stdio(false);
  std::cin.tie(nullptr);
  
  ll n,m,k;
  cin>>n>>m>>k;
  V<ll> A(n+1);
  A[0]=1;
  rep(i,n) A[i+1]=A[i]*(k+1);
  rep(i,m){
    int u,v;
    cin>>u>>v;
    u--,v--;
    G[u].push_back(v);
    G[v].push_back(u);
  }
  rep(i,n) dp[A[i]][i]=1;
  for(int i=0;i<A[n];i++){
    for(int j=0;j<n;j++){
      if(dp[i][j]==0) continue;
      for(auto nv:G[j]){
        if((i/A[nv])%(k+1)==k) continue;
        dp[i+A[nv]][nv]=(dp[i+A[nv]][nv]+dp[i][j])%MOD;
      }
    }
  }
  ll ans=0;
  rep(i,n){
    ans=(ans+dp[A[n]-1][i])%MOD;
  }
  cout<<ans<<endl;
    
  return 0;
}
0