結果

問題 No.2733 Just K-times TSP
ユーザー ttkkggww
提出日時 2024-05-02 08:51:26
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 138 ms / 2,000 ms
コード長 1,361 bytes
コンパイル時間 3,456 ms
コンパイル使用メモリ 254,148 KB
最終ジャッジ日時 2025-02-21 10:08:00
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using ll = long long;
using mint = modint998244353;

mint dp[6][9][9][9][9][9][9];
int N, M, K;
vector<vector<int>> G;

int indies[6];

void dfs(int depth) {

  if (depth == N) {
    for (int i = 0; i < N; i++) {
      for (auto to : G[i]) {
        if (indies[to] == K)
          continue;
        mint tmp = dp[i][indies[0]][indies[1]][indies[2]][indies[3]][indies[4]]
                     [indies[5]];
        indies[to]++;
        dp[to][indies[0]][indies[1]][indies[2]][indies[3]][indies[4]]
          [indies[5]] += tmp;
        indies[to]--;
      }
    }
    return;
  }
  for (int i = 0; i <= K; i++) {
    indies[depth] = i;
    dfs(depth + 1);
  }
}

void solve() {
  for (int i = 0; i < N; i++) {
    indies[i] = 1;
    dp[i][indies[0]][indies[1]][indies[2]][indies[3]][indies[4]][indies[5]] = 1;
    indies[i] = 0;
  }
  dfs(0);
  mint ans = 0;
  for (int i = 0; i < N; i++) {
    ans +=
        dp[i][indies[0]][indies[1]][indies[2]][indies[3]][indies[4]][indies[5]];
  }
  cout << ans.val() << endl;
}

signed main() {
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  cin >> N >> M >> K;
  G.resize(N);
  for (int i = 0; i < M; i++) {
    int a, b;
    cin >> a >> b;
    a--;
    b--;
    G[a].push_back(b);
    G[b].push_back(a);
  }
  solve();
}
0