結果

問題 No.2733 Just K-times TSP
ユーザー Today03
提出日時 2024-04-19 23:19:54
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,346 bytes
コンパイル時間 4,378 ms
コンパイル使用メモリ 273,964 KB
実行使用メモリ 86,400 KB
最終ジャッジ日時 2024-10-11 17:53:59
合計ジャッジ時間 14,516 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 30 TLE * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int INF = 1e9 + 10;
const ll INFL = 4e18;

const int MOD = 998244353;

int N, M, K;
vector<int> G[6];
map<int, int> dp[6];
vector<int> v;

int g(vector<int> &v) {
    int ret = 0;
    int tmp = 1;
    for (int x : v) {
        ret += x * tmp;
        tmp *= K + 1;
    }
    return ret;
}

int cnt = 0;

int F(int now) {
    cnt++;
    if (dp[now].count(g(v))) {
        return dp[now][g(v)];
    }

    bool ok = true;
    for (int i = 0; i < N; i++) {
        if (v[i] != K) {
            ok = false;
        }
    }

    if (ok) {
        return dp[now][g(v)] = 1;
    }

    int ret = 0;
    for (int nxt : G[now]) {
        if (v[nxt] < K) {
            v[nxt]++;
            ret += F(nxt);
            ret %= MOD;
            v[nxt]--;
        }
    }

    return dp[now][g(v)] = ret;
};

int main() {
    cin >> N >> M >> K;

    for (int i = 0; i < M; i++) {
        int u, v;
        cin >> u >> v;
        u--;
        v--;
        G[u].push_back(v);
        G[v].push_back(u);
    }

    v = vector<int>(N, 0);

    int ans = 0;
    for (int i = 0; i < N; i++) {
        v[i]++;
        ans += F(i);
        ans %= MOD;
        v[i]--;
    }

    cout << ans << endl;
}
0