結果
問題 | No.2733 Just K-times TSP |
ユーザー |
![]() |
提出日時 | 2024-04-19 21:57:09 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 1,021 ms / 2,000 ms |
コード長 | 13,412 bytes |
コンパイル時間 | 6,800 ms |
コンパイル使用メモリ | 325,548 KB |
実行使用メモリ | 17,792 KB |
最終ジャッジ日時 | 2024-10-11 15:01:22 |
合計ジャッジ時間 | 10,436 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 32 |
ソースコード
#include <bits/stdc++.h>using namespace std;using ll = long long;using ld = long double;using ull = unsigned long long;using pll = pair<ll, ll>;using tlll = tuple<ll, ll, ll>;using tllll = tuple<ll, ll, ll, ll>;using vll = vector<ll>;using vpll = vector<pll>;using vtlll = vector<tlll>;using vtllll = vector<tllll>;using vstr = vector<string>;using vvl = vector<vector<ll>>;constexpr ll INF = 4'000'000'000'000'000'037;template <class T, class U> bool chmin(T &a, const U &b) { if (a > b) { a = b; return true; } return false; }template <class T, class U> bool chmax(T &a, const U &b) { if (a < b) { a = b; return true; } return false; }ll safemod(ll a, ll m) { ll res = a % m; if (res < 0) res += m; return res; }ll divfloor(ll a, ll b) { if (b < 0) a = -a, b = -b; return (a - safemod(a, b)) / b; }ll divceil(ll a, ll b) { if (b < 0) a = -a, b = -b; return divfloor(a + b - 1, b); }ll pow_ll(ll a, ll b) { if (a == 0 || a == 1) return a; if (a == -1) return b & 1 ? -1 : 1; ll res = 1; for (int i = 0; i < b; i++) res *= a; returnres; }ll mul_limited(ll a, ll b, ll m = INF) { return b == 0 ? 0 : a > m / b ? m : a * b; }ll pow_limited(ll a, ll b, ll m = INF) { if (a == 0 || a == 1) return a; ll res = 1; for (int i = 0; i < b; i++) { if (res > m / a) return m; res *=a; } return res; }#define overload4(_1, _2, _3, _4, name, ...) name#define rep1(i, n) for (ll i = 0; i < ll(n); i++)#define rep2(i, l, r) for (ll i = ll(l); i < ll(r); i++)#define rep3(i, l, r, d) for (ll i = ll(l); (d) > 0 ? i < ll(r) : i > ll(r); i += d)#define rep(...) overload4(__VA_ARGS__, rep3, rep2, rep1)(__VA_ARGS__)#define all(a) (a).begin(), (a).end()template <class T, class F = decltype(plus<>())> vector<T> cuml(vector<T> v, const F &op = plus<>(), const T &e = 0) { v.emplace_back(e);exclusive_scan(v.begin(), v.end(), v.begin(), e, op); return v; }template <class T, class F = decltype(plus<>())> vector<T> cumr(vector<T> v, const F &op = plus<>(), const T &e = 0) { v.insert(v.begin(), e);exclusive_scan(v.rbegin(), v.rend(), v.rbegin(), e, op); return v; }template <class T> vector<T> adjd(vector<T> v) { adjacent_difference(v.begin(), v.end(), v.begin()); return v; }template <class T = ll> struct max_op { T operator()(const T &a, const T &b) const { return max(a, b); } };template <class T = ll> struct min_op { T operator()(const T &a, const T &b) const { return min(a, b); } };struct max_e { ll operator()() const { return -INF; } };struct min_e { ll operator()() const { return INF; } };template <class T> vector<T> cumlmax(const vector<T> &v) { return cuml(v, max_op<T>(), max_e()()); }template <class T> vector<T> cumrmax(const vector<T> &v) { return cumr(v, max_op<T>(), max_e()()); }template <class T> vector<T> cumlmin(const vector<T> &v) { return cuml(v, min_op<T>(), min_e()()); }template <class T> vector<T> cumrmin(const vector<T> &v) { return cumr(v, min_op<T>(), min_e()()); }template <class T> vector<T> sorted(vector<T> v) { sort(v.begin(), v.end()); return v; }template <class T> vector<T> reversed(const vector<T> &v) { return {v.rbegin(), v.rend()}; }template <class T> void unique(vector<T> &v) { v.erase(unique(v.begin(), v.end()), v.end()); }template <class T> void sortunique(vector<T> &v) { sort(v.begin(), v.end()); v.erase(unique(v.begin(), v.end()), v.end()); }template <class T> void rotate(vector<T> v, int k) { rotate(v.begin(), v.begin() + k, v.end()); }template <class T> vector<T> rotated(vector<T> v, int k) { rotate(v.begin(), v.begin() + k, v.end()); return v; }string sorted(string s) { sort(s.begin(), s.end()); return s; }string reversed(const string &s) { return {s.rbegin(), s.rend()}; }void unique(string &s) { s.erase(unique(s.begin(), s.end()), s.end()); }void sortunique(string &s) { sort(s.begin(), s.end()); s.erase(unique(s.begin(), s.end()), s.end()); }void rotate(string &s, int k) { rotate(s.begin(), s.begin() + k, s.end()); }string rotated(string s, int k) { rotate(s.begin(), s.begin() + k, s.end()); return s; }template <class T> vector<vector<T>> transposed(const vector<vector<T>> &a) { const size_t n = a.size(), m = a.at(0).size(); vector<vector<T>> b(m,vector<T>(n)); for (size_t i = 0; i < n; i++) for (size_t j = 0; j < m; j++) b[j][i] = a[i].at(j); return b; }template <class T, const size_t m> array<vector<T>, m> transposed(const vector<array<T, m>> &a) { const size_t n = a.size(); array<vector<T>, m> b; b.fill(vector<T>(n)); for (size_t i = 0; i < n; i++) for (size_t j = 0; j < m; j++) b[j][i] = a[i][j]; return b; }template <class T, const size_t n> vector<array<T, n>> transposed(const array<vector<T>, n> &a) { const size_t m = a.at(0).size(); vector<array<T, n>> b(m); for (size_t i = 0; i < n; i++) for (size_t j = 0; j < m; j++) b[j][i] = a[i].at(j); return b; }template <class T, class U> pair<vector<T>, vector<U>> transposed(const vector<pair<T, U>> &a) { const size_t n = a.size(); vector<T> b(n); vector<U>c(n); for (size_t i = 0; i < n; i++) b[i] = a[i].first, c[i] = a[i].second; return make_pair(b, c); }template <class T, class U> vector<pair<T, U>> transposed(const pair<vector<T>, vector<U>> &a) { const size_t n = a.first.size(); vector<pair<T, U>>b(n); for (size_t i = 0; i < n; i++) b[i] = make_pair(a.first[i], a.second.at(i)); return b; }template <class T1, class T2, class T3> tuple<vector<T1>, vector<T2>, vector<T3>> transposed(const vector<tuple<T1, T2, T3>> &a) { const size_t n = a.size(); vector<T1> b(n); vector<T2> c(n); vector<T3> d(n); for (size_t i = 0; i < n; i++) b[i] = get<0>(a[i]), c[i] = get<1>(a[i]), d[i] = get<2>(a[i]); return make_tuple(b, c, d); }template <class T1, class T2, class T3> vector<tuple<T1, T2, T3>> transposed(const tuple<vector<T1>, vector<T2>, vector<T3>> &a) { const size_t n =get<0>(a).size(); vector<tuple<T1, T2, T3>> b(n); for (size_t i = 0; i < n; i++) b[i] = make_tuple(get<0>(a)[i], get<1>(a).at(i), get<2>(a).at(i)); return b; }template <class T1, class T2, class T3, class T4> tuple<vector<T1>, vector<T2>, vector<T3>, vector<T4>> transposed(const vector<tuple<T1, T2, T3, T4>> &a) { const size_t n = a.size(); vector<T1> b(n); vector<T2> c(n); vector<T3> d(n); vector<T4> e(n); for (size_t i = 0; i < n; i++) b[i] = get<0>(a[i]), c[i] = get<1>(a[i]), d[i] = get<2>(a[i]), e[i] = get<3>(a[i]); return make_tuple(b, c, d, e); }template <class T1, class T2, class T3, class T4> vector<tuple<T1, T2, T3, T4>> transposed(const tuple<vector<T1>, vector<T2>, vector<T3>, vector<T4>> &a) { const size_t n = get<0>(a).size(); vector<tuple<T1, T2, T3, T4>> b(n); for (size_t i = 0; i < n; i++) b[i] = make_tuple(get<0>(a)[i],get<1>(a).at(i), get<2>(a).at(i), get<3>(a).at(i)); return b; }template <class T = ll> vector<T> digitvec(const string &s) { int n = s.size(); vector<T> a(n); for (int i = 0; i < n; i++) a[i] = s[i] - '0'; returna; }#if __cplusplus < 202002Lull bit_ceil(ull x) { ull y = 1; while (y < x) y <<= 1; return y; }ull bit_floor(ull x) { ull y = 1; while (y <= x) y <<= 1; return y >> 1; }ull bit_width(ull x) { ull y = 1, z = 0; while (y <= x) y <<= 1, z++; return z; }ull countr_zero(ull x) { return __builtin_ctzll(x); }ull popcount(ull x) { return __builtin_popcountll(x); }ull has_single_bit(ull x) { return popcount(x) == 1; }#endifull lsb_index(ull x) { assert(x != 0); return countr_zero(x); }ull msb_index(ull x) { assert(x != 0); return bit_width(x) - 1; }ull lsb_mask(ull x) { assert(x != 0); return x & -x; }ull msb_mask(ull x) { assert(x != 0); return bit_floor(x); }#if __has_include(<atcoder/all>)#include <atcoder/all>using namespace atcoder;template <class T, internal::is_modint_t<T> * = nullptr> istream &operator>>(istream &is, T &a) { ll v; is >> v; a = v; return is; }template <class T, internal::is_modint_t<T> * = nullptr> ostream &operator<<(ostream &os, const T &a) { os << a.val(); return os; }#endiftemplate <class T, class U> istream &operator>>(istream &is, pair<T, U> &p) { is >> p.first >> p.second; return is; }template <class T1, class T2, class T3> istream &operator>>(istream &is, tuple<T1, T2, T3> &t) { is >> get<0>(t) >> get<1>(t) >> get<2>(t); return is; }template <class T1, class T2, class T3, class T4> istream &operator>>(istream &is, tuple<T1, T2, T3, T4> &t) { is >> get<0>(t) >> get<1>(t) >> get<2>(t) >> get<3>(t); return is; }template <class T, const size_t N> istream &operator>>(istream &is, array<T, N> &a) { for (size_t i = 0; i < N; i++) is >> a[i]; return is; }template <class... T> void input(T&... a) { (cin >> ... >> a); }template <class T> void inputvec(int n, vector<T> &v) { v.resize(n); for (int i = 0; i < n; i++) cin >> v[i]; }template <class T, class... Ts> void inputvec(int n, vector<T>& v, vector<Ts>&... vs) { inputvec(n, v); inputvec(n, vs...); }template <class T> void inputvec2(int n, int m, vector<vector<T>> &v) { v.assign(n, vector<T>(m)); for (int i = 0; i < n; i++) for (int j = 0; j < m;j++) cin >> v[i][j]; }template <class T, class... Ts> void inputvec2(int n, int m, vector<T>& v, vector<Ts>&... vs) { inputvec2(n, m, v); inputvec2(n, m, vs...); }template <class T, class... Ts> void print(const T& a, const Ts&... b) { cout << a; (cout << ... << (cout << ' ', b)); cout << '\n'; }#define INT(...) int __VA_ARGS__; input(__VA_ARGS__)#define LL(...) ll __VA_ARGS__; input(__VA_ARGS__)#define STR(...) string __VA_ARGS__; input(__VA_ARGS__)#define VINT(n, ...) vector<int> __VA_ARGS__; inputvec(n, __VA_ARGS__)#define VLL(n, ...) vector<ll> __VA_ARGS__; inputvec(n, __VA_ARGS__)#define VSTR(n, ...) vector<string> __VA_ARGS__; inputvec(n, __VA_ARGS__)#define VPLL(n, ...) vector<pll> __VA_ARGS__; inputvec(n, __VA_ARGS__)#define VTLLL(n, ...) vector<tlll> __VA_ARGS__; inputvec(n, __VA_ARGS__)#define VTLLLL(n, ...) vector<tllll> __VA_ARGS__; inputvec(n, __VA_ARGS__)#define AL(n, ...) array<ll, n> __VA_ARGS__; input(__VA_ARGS__)#define VAL(n, m, ...) vector<array<ll, m>> __VA_ARGS__; inputvec(n, __VA_ARGS__)#define VVL(n, m, ...) vector<vector<ll>> __VA_ARGS__; inputvec2(n, m, __VA_ARGS__)#define FINALANS(x) do { cout << (x) << '\n'; exit(0); } while (false)template <class T> void printvec(const vector<T> &v) { int n = v.size(); for (int i = 0; i < n; i++) cout << v[i] << (i == n - 1 ? "" : " "); cout <<'\n'; }template <class T> void printvect(const vector<T> &v) { for (auto &vi : v) cout << vi << '\n';}template <class T> void printvec2(const vector<vector<T>> &v) { for (auto &vi : v) printvec(vi); }#if __has_include(<atcoder/all>)using mint = modint998244353;//using mint = modint1000000007;//using mint = modint;#define MINT(...) mint __VA_ARGS__; input(__VA_ARGS__);#define VMINT(n, ...) vector<mint> __VA_ARGS__; inputvec(n, __VA_ARGS__)#endif#ifdef LOCAL // https://zenn.dev/sassan/articles/19db660e4da0a4#include "/Users/Shared/cpp-dump-main/dump.hpp"#define dump(...) cpp_dump(__VA_ARGS__)CPP_DUMP_DEFINE_EXPORT_OBJECT(mint, val());CPP_DUMP_DEFINE_DANGEROUS_EXPORT_OBJECT(val());#else#define dump(...)#endiftemplate <class Cost>struct Edge{int from, to, index;Cost cost;Edge() {}Edge(int s, int t, Cost c = 1, int i = -1) : from(s), to(t), index(i), cost(c) {}operator int() const { return to; }};template <class Cost>struct Graph : vector<vector<Edge<Cost>>>{int n, m;vector<Edge<Cost>> edges;Graph() : n(0), m(0) {}Graph(int n) : vector<vector<Edge<Cost>>>(n), n(n), m(0) {}int add_edge(int s, int t, Cost c = 1){(*this).at(s).emplace_back(s, t, c, m);edges.emplace_back(s, t, c, m);return m++;}int add_edge2(int s, int t, Cost c = 1){(*this).at(s).emplace_back(s, t, c, m);(*this).at(t).emplace_back(t, s, c, m);edges.emplace_back(s, t, c, m);return m++;}};template <class Cost, const bool is_weighted, const bool is_directed, const int base_index>Graph<Cost> input_graph(int n, int m, Cost weight_default = 1){Graph<Cost> g(n);for (int i = 0; i < m; i++){int a, b;Cost c;cin >> a >> b;if (is_weighted) cin >> c;else c = weight_default;a -= base_index, b -= base_index;if (is_directed) g.add_edge(a, b, c);else g.add_edge2(a, b, c);}return g;}template <class Cost, const bool is_directed, const int base_index>Graph<Cost> input_parents(int n){Graph<Cost> g(n);for (int i = 1; i < n; i++){int p;cin >> p;p -= base_index;if (is_directed) g.add_edge(p, i);else g.add_edge2(p, i);}return g;}int main(){LL(N, M, K);auto G = input_graph<ll, false, false, 1>(N, M);auto f = [&](ll b){vll A(N);ll bb = b;rep(j, N){A.at(j) = bb % (K + 1);bb /= (K + 1);}return A;};vector dp(N, vector(pow_ll(K + 1, N), mint(0)));rep(i, N) dp.at(i).at(pow_ll(K + 1, i)) = 1;rep(b, pow_ll(K + 1, N)) rep(i, N){if (dp.at(i).at(b) == 0)continue;vll A = f(b);for (auto j : G.at(i)){if (A.at(j) == K) continue;ll nb = b + pow_ll(K + 1, j);dp.at(j).at(nb) += dp.at(i).at(b);dump(i, b, A, j, nb, f(nb));}}mint ans = 0;rep(i, N){mint tmp = dp.at(i).back();ans += tmp;}cout << ans << endl;rep(i, N) rep(b, pow_ll(K + 1, N)){vll A(N);ll bb = b;rep(j, N){A.at(j) = bb % (K + 1);bb /= (K + 1);}dump(i, b, A, dp.at(i).at(b));}}