#line 1 "No_114_\u9060\u3044\u672a\u6765.cpp" #define YRSD #line 2 "YRS/all.hpp" #line 2 "YRS/aa/head.hpp" #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #define TE template #define TES template #define Z auto #define ep emplace_back #define eb emplace #define fi first #define se second #define all(x) (x).begin(), (x).end() #define OV4(a, b, c, d, e, ...) e #define FOR1(a) for (int _ = 0; _ < (a); ++_) #define FOR2(i, a) for (int i = 0; i < (a); ++i) #define FOR3(i, a, b) for (int i = (a); i < (b); ++i) #define FOR4(i, a, b, c) for (int i = (a); i < (b); i += (c)) #define FOR(...) OV4(__VA_ARGS__, FOR4, FOR3, FOR2, FOR1)(__VA_ARGS__) #define FOR1_R(a) for (int _ = (a) - 1; _ >= 0; --_) #define FOR2_R(i, a) for (int i = (a) - 1; i >= 0; --i) #define FOR3_R(i, a, b) for (int i = (b) - 1; i >= (a); --i) #define FOR4_R(i, a, b, c) for (int i = (b) - 1; i >= (a); i -= (c)) #define FOR_R(...) OV4(__VA_ARGS__, FOR4_R, FOR3_R, FOR2_R, FOR1_R)(__VA_ARGS__) #define FOR_subset(t, s) for (int t = (s); t > -1; t = (t == 0 ? -1 : (t - 1) & s)) #define sort ranges::sort using namespace std; TE using vc = vector; TE using vvc = vc>; TE using T1 = tuple; TE using T2 = tuple; TE using T3 = tuple; TE using T4 = tuple; TE using max_heap = priority_queue; TE using min_heap = priority_queue, greater>; using u8 = unsigned char; using uint = unsigned int; using ll = long long; using ull = unsigned long long; using ld = long double; using i128 = __int128; using u128 = __uint128_t; using f128 = __float128; using u16 = uint16_t; using PII = pair; using PLL = pair; #ifdef YRSD constexpr bool dbg = 1; #else constexpr bool dbg = 0; #endif #line 2 "YRS/IO/IO.hpp" istream &operator>>(istream &I, i128 &x) { static string s; I >> s; int f = s[0] == '-'; x = 0; const int N = (int)s.size(); FOR(i, f, N) x = x * 10 + s[i] - '0'; if (f) x = -x; return I; } ostream &operator<<(ostream &O, i128 x) { static string s; s.clear(); bool f = x < 0; if (f) x = -x; while (x) s += '0' + x % 10, x /= 10; if (s.empty()) s += '0'; if (f) s += '-'; reverse(all(s)); return O << s; } istream &operator>>(istream &I, f128 &x) { static string s; I >> s, x = stold(s); return I; } ostream &operator<<(ostream &O, const f128 x) { return O << ld(x); } template istream &operator>>(istream &I, tuple &t) { return apply([&I](Z &...s) { ((I >> s), ...); }, t), I; } template istream &operator>>(istream &I, pair &x) { return I >> x.fi >> x.se; } template ostream &operator<<(ostream &O, const pair &x) { return O << x.fi << ' ' << x.se; } TE requires requires(T &c) { begin(c); end(c); } and (not is_same_v, string>) istream &operator>>(istream &I, T &c) { for (Z &e : c) I >> e; return I; } TE requires requires(const T &c) { begin(c); end(c); } and (not is_same_v, const char*>) and (not is_same_v, string>) and (not is_array_v> or not is_same_v>, char>) ostream &operator<<(ostream &O, const T &a) { if (a.empty()) return O; Z i = a.begin(); O << *i++; for (; i != a.end(); ++i) O << ' ' << *i; return O; } void IN() {} TE void IN(T &x, Z &...s) { cin >> x, IN(s...); } void print() { cout << '\n'; } TES void print(T &&x, S &&...y) { cout << x; if constexpr (sizeof...(S)) cout << ' '; print(forward(y)...); } void put() { cout << ' '; } TES void put(T &&x, S &&...y) { cout << x; if constexpr (sizeof...(S)) cout << ' '; put(forward(y)...); } #define INT(...) int __VA_ARGS__; IN(__VA_ARGS__) #define UINT(...) uint __VA_ARGS__; IN(__VA_ARGS__) #define LL(...) ll __VA_ARGS__; IN(__VA_ARGS__) #define ULL(...) ull __VA_ARGS__; IN(__VA_ARGS__) #define I128(...) i128 __VA_ARGS__; IN(__VA_ARGS__) #define STR(...) string __VA_ARGS__; IN(__VA_ARGS__) #define CH(...) char __VA_ARGS__; IN(__VA_ARGS__) #define REAL(...) re __VA_ARGS__; IN(__VA_ARGS__) #define VEC(T, a, n) vc a(n); IN(a) void YES(bool o = 1) { print(o ? "YES" : "NO"); } void Yes(bool o = 1) { print(o ? "Yes" : "No"); } void yes(bool o = 1) { print(o ? "yes" : "no"); } void NO(bool o = 1) { YES(not o); } void No(bool o = 1) { Yes(not o); } void no(bool o = 1) { yes(not o); } void ALICE(bool o = 1) { print(o ? "ALICE" : "BOB"); } void Alice(bool o = 1) { print(o ? "Alice" : "Bob"); } void alice(bool o = 1) { print(o ? "alice" : "bob"); } void BOB(bool o = 1) { ALICE(not o); } void Bob(bool o = 1) { Alice(not o); } void bob(bool o = 1) { alice(not o); } void POSSIBLE(bool o = 1) { print(o ? "POSSIBLE" : "IMPOSSIBLE"); } void Possible(bool o = 1) { print(o ? "Possible" : "Impossible"); } void possible(bool o = 1) { print(o ? "possible" : "impossible"); } void IMPOSSIBLE(bool o = 1) { POSSIBLE(not o); } void Impossible(bool o = 1) { Possible(not o); } void impossible(bool o = 1) { possible(not o); } void TAK(bool o = 1) { print(o ? "TAK" : "NIE"); } void NIE(bool o = 1) { TAK(not o); } #line 5 "YRS/all.hpp" #if (__cplusplus >= 202002L) #include constexpr ld pi = numbers::pi; #endif TE constexpr T inf = numeric_limits::max(); template <> constexpr i128 inf = i128(inf) * 2'000'000'000'000'000'000; template constexpr pair inf> = {inf, inf}; TE constexpr static inline int pc(T x) { return popcount(make_unsigned_t(x)); } constexpr static inline ll len(const Z &a) { return a.size(); } void reverse(Z &a) { reverse(all(a)); } void unique(Z &a) { sort(a); a.erase(unique(all(a)), a.end()); } TE vc inverse(const vc &a) { int N = len(a); vc b(N, -1); FOR(i, N) if (a[i] != -1) b[a[i]] = i; return b; } Z QMAX(const Z &a) { return *max_element(all(a)); } Z QMIN(const Z &a) { return *min_element(all(a)); } constexpr bool chmax(Z &a, const Z &b) { return (a < b ? a = b, 1 : 0); } constexpr bool chmin(Z &a, const Z &b) { return (a > b ? a = b, 1 : 0); } vc argsort(const Z &a) { vc I(len(a)); iota(all(I), 0); sort(I, [&](int i, int k) { return a[i] < a[k] or (a[i] == a[k] and i < k); }); return I; } TE vc rearrange(const vc &a, const vc &I) { int N = len(I); vc b(N); FOR(i, N) b[i] = a[I[i]]; return b; } template vc pre_sum(const vc &a) { int N = len(a); vc c(N + 1); FOR(i, N) c[i + 1] = c[i] + a[i]; if (of == 0) c.erase(c.begin()); return c; } TE constexpr static int topbit(T x) { if (x == 0) return - 1; if constexpr (sizeof(T) <= 4) return 31 - __builtin_clz(x); else return 63 - __builtin_clzll(x); } TE constexpr static int lowbit(T x) { if (x == 0) return -1; if constexpr (sizeof(T) <= 4) return __builtin_ctz(x); else return __builtin_ctzll(x); } TE constexpr T floor(T x, T y) { return x / y - (x % y and (x ^ y) < 0); } TE constexpr T ceil(T x, T y) { return floor(x + y - 1, y); } TE constexpr T bmod(T x, T y) { return x - floor(x, y) * y; } TE constexpr pair divmod(T x, T y) { T q = floor(x, y); return pair{q, x - q * y}; } template T SUM(const Z &v) { return accumulate(all(v), T(0)); } int lb(const Z &a, Z x) { return lower_bound(all(a), x) - a.begin(); } int lb(Z l, Z r, Z x) requires(is_same_v) { return lower_bound(l, r, x) - l; } int ub(const Z &a, Z x) { return upper_bound(all(a), x) - a.begin(); } int ub(Z l, Z r, Z x) requires(is_same_v) { return upper_bound(l, r, x) - l; } template ll bina(Z f, ll l, ll r) { if constexpr (ck) assert(f(l)); while (abs(l - r) > 1) { ll x = (r + l) >> 1; (f(x) ? l : r) = x; } return l; } TE T bina_real(Z f, T l, T r, int c = 100) { while (c--) { T x = (l + r) / 2; (f(x) ? l : r) = x; } return (l + r) / 2; } Z pop(Z &s) { if constexpr (requires { s.pop_back(); }) { Z x = s.back(); return s.pop_back(), x; } else if constexpr (requires { s.top(); }) { Z x = s.top(); return s.pop(), x; } else { Z x = s.front(); return s.pop(), x; } } void setp(int x) { cout << fixed << setprecision(x); } TE inline void sh(vc &a, int N) { a.resize(N, T(0)); } #line 1 "YRS/debug.hpp" #ifdef YRSD void DBG() { cerr << "]" << endl; } TES void DBG(T &&x, S &&...y) { cerr << x; if constexpr (sizeof...(S)) cerr << ", "; DBG(forward(y)...); } #define debug(...) cerr << "[" << __LINE__ << "]: [" #__VA_ARGS__ "] = [", DBG(__VA_ARGS__) void ERR() { cerr << endl; } TES void ERR(T &&x, S &&...y) { cerr << x; if constexpr (sizeof...(S)) cerr << ", "; ERR(forward(y)...); } #define err(...) cerr << "[" << __LINE__ << "]: ", ERR(__VA_ARGS__) #define asser assert #else #define debug(...) void(0721) #define err(...) void(0721) #define asser(...) void(0721) #endif #line 4 "No_114_\u9060\u3044\u672a\u6765.cpp" // #include "YRS/IO/fast_io.hpp" // #include "YRS/random/rng.hpp" #line 2 "YRS/g/steiner.hpp" #line 2 "YRS/g/Basic.hpp" #line 2 "YRS/ds/basic/hashmap.hpp" TE struct hash_map { uint ls, msk; vc ke; vc val; vc vis; ull hash(ull x) const { static const ull FIXED_RANDOM = std::chrono::steady_clock::now().time_since_epoch().count(); x += FIXED_RANDOM; x = (x ^ (x >> 30)) * 0xbf58476d1ce4e5b9; x = (x ^ (x >> 27)) * 0x94d049bb133111eb; return (x ^ (x >> 31)) & msk; } void extend() { vc> dat; const int N = len(vis); dat.reserve(N / 2 - ls); FOR(i, N) if (vis[i]) dat.ep(ke[i], val[i]); build(dat.size() << 1); for (Z &[a, b] : dat) (*this)[a] = b; } hash_map(uint N = 0) { build(N); } void build(uint N) { uint k = 8; while (k < (N << 1)) k <<= 1; ls = k >> 1, msk = k - 1; ke.resize(k), val.resize(k), vis.assign(k, 0); } void clear() { fill(all(vis), 0); ls = (msk + 1) >> 1; } ll size() const { return vis.size() / 2 - ls; } int id(ull k) const { int i = hash(k); while (vis[i] and ke[i] != k) i = (i + 1) & msk; return i; } T &operator[](ull k) { if (ls == 0) extend(); int i = id(k); if (not vis[i]) { vis[i] = 1; ke[i] = k; val[i] = T {}; --ls; } return val[i]; } T get(ull k, T fail) const { int i = id(k); return (vis[i] ? val[i] : fail); } bool contains(ull k) const { int i = id(k); return vis[i] and ke[i] == k; } vc> get_all() const { int N = len(vis); vc> s; FOR(i, N) if (vis[i]) s.ep(ke[i], val[i]); return s; } // f(ke, val); void enumerate_all(Z f) const { const int N = len(vis); FOR(i, N) if (vis[i]) f(ke[i], val[i]); } }; #line 4 "YRS/g/Basic.hpp" // https://www.luogu.com.cn/problem/P5318 TE struct edge { int f, to; T w; int id; }; template struct graph { static constexpr bool is_dir = dir; int N, M; using cost_type = T; using ee = edge; vc es; vc in; vc c; bool ok; bool isok() { return ok; } struct px { const graph *g; int l, r; px(const graph *g, int l, int r) : g(g), l(l), r(r) {} const ee *begin() const { if (l == r) return 0; return &g->c[l]; } const ee *end() const { if (l == r) return 0; return &g->c[r]; } }; px operator[](int i) const { assert(ok); return {this, in[i], in[i + 1]}; } graph() : N(0), M(0), ok(0) {} graph(int N) : N(N), M(0), ok(0) {} void add(int f, int t, T w = 1, int i = -1) { assert(not ok); assert(-1 < f and -1 < t and t < N and f < N); if (i == -1) i = M; Z e = ee({f, t, w, i}); es.ep(e); ++M; } void build() { assert(not ok); ok = 1; in.assign(N + 1, 0); for (Z &&e : es) { in[e.f + 1]++; if (not dir) in[e.to + 1]++; } FOR(i, N) in[i + 1] += in[i]; Z cc = in; c.resize(in.back() + 1); for (Z &&e : es) { c[cc[e.f]++] = e; if (not dir) c[cc[e.to]++] = ee({e.to, e.f, e.w, e.id}); } } template void read_tree() { sc(N - 1); } template void read_graph(int M) { sc(M); } template void sc() { sc(N - 1); } template void sc(int M) { es.reserve(M * (dir ? 1 : 2)); FOR(M) { INT(x, y); x -= of, y -= of; if (not wt) { add(x, y); } else { T w; IN(w); add(x, y, w); } } build(); } vc deg() { vc in(N); for (Z &&e : es) ++in[e.f], ++in[e.to]; return in; } pair, vc> deg_inout() { vc in(N), ou(N); for (Z &&e : es) ++in[e.to], ++ou[e.f]; return {in, ou}; } vc ni; vc vis; // 使G中的顶点V[i]在新图表中为i // {G, es} // sum(deg(v))的计算量 // 注意它可能大于新图表的n+M graph rearrange(const vc &v, bool keep_eid = 0) { if (len(ni) != N) ni.assign(N, -1); int N = len(v); FOR(i, N) ni[v[i]] = i; graph g(N); vc s; FOR(i, N) { for (Z &&e : (*this)[v[i]]) { if (len(vis) <= e.id) vis.resize(e.id + 1); if (vis[e.id]) continue; int f = e.f, to = e.to; if (ni[f] != -1 and ni[to] != -1) { s.ep(e.id); vis[e.id] = 1; int id = (keep_eid ? e.id : -1); g.add(ni[f], ni[to], e.w, id); } } } FOR(i, N) ni[v[i]] = -1; for (int i : s) vis[i] = 0; return g.build(), g; } ull has(ull x, ull y) { if (not dir and x > y) swap(x, y); return x * N + y; } hash_map mp; int get_eid(ull x, ull y) { if (mp.size() == 0) { mp.build(N - 1); for (Z &&e : es) { ull x = e.f, y = e.to; ull k = has(x, y); mp[k] = e.id; } } return mp.get(has(x, y), -1); } graph rev() const { static_assert(dir); graph ng(N); for (Z &&[f, t, w, id] : es) ng.add(t, f, w, id); return ng; } }; #line 2 "YRS/ds/basic/retsu.hpp" TE struct retsu { int N, M; vc a; retsu(int N, int M, T bs = T()) : N(N), M(M), a(N * M, bs) {} T* operator[](int i) { return a.data() + i * M; } const T* operator[](int i) const { return a.data() + i * M; } void fill(T x) { std::fill(all(a), x); } vc> to_vector() const { vector res(N, vc(M)); FOR(i, N) FOR(k, M) res[i][k] = a[i * M + k]; return res; } }; TE istream &operator>>(istream &I, retsu &a) { for (Z &e : a.a) I >> e; return I; } TE ostream &operator<<(ostream &O, retsu &a) { FOR(i, a.N) FOR(k, a.M) O << a[i][k] << " \n"[k + 1 == a.M and i + 1 != a.N]; return O; } #ifdef FIO TE rd(retsu &a) { for (Z &e : a.a) rd(e); } TE wt(retsu &a) { FOR(i, a.N) { FOR(k, a.M) { if (k) wt(' '); wt(a[i][k]); } if (i != a.M) wt('\n'); } } #endif #line 5 "YRS/g/steiner.hpp" // in: (g, ps, p_w) // return {cost, vs, es} // O(3^kn + 2^k(N+M)log N), k: set size TE tuple, vc> steiner_tree( graph &g, vc S, vc w = {}) { assert(g.isok()); assert(not S.empty()); int N = g.N, M = g.M, K = len(S); if (w.empty()) w.assign(N, 0); vc> DP(1 << K, vc(N, inf)); FOR(i, N) DP[0][i] = w[i]; retsu fa(1 << K, N, -1); FOR(s, 1, 1 << K) { Z &dp = DP[s]; for (int b = s, k = lowbit(b); b; b -= 1 << k, k = lowbit(b)) { int x = S[k]; chmin(dp[x], DP[s ^ 1 << k][x]); } FOR_subset(t, s) { if (t == 0 or t == s) continue; FOR(x, N) if (chmin(dp[x], DP[t][x] + DP[s ^ t][x] - w[x])) fa[s][x] = 2 * t; } min_heap> q; FOR(i, N) q.eb(dp[i], i); while (not q.empty()) { Z [d, n] = pop(q); if (d != dp[n]) continue; for (Z &&e : g[n]) { if (chmin(dp[e.to], d + e.w + w[e.to])) { fa[s][e.to] = 2 * e.id + 1; q.eb(dp[e.to], e.to); } } } } vc vis(N), vis_e(M); vc v_to_k(N, -1); FOR(k, K) v_to_k[S[k]] = k; vc q; int rt = min_element(all(DP.back())) - DP.back().begin(); q.ep((1 << K) - 1, rt); vis[rt] = 1; while (not q.empty()) { Z [s, n] = pop(q); if (s == 0) continue; if (fa[s][n] == -1) { int k = v_to_k[n]; assert(k != -1 and s >> k & 1); q.ep(s ^ 1 << k, n); continue; } else if (fa[s][n] & 1) { int i = fa[s][n] >> 1; Z &e = g.es[i]; int w = n ^ e.f ^ e.to; vis[w] = vis_e[i] = 1; q.ep(s, w); continue; } else { int t = fa[s][n] >> 1; q.ep(t, n), q.ep(s ^ t, n); } } vc vs, es; FOR(i, N) if (vis[i]) vs.ep(i); FOR(i, M) if (vis_e[i]) es.ep(i); T s = 0; for (int i : vs) s += w[i]; for (int i : es) s += g.es[i].w; assert(s == DP.back()[rt]); return {s, vs, es}; } #line 2 "YRS/ds/unionfind/dsu.hpp" struct dsu { int c; vc fa; dsu(int N = 0) : c(N), fa(N, -1) {} int f(int x) { while (fa[x] >= 0) { int p = fa[fa[x]]; if (p < 0) return fa[x]; x = fa[x] = p; } return x; } int operator[](int x) { return f(x); } bool merge(int x, int y) { x = f(x), y = f(y); if (x == y) return false; if (fa[x] > fa[y]) swap(x, y); fa[x] += fa[y]; fa[y] = x; --c; return true; } // set fa[y] = x; bool set(int x, int y) { x = f(x), y = f(y); if (x == y) return false; fa[x] += fa[y]; fa[y] = x; --c; return true; } int size(int x) { return -fa[f(x)]; } int count() const { return c; } bool same(int x, int y) { return f(x) == f(y); } void build(int N) { fa.assign(N, -1), c = N; } void reset() { std::fill(all(fa), -1), c = len(fa); } vc> get_group() { int N = len(fa); vc> v(N), s; FOR(i, N) v[f(i)].ep(i); FOR(i, N) if (not v[i].empty()) s.ep(v[i]); return s; } void pr() { const int N = len(fa); vc res(N); FOR(i, N) res[i] = f(i); print("fa:", res); } }; #line 8 "No_114_\u9060\u3044\u672a\u6765.cpp" #define tests 0 #define fl 0 #define DB 10 void Yorisou() { INT(N, M, K); graph g(N); g.read_graph<1>(M); VEC(int, s, K); FOR(i, K) --s[i]; if (len(s) <= 15) return print(get<0>(steiner_tree(g, s))); vc is(N); for (int x : s) is[x] = 1; vc n; FOR(i, N) if (is[i] == 0) n.ep(i); int sz = len(n), ans = inf; Z es = g.es; sort(es, [&](Z &a, Z &b) { return a.w < b.w; }); FOR(m, 1 << sz) { fill(all(is), 0); FOR(i, sz) if (m >> i & 1) is[n[i]] = 1; for (int x : s) is[x] = 1; dsu g(N); int sm = 0; for (Z &&e : es) { if (is[e.f] and is[e.to] and g.merge(e.f, e.to)) sm += e.w; if (sm >= ans) break; } bool o = 1; FOR(i, 1, K) o &= g[s[i]] == g[s[i - 1]]; if (o) chmin(ans, sm); } print(ans); } #line 1 "YRS/aa/main.hpp" int main() { cin.tie(nullptr)->sync_with_stdio(0); int T = 1; if (fl) cerr.tie(0); if (tests and not fl) IN(T); for (int i = 0; i < T or fl; ++i) { Yorisou(); if (fl and i % DB == 0) cerr << "Case: " << i << '\n'; } return 0; } #line 43 "No_114_\u9060\u3044\u672a\u6765.cpp"