// BEGIN: ../sakumon/bonsai/minimum_cut/sol_m1une.cpp #line 1 "..::sakumon::bonsai::minimum_cut::sol_m1une.cpp" // BEGIN: pch.hpp #line 3 "pch.hpp" #if defined(__GNUC__) && !defined(__clang__) #pragma GCC optimize("O3") #pragma GCC optimize("unroll-loops") #endif #define dump(...) #define CPP_DUMP_SET_OPTION(...) #define CPP_DUMP_DEFINE_EXPORT_OBJECT(...) #define CPP_DUMP_DEFINE_EXPORT_ENUM(...) #define CPP_DUMP_DEFINE_DANGEROUS_EXPORT_OBJECT(...) // BEGIN: template.hpp #line 3 "template.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 #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include // BEGIN: utilities/fast_io.hpp #line 3 "utilities::fast_io.hpp" #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include namespace m1une { namespace utilities { namespace internal { // Detect std::begin(x), std::end(x). template struct is_range : std::false_type {}; template struct is_range())), decltype(std::end(std::declval())) >> : std::true_type {}; template inline constexpr bool is_range_v = is_range::value; template using range_reference_t = decltype(*std::begin(std::declval())); template using range_value_t = std::remove_cv_t>>; template struct range_stored_value { using type = range_value_t; }; template struct range_stored_value>::value_type>> { using type = typename std::remove_cv_t>::value_type; }; template using range_stored_value_t = typename range_stored_value::type; // Treat strings and C strings as scalar output objects, not as ranges. template struct is_char_array : std::false_type {}; template struct is_char_array : std::bool_constant, char>> {}; template struct is_string_like : std::bool_constant< std::is_same_v, std::string> || std::is_same_v, const char*> || std::is_same_v, char*> || is_char_array>::value > {}; template inline constexpr bool is_string_like_v = is_string_like::value; // ModInt-like type: x.val() is printable, and x can be assigned from long long. template struct has_val_method : std::false_type {}; template struct has_val_method().val())>> : std::true_type {}; template inline constexpr bool has_val_method_v = has_val_method::value; template struct has_static_mod_raw : std::false_type {}; template struct has_static_mod_raw< T, std::void_t()))>> : std::true_type {}; template inline constexpr bool has_static_mod_raw_v = has_static_mod_raw::value; // libstdc++ before GCC 16 does not classify __int128 as an integral type in // strict ISO modes such as -std=c++23. Keep the fast-I/O interface independent // of that implementation detail. template inline constexpr bool is_integral_v = std::is_integral_v || std::is_same_v, __int128_t> || std::is_same_v, __uint128_t>; template inline constexpr bool is_signed_v = std::is_signed_v || std::is_same_v, __int128_t>; template struct make_unsigned { using type = std::make_unsigned_t; }; template <> struct make_unsigned<__int128_t> { using type = __uint128_t; }; template <> struct make_unsigned<__uint128_t> { using type = __uint128_t; }; template using make_unsigned_t = typename make_unsigned>::type; } // namespace internal struct FastInput { static constexpr int buffer_size = 1 << 20; private: std::FILE* _stream; char _buffer[buffer_size]; int _position; int _length; int _file_descriptor; bool _streaming; bool refill() { _position = 0; if (_streaming) { ssize_t length; do { length = ::read(_file_descriptor, _buffer, buffer_size); } while (length < 0 && errno == EINTR); if (length <= 0) { _length = 0; return false; } _length = int(length); } else { _length = int(std::fread(_buffer, 1, buffer_size, _stream)); } return _length != 0; } template bool read_integer_from_stream(T& value) { if (!skip_spaces()) return false; int c = read_char_raw(); bool negative = false; if (c == '-') { negative = true; c = read_char_raw(); } if constexpr (internal::is_signed_v) { T result = 0; while ('0' <= c && c <= '9') { result = negative ? result * 10 - (c - '0') : result * 10 + (c - '0'); c = read_char_raw(); } value = result; } else { T result = 0; while ('0' <= c && c <= '9') { result = result * 10 + T(c - '0'); c = read_char_raw(); } value = negative ? T(0) - result : result; } return true; } bool prepare_number() { if (_length - _position >= 64) return true; const int remaining = _length - _position; if (remaining > 0) std::memmove(_buffer, _buffer + _position, remaining); const int added = int(std::fread(_buffer + remaining, 1, buffer_size - remaining, _stream)); _position = 0; _length = remaining + added; if (_length < buffer_size) _buffer[_length] = '\0'; return _length != 0; } public: explicit FastInput(std::FILE* stream = stdin) : _stream(stream), _position(0), _length(0), _file_descriptor(::fileno(stream)), _streaming([&] { struct stat status; return _file_descriptor >= 0 && ::fstat(_file_descriptor, &status) == 0 && !S_ISREG(status.st_mode); }()) {} FastInput(const FastInput&) = delete; FastInput& operator=(const FastInput&) = delete; int read_char_raw() { if (_position == _length && !refill()) return EOF; return _buffer[_position++]; } bool skip_spaces() { int c = read_char_raw(); while (c != EOF && c <= ' ') c = read_char_raw(); if (c == EOF) return false; --_position; return true; } bool read(char& value) { if (!skip_spaces()) return false; value = char(read_char_raw()); return true; } bool read(std::string& value) { if (!skip_spaces()) return false; value.clear(); while (true) { const int begin = _position; while (_position < _length && static_cast(_buffer[_position]) > ' ') { ++_position; } value.append(_buffer + begin, _position - begin); if (_position < _length) { ++_position; return true; } if (!refill()) return true; } } bool read(bool& value) { int x; if (!read(x)) return false; value = x != 0; return true; } template std::enable_if_t< internal::is_integral_v && !std::is_same_v, bool> && !std::is_same_v, char>, bool > read(T& value) { if (_streaming) return read_integer_from_stream(value); if (!prepare_number()) return false; int c = static_cast(_buffer[_position++]); while (c <= ' ') c = static_cast(_buffer[_position++]); bool negative = false; if (c == '-') { negative = true; c = static_cast(_buffer[_position++]); } if constexpr (internal::is_signed_v) { T result = 0; while ('0' <= c && c <= '9') { const int first = c - '0'; const int second = static_cast(_buffer[_position]) - '0'; if (0 <= second && second <= 9) { result = negative ? result * 100 - (first * 10 + second) : result * 100 + (first * 10 + second); ++_position; } else { result = negative ? result * 10 - first : result * 10 + first; } c = static_cast(_buffer[_position++]); } value = result; } else { T result = 0; while ('0' <= c && c <= '9') { const unsigned first = unsigned(c - '0'); const int second = static_cast(_buffer[_position]) - '0'; if (0 <= second && second <= 9) { result = result * 100 + T(first * 10 + unsigned(second)); ++_position; } else { result = result * 10 + T(first); } c = static_cast(_buffer[_position++]); } value = negative ? T(0) - result : result; } if (_position > _length) _position = _length; return true; } template std::enable_if_t, bool> read(T& value) { if (!skip_spaces()) return false; int c = read_char_raw(); bool negative = false; if (c == '-' || c == '+') { negative = c == '-'; c = read_char_raw(); } long double result = 0; while ('0' <= c && c <= '9') { result = result * 10 + (c - '0'); c = read_char_raw(); } if (c == '.') { long double place = 0.1L; c = read_char_raw(); while ('0' <= c && c <= '9') { result += (c - '0') * place; place *= 0.1L; c = read_char_raw(); } } if (c == 'e' || c == 'E') { c = read_char_raw(); bool exponent_negative = false; if (c == '-' || c == '+') { exponent_negative = c == '-'; c = read_char_raw(); } int exponent = 0; while ('0' <= c && c <= '9') { exponent = exponent * 10 + (c - '0'); c = read_char_raw(); } long double scale = 1; long double power = 10; while (exponent > 0) { if (exponent & 1) scale *= power; power *= power; exponent >>= 1; } result = exponent_negative ? result / scale : result * scale; } value = static_cast(negative ? -result : result); return true; } template std::enable_if_t< internal::has_val_method_v && !internal::is_integral_v && !internal::is_range_v, bool > read(T& value) { long long x; if (!read(x)) return false; if constexpr (internal::has_static_mod_raw_v) { if (x >= 0 && uint64_t(x) < uint64_t(T::mod())) { value = T::raw(uint32_t(x)); } else { value = T(x); } } else { value = T(x); } return true; } template bool read(std::pair& value) { if (!read(value.first)) return false; return read(value.second); } template std::enable_if_t< internal::is_range_v && !internal::is_string_like_v, bool > read(Range& range) { using StoredValue = internal::range_stored_value_t; constexpr bool nested = internal::is_range_v && !internal::is_string_like_v; for (auto&& value : range) { if constexpr (std::is_same_v && !nested) { bool x; if (!read(x)) return false; value = x; } else { if (!read(value)) return false; } } return true; } template bool read(First& first, Second& second, Rest&... rest) { if (!read(first)) return false; return read(second, rest...); } template FastInput& operator>>(T& value) { if (!read(value)) std::abort(); return *this; } }; struct FastOutput { static constexpr int buffer_size = 1 << 20; private: inline static const auto digit_quads = [] { std::array result{}; for (int i = 0; i < 10000; i++) { int value = i; for (int j = 3; j >= 0; j--) { result[4 * i + j] = char('0' + value % 10); value /= 10; } } return result; }(); std::FILE* _stream; char _buffer[buffer_size]; int _position; int _precision; std::chars_format _float_format; char _range_separator; public: explicit FastOutput(std::FILE* stream = stdout) : _stream(stream), _position(0), _precision(6), _float_format(std::chars_format::general), _range_separator(' ') {} FastOutput(const FastOutput&) = delete; FastOutput& operator=(const FastOutput&) = delete; ~FastOutput() { flush(); } void flush() { if (_position != 0) { std::fwrite(_buffer, 1, _position, _stream); _position = 0; } std::fflush(_stream); } void write_char(char c) { if (_position == buffer_size) flush(); _buffer[_position++] = c; } void write(const char* s) { while (*s != '\0') write_char(*s++); } void write(const std::string& s) { std::size_t position = 0; while (position < s.size()) { if (_position == buffer_size) flush(); const std::size_t copied = std::min(buffer_size - _position, s.size() - position); std::memcpy(_buffer + _position, s.data() + position, copied); _position += int(copied); position += copied; } } void write(char c) { write_char(c); } void write(bool value) { write_char(value ? '1' : '0'); } template std::enable_if_t> write(T value) { char digits[128]; auto [end, error] = std::to_chars( digits, digits + sizeof(digits), value, _float_format, _precision ); if (error != std::errc()) std::abort(); for (const char* pointer = digits; pointer != end; pointer++) { write_char(*pointer); } } template std::enable_if_t< internal::is_integral_v && !std::is_same_v, bool> && !std::is_same_v, char> > write(T value) { using Raw = std::remove_cv_t; using Unsigned = internal::make_unsigned_t; Unsigned magnitude; if constexpr (internal::is_signed_v) { if (value < 0) { write_char('-'); magnitude = Unsigned(0) - Unsigned(value); } else { magnitude = Unsigned(value); } } else { magnitude = value; } if (magnitude == 0) { write_char('0'); return; } unsigned chunks[16]; int count = 0; while (magnitude >= 10000) { const Unsigned quotient = magnitude / 10000; chunks[count++] = unsigned(magnitude - quotient * 10000); magnitude = quotient; } if (_position > buffer_size - 64) flush(); const unsigned leading = unsigned(magnitude); const char* first = digit_quads.data() + 4 * leading; int skip = leading < 10 ? 3 : leading < 100 ? 2 : leading < 1000 ? 1 : 0; for (; skip < 4; skip++) _buffer[_position++] = first[skip]; while (count--) { const char* digits = digit_quads.data() + 4 * chunks[count]; std::memcpy(_buffer + _position, digits, 4); _position += 4; } } template std::enable_if_t< internal::has_val_method_v && !internal::is_integral_v && !internal::is_range_v > write(const T& value) { write(value.val()); } template void write(const std::pair& value) { write(value.first); write_char(' '); write(value.second); } template std::enable_if_t< internal::is_range_v && !internal::is_string_like_v > write(const Range& range) { using StoredValue = internal::range_stored_value_t; constexpr bool nested = internal::is_range_v && !internal::is_string_like_v; bool first = true; for (const auto& value : range) { if (!first) write_char(nested ? '\n' : _range_separator); first = false; if constexpr (std::is_same_v && !nested) { write(static_cast(value)); } else { write(value); } } } template void print(const First& first, const Rest&... rest) { write(first); ((write_char(' '), write(rest)), ...); } void println() { write_char('\n'); } void set_precision(int precision) { _precision = precision; } void set_fixed(int precision = 6) { _float_format = std::chars_format::fixed; _precision = precision; } void set_general(int precision = 6) { _float_format = std::chars_format::general; _precision = precision; } void set_range_separator(char separator) { _range_separator = separator; } template void println(const Args&... args) { print(args...); write_char('\n'); } template FastOutput& operator<<(const T& value) { write(value); return *this; } }; } // namespace utilities } // namespace m1une // END: utilities/fast_io.hpp #line 103 "template.hpp" using namespace std; namespace m1une { namespace template_io { inline utilities::FastInput& input() { static utilities::FastInput instance; return instance; } inline utilities::FastOutput& output() { static utilities::FastOutput instance; return instance; } } // namespace template_io } // namespace m1une using ll = long long; using u32 = unsigned int; using u64 = unsigned long long; using i128 = __int128; using u128 = unsigned __int128; #ifdef __SIZEOF_FLOAT128__ using f128 = __float128; #endif template constexpr T infty = 0; template <> constexpr int infty = 1'000'000'000; template <> constexpr ll infty = ll(infty) * infty * 2; template <> constexpr u32 infty = infty; template <> constexpr u64 infty = infty; template <> constexpr i128 infty = i128(infty) * infty; template <> constexpr double infty = infty; template <> constexpr long double infty = infty; using pi = pair; using pl = pair; using vi = vector; using vl = vector; template using vc = vector; template using vvc = vector>; using vvi = vvc; using vvl = vvc; template using vvvc = vector>; template using vvvvc = vector>; template using vvvvvc = vector>; template using pqg = std::priority_queue, greater>; template using umap = unordered_map; // template // using tree = __gnu_pbds::tree, // __gnu_pbds::rb_tree_tag, // __gnu_pbds::tree_order_statistics_node_update>; #define vv(type, name, h, ...) vector> name(h, vector(__VA_ARGS__)) #define vvv(type, name, h, w, ...) \ vector>> name(h, vector>(w, vector(__VA_ARGS__))) #define vvvv(type, name, a, b, c, ...) \ vector>>> name( \ a, vector>>(b, vector>(c, vector(__VA_ARGS__)))) #define overload4(a, b, c, d, e, ...) e #define overload3(a, b, c, d, ...) d // FOR(a) := for (ll _ = 0; _ < (ll)a; ++_) // FOR(i, a) := for (ll i = 0; i < (ll)a; ++i) // FOR(i, a, b) := for (ll i = a; i < (ll)b; ++i) // FOR(i, a, b, c) := for (ll i = a; i < (ll)b; i += (c)) // FOR_R(a) := for (ll i = (a) - 1; i >= 0; --i) // FOR_R(i, a) := for (ll i = (a) - 1; i >= 0; --i) // FOR_R(i, a, b) := for (ll i = (b) - 1; i >= (ll)a; --i) #define FOR1(a) for (ll _ = 0; _ < (ll)a; ++_) #define FOR2(i, a) for (ll i = 0; i < (ll)a; ++i) #define FOR3(i, a, b) for (ll i = a; i < (ll)b; ++i) #define FOR4(i, a, b, c) for (ll i = a; i < (ll)b; i += (c)) #define FOR1_R(a) for (ll i = (a) - 1; i >= 0; --i) #define FOR2_R(i, a) for (ll i = (a) - 1; i >= 0; --i) #define FOR3_R(i, a, b) for (ll i = (b) - 1; i >= (ll)a; --i) #define FOR(...) overload4(__VA_ARGS__, FOR4, FOR3, FOR2, FOR1)(__VA_ARGS__) #define FOR_R(...) overload3(__VA_ARGS__, FOR3_R, FOR2_R, FOR1_R)(__VA_ARGS__) #define FORI1(a) for (int _ = 0; _ < (int)a; ++_) #define FORI2(i, a) for (int i = 0; i < (int)a; ++i) #define FORI3(i, a, b) for (int i = a; i < (int)b; ++i) #define FORI4(i, a, b, c) for (int i = a; i < (int)b; i += (c)) #define FORI1_R(a) for (int i = (a) - 1; i >= 0; --i) #define FORI2_R(i, a) for (int i = (a) - 1; i >= 0; --i) #define FORI3_R(i, a, b) for (int i = (b) - 1; i >= (int)a; --i) #define FORI(...) overload4(__VA_ARGS__, FORI4, FORI3, FORI2, FORI1)(__VA_ARGS__) #define FORI_R(...) overload3(__VA_ARGS__, FORI3_R, FORI2_R, FORI1_R)(__VA_ARGS__) #define FOR_subset(t, s) for (int t = (s); t >= 0; t = (t == 0 ? -1 : (t - 1) & (s))) #define all(x) x.begin(), x.end() #define rall(x) x.rbegin(), x.rend() int popcnt(int x) { return __builtin_popcount(x); } int popcnt(u32 x) { return __builtin_popcount(x); } int popcnt(ll x) { return __builtin_popcountll(x); } int popcnt(u64 x) { return __builtin_popcountll(x); } int popcnt_mod_2(int x) { return __builtin_parity(x); } int popcnt_mod_2(u32 x) { return __builtin_parity(x); } int popcnt_mod_2(ll x) { return __builtin_parityll(x); } int popcnt_mod_2(u64 x) { return __builtin_parityll(x); } // (0, 1, 2, 3, 4) -> (-1, 0, 1, 1, 2) int topbit(int x) { return (x == 0 ? -1 : 31 - __builtin_clz(x)); } int topbit(u32 x) { return (x == 0 ? -1 : 31 - __builtin_clz(x)); } int topbit(ll x) { return (x == 0 ? -1 : 63 - __builtin_clzll(x)); } int topbit(u64 x) { return (x == 0 ? -1 : 63 - __builtin_clzll(x)); } // (0, 1, 2, 3, 4) -> (-1, 0, 1, 0, 2) int lowbit(int x) { return (x == 0 ? -1 : __builtin_ctz(x)); } int lowbit(u32 x) { return (x == 0 ? -1 : __builtin_ctz(x)); } int lowbit(ll x) { return (x == 0 ? -1 : __builtin_ctzll(x)); } int lowbit(u64 x) { return (x == 0 ? -1 : __builtin_ctzll(x)); } template T floor(T a, T b) { return a / b - (a % b && (a ^ b) < 0); } template T ceil(T x, T y) { return floor(x + y - 1, y); } template T bmod(T x, T y) { return x - y * floor(x, y); } template pair divmod(T x, T y) { T q = floor(x, y); return {q, x - q * y}; } template T POW(U x_, int n) { T x = x_; T ret = 1; while (n > 0) { if (n & 1) ret *= x; x *= x; n >>= 1; } return ret; } template T SUM(const vector& A) { T sm = 0; for (auto&& a : A) sm += a; return sm; } #define LB(c, x) distance((c).begin(), lower_bound(all(c), (x))) #define UB(c, x) distance((c).begin(), upper_bound(all(c), (x))) #define UNIQUE(x) sort(all(x)), x.erase(unique(all(x)), x.end()), x.shrink_to_fit() template inline bool chmax(T& a, const S& b) { return (a < b ? a = b, 1 : 0); } template inline bool chmin(T& a, const S& b) { return (a > b ? a = b, 1 : 0); } // ? は -1 vc s_to_vi(const string& S, char first_char) { vc A(S.size()); FOR(i, S.size()) { A[i] = (S[i] != '?' ? S[i] - first_char : -1); } return A; } template vector cumsum(vector& A, int off = 1) { int N = A.size(); vector B(N + 1); FOR(i, N) { B[i + 1] = B[i] + A[i]; } if (off == 0) B.erase(B.begin()); return B; } template vector argsort(const vector& A) { vector ids(A.size()); iota(all(ids), 0); sort(all(ids), [&](int i, int j) { return (A[i] == A[j] ? i < j : A[i] < A[j]); }); return ids; } // A[I[0]], A[I[1]], ... template vc rearrange(const vc& A, const vc& I) { vc B(I.size()); FOR(i, I.size()) B[i] = A[I[i]]; return B; } template constexpr auto min(T... a) { return min(initializer_list>{a...}); } template constexpr auto max(T... a) { return max(initializer_list>{a...}); } template bool scan(Ts&... values) { return m1une::template_io::input().read(values...); } template void print(const Ts&... values) { m1une::template_io::output().println(values...); } void YESNO(bool b) { m1une::template_io::output().println(b ? "YES" : "NO"); } void YesNo(bool b) { m1une::template_io::output().println(b ? "Yes" : "No"); } void YES() { m1une::template_io::output().println("YES"); } void NO() { m1une::template_io::output().println("NO"); } void Yes() { m1une::template_io::output().println("Yes"); } void No() { m1une::template_io::output().println("No"); } // END: template.hpp #line 29 "pch.hpp" // END: pch.hpp #line 2 "..::sakumon::bonsai::minimum_cut::sol_m1une.cpp" auto& fastin = m1une::template_io::input(); auto& fastout = m1une::template_io::output(); // BEGIN: graph/flow/max_flow.hpp #line 3 "graph::flow::max_flow.hpp" #include #include #include #include #include namespace m1une { namespace flow { template struct MaxFlow { struct Edge { int from; int to; Cap cap; Cap flow; }; private: struct InternalEdge { int to; int rev; Cap cap; }; struct Position { int from; int edge; }; int _n; std::vector _pos; std::vector> _g; Cap highest_label_preflow_push(int s, int t) { const int dead = 2 * _n; const int unreachable = _n + 1; std::vector excess(_n, Cap(0)); std::vector state(8 * std::size_t(_n) + 2); int* height = state.data(); int* height_count = height + _n; int* current = height_count + dead + 1; int* queue = current + _n; int* next = queue + _n; int* bucket_head = next + _n; std::vector active(_n, false); int highest = -1; long long work = 0; const long long arc_count = 2LL * static_cast(_pos.size()); const long long work_limit = std::max(1LL, 4 * arc_count + _n); auto activate = [&](int v) { if (v == s || v == t || active[v] || excess[v] == Cap(0) || height[v] >= dead) { return; } active[v] = true; next[v] = bucket_head[height[v]]; bucket_head[height[v]] = v; highest = std::max(highest, height[v]); }; auto rebuild_buckets = [&]() { std::fill(bucket_head, bucket_head + dead + 1, -1); std::fill(active.begin(), active.end(), false); highest = -1; for (int v = 0; v < _n; v++) activate(v); }; auto global_relabel = [&]() { std::fill(height, height + _n, unreachable); std::fill(height_count, height_count + dead + 1, 0); std::fill(current, current + _n, 0); int head = 0; int tail = 0; height[t] = 0; height[s] = _n; queue[tail++] = t; while (head != tail) { int v = queue[head++]; for (const auto& e : _g[v]) { if (e.to == s || height[e.to] != unreachable) continue; const auto& reverse = _g[e.to][e.rev]; if (reverse.cap == Cap(0)) continue; height[e.to] = height[v] + 1; queue[tail++] = e.to; } } for (int v = 0; v < _n; v++) height_count[height[v]]++; rebuild_buckets(); work = 0; }; auto gap = [&](int empty_height) { for (int v = 0; v < _n; v++) { if (v == s || v == t || height[v] <= empty_height || height[v] >= _n) { continue; } height_count[height[v]]--; height[v] = unreachable; height_count[height[v]]++; current[v] = 0; } rebuild_buckets(); }; auto relabel = [&](int v) -> bool { int old_height = height[v]; int new_height = dead; work += int(_g[v].size()); for (const auto& e : _g[v]) { if (e.cap != Cap(0)) { new_height = std::min(new_height, height[e.to] + 1); } } height_count[old_height]--; height[v] = std::min(new_height, dead); height_count[height[v]]++; current[v] = 0; if (old_height < _n && height_count[old_height] == 0) { gap(old_height); return true; } return false; }; auto push = [&](int v, InternalEdge& e) { Cap sent = std::min(excess[v], e.cap); bool was_zero = excess[e.to] == Cap(0); e.cap -= sent; _g[e.to][e.rev].cap += sent; excess[v] -= sent; excess[e.to] += sent; if (was_zero) activate(e.to); }; auto discharge = [&](int v) { while (excess[v] != Cap(0) && height[v] < dead) { if (current[v] == int(_g[v].size())) { if (relabel(v)) return; continue; } auto& e = _g[v][current[v]]; work++; if (e.cap != Cap(0) && height[v] == height[e.to] + 1) { push(v, e); } else { current[v]++; } } activate(v); }; for (auto& e : _g[s]) { if (e.to == s || e.cap == Cap(0)) continue; Cap sent = e.cap; e.cap = Cap(0); _g[e.to][e.rev].cap += sent; excess[e.to] += sent; } global_relabel(); while (highest >= 0) { if (bucket_head[highest] == -1) { highest--; continue; } int v = bucket_head[highest]; bucket_head[highest] = next[v]; if (!active[v] || height[v] != highest) continue; active[v] = false; discharge(v); if (work >= work_limit) global_relabel(); } return excess[t]; } public: MaxFlow() : MaxFlow(0) {} explicit MaxFlow(int n) : _n(n), _g(n) { assert(0 <= n); } int size() const { return _n; } int edge_count() const { return int(_pos.size()); } void reserve_edges(int edge_count) { assert(0 <= edge_count); _pos.reserve(edge_count); if (_n == 0 || edge_count == 0 || 2 * std::size_t(edge_count) < std::size_t(_n)) { return; } const std::size_t average_degree = (3 * std::size_t(edge_count) + std::size_t(_n) - 1) / std::size_t(_n); for (auto& edges : _g) edges.reserve(average_degree); } void reserve_edges(int edge_count, const std::vector& degrees) { assert(0 <= edge_count); assert(int(degrees.size()) == _n); _pos.reserve(edge_count); for (int v = 0; v < _n; v++) { assert(0 <= degrees[v]); _g[v].reserve(degrees[v]); } } int add_edge(int from, int to, Cap cap) { assert(0 <= from && from < _n); assert(0 <= to && to < _n); assert(Cap(0) <= cap); int id = int(_pos.size()); int from_id = int(_g[from].size()); int to_id = int(_g[to].size()); if (from == to) to_id++; _pos.push_back(Position{from, from_id}); _g[from].push_back(InternalEdge{to, to_id, cap}); _g[to].push_back(InternalEdge{from, from_id, Cap(0)}); return id; } int add_undirected_edge(int first, int second, Cap cap) { static_assert(std::numeric_limits::is_signed); assert(0 <= first && first < _n); assert(0 <= second && second < _n); assert(Cap(0) <= cap); assert(cap <= std::numeric_limits::max() / Cap(2)); int id = int(_pos.size()); int first_id = int(_g[first].size()); int second_id = int(_g[second].size()); if (first == second) second_id++; _pos.push_back(Position{first, ~first_id}); _g[first].push_back(InternalEdge{second, second_id, cap}); _g[second].push_back(InternalEdge{first, first_id, cap}); return id; } Edge get_edge(int i) const { assert(0 <= i && i < int(_pos.size())); const auto& position = _pos[i]; int from = position.from; bool undirected = position.edge < 0; int idx = undirected ? ~position.edge : position.edge; const auto& e = _g[from][idx]; const auto& re = _g[e.to][e.rev]; if (undirected) { return Edge{ from, e.to, (e.cap + re.cap) / Cap(2), (re.cap - e.cap) / Cap(2) }; } return Edge{from, e.to, e.cap + re.cap, re.cap}; } std::vector edges() const { std::vector result; result.reserve(_pos.size()); for (int i = 0; i < int(_pos.size()); i++) result.push_back(get_edge(i)); return result; } void change_edge(int i, Cap new_cap, Cap new_flow) { assert(0 <= i && i < int(_pos.size())); assert(Cap(0) <= new_cap); auto& position = _pos[i]; int from = position.from; bool undirected = position.edge < 0; int idx = undirected ? ~position.edge : position.edge; auto& e = _g[from][idx]; auto& re = _g[e.to][e.rev]; if (undirected) { assert(new_cap <= std::numeric_limits::max() / Cap(2)); assert(-new_cap <= new_flow && new_flow <= new_cap); e.cap = new_cap - new_flow; re.cap = new_cap + new_flow; } else { assert(Cap(0) <= new_flow && new_flow <= new_cap); e.cap = new_cap - new_flow; re.cap = new_flow; } } Cap max_flow(int s, int t) { assert(0 <= s && s < _n); assert(0 <= t && t < _n); assert(s != t); return highest_label_preflow_push(s, t); } Cap max_flow_push_relabel(int s, int t) { assert(0 <= s && s < _n); assert(0 <= t && t < _n); assert(s != t); return highest_label_preflow_push(s, t); } Cap max_flow_dinic(int s, int t) { return max_flow(s, t, std::numeric_limits::max()); } Cap max_flow(int s, int t, Cap flow_limit) { assert(0 <= s && s < _n); assert(0 <= t && t < _n); assert(s != t); std::vector work(3 * std::size_t(_n)); int* level = work.data(); int* iter = level + _n; int* queue = iter + _n; auto bfs = [&]() -> bool { std::fill(level, level + _n, -1); int head = 0; int tail = 0; level[s] = 0; queue[tail++] = s; while (head != tail) { int v = queue[head++]; for (const auto& e : _g[v]) { if (level[e.to] != -1 || e.cap == Cap(0)) continue; level[e.to] = level[v] + 1; if (e.to == t) return true; queue[tail++] = e.to; } } return level[t] != -1; }; auto dfs = [&](auto&& self, int v, Cap up) -> Cap { if (v == s) return up; Cap result = Cap(0); const int current_level = level[v]; auto& edges = _g[v]; const int edge_count = int(edges.size()); for (int& i = iter[v]; i < edge_count; i++) { auto& e = edges[i]; if (level[e.to] + 1 != current_level) continue; auto& reverse = _g[e.to][e.rev]; if (reverse.cap == Cap(0)) continue; Cap d = self( self, e.to, std::min(up - result, reverse.cap) ); if (d == Cap(0)) continue; e.cap += d; reverse.cap -= d; result += d; if (result == up) return result; } level[v] = _n; return result; }; Cap flow = 0; while (flow < flow_limit && bfs()) { std::fill(iter, iter + _n, 0); flow += dfs(dfs, t, flow_limit - flow); } return flow; } std::vector min_cut(int s) const { assert(0 <= s && s < _n); std::vector visited(_n, false); std::vector queue(_n); int head = 0; int tail = 0; visited[s] = true; queue[tail++] = s; while (head != tail) { int v = queue[head++]; for (const auto& e : _g[v]) { if (e.cap == Cap(0) || visited[e.to]) continue; visited[e.to] = true; queue[tail++] = e.to; } } return visited; } }; } // namespace flow } // namespace m1une // END: graph/flow/max_flow.hpp #line 7 "..::sakumon::bonsai::minimum_cut::sol_m1une.cpp" void solve() { ll n, m, s, t; scan(n, m, s, t); --s, --t; m1une::flow::MaxFlow mf(n); FOR(m) { ll u, v, c; scan(u, v, c); --u, --v; mf.add_edge(u, v, c); } ll ans = mf.max_flow_push_relabel(s, t); print(ans); } int main() { CPP_DUMP_SET_OPTION(max_line_width, 80); CPP_DUMP_SET_OPTION(log_label_func, cpp_dump::log_label::filename()); CPP_DUMP_SET_OPTION(enable_asterisk, true); int T = 1; // cin >> T; while (T--) solve(); return 0; } // END: ../sakumon/bonsai/minimum_cut/sol_m1une.cpp