結果
問題 | No.1678 Coin Trade (Multiple) |
ユーザー |
![]() |
提出日時 | 2021-09-10 21:40:40 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 278 ms / 5,000 ms |
コード長 | 3,810 bytes |
コンパイル時間 | 1,820 ms |
コンパイル使用メモリ | 114,392 KB |
最終ジャッジ日時 | 2025-01-24 09:58:44 |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 56 |
ソースコード
//#define NDEBUG #pragma warning(disable : 4146) #include <algorithm> #include <cassert> #include <cstddef> #include <cstdint> #include <iomanip> #include <iostream> #include <utility> #include <vector> namespace n91 { using i32 = std::int32_t; using i64 = std::int64_t; using u32 = std::uint32_t; using u64 = std::uint64_t; using isize = std::ptrdiff_t; using usize = std::size_t; using f64 = double; struct rep { struct itr { usize i; constexpr itr(const usize i) noexcept : i(i) {} void operator++() noexcept { ++i; } constexpr usize operator*() const noexcept { return i; } constexpr bool operator!=(const itr x) const noexcept { return i != x.i; } }; const itr f, l; constexpr rep(const usize f, const usize l) noexcept : f(std::min(f, l)), l(l) {} constexpr auto begin() const noexcept { return f; } constexpr auto end() const noexcept { return l; } }; struct revrep { struct itr { usize i; constexpr itr(const usize i) noexcept : i(i) {} void operator++() noexcept { --i; } constexpr usize operator*() const noexcept { return i; } constexpr bool operator!=(const itr x) const noexcept { return i != x.i; } }; const itr f, l; constexpr revrep(const usize f, const usize l) noexcept : f(l - 1), l(std::min(f, l) - 1) {} constexpr auto begin() const noexcept { return f; } constexpr auto end() const noexcept { return l; } }; template <class T> auto md_vec(const usize n, const T& value) { return std::vector<T>(n, value); } template <class... Args> auto md_vec(const usize n, Args... args) { return std::vector<decltype(md_vec(args...))>(n, md_vec(args...)); } template <class T> constexpr T difference(const T& a, const T& b) noexcept { return a < b ? b - a : a - b; } template <class T> void chmin(T& a, const T& b) noexcept { if (b < a) a = b; } template <class T> void chmax(T& a, const T& b) noexcept { if (a < b) a = b; } template <class F> class rec_lambda { F f; public: rec_lambda(F&& f_) : f(std::forward<F>(f_)) {} template <class... Args> auto operator()(Args &&... args) const { return f(*this, std::forward<Args>(args)...); } }; template <class T> T scan() { T ret; std::cin >> ret; return ret; } constexpr char eoln = '\n'; i64 floor_div(const i64 n, const i64 d) { assert(d != 0); return n / d - static_cast<i64>((n ^ d) < 0 && n % d != 0); } i64 ceil_div(const i64 n, const i64 d) { assert(d != 0); return n / d + static_cast<i64>((n ^ d) >= 0 && n % d != 0); } #ifdef N91_LOCAL #define OJ_LOCAL(a, b) b #else #define OJ_LOCAL(a, b) a #endif } // namespace n91 #include <atcoder/mincostflow> namespace n91 { void main_() { const usize n = scan<usize>(); const i64 k = scan<i64>(); atcoder::mcf_graph<int, long long> g(n); std::vector<i64> d(n, 0); std::vector<i64> a(n); for (const usize i : rep(0, n)) { std::cin >> a[i]; const usize m = scan<usize>(); std::vector<usize> b(m); if (i != 0) { chmin(d[i], d[i - 1]); } for (auto& e : b) { std::cin >> e; e -= 1; chmin(d[i], d[e] + a[e] - a[i]); } for (const auto& e : b) { g.add_edge(e, i, 1, a[e] - a[i] + d[e] - d[i]); } } for (const usize i : rep(1, n)) { g.add_edge(i - 1, i, k, d[i - 1] - d[i]); } auto [flow, cost] = g.flow(0, n - 1, k); assert(flow == k); cost += k * (d[n - 1] - d[0]); std::cout << -cost << eoln; } } // namespace n91 int main() { //* std::ios::sync_with_stdio(false); std::cin.tie(nullptr); //*/ std::cout << std::setprecision(20); n91::main_(); return 0; }