結果

問題 No.957 植林
ユーザー PachicobuePachicobue
提出日時 2019-12-20 22:18:47
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,482 ms / 2,000 ms
コード長 8,555 bytes
コンパイル時間 2,611 ms
コンパイル使用メモリ 219,880 KB
実行使用メモリ 12,928 KB
最終ジャッジ日時 2024-07-18 00:25:25
合計ジャッジ時間 34,384 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 113 ms
11,264 KB
testcase_04 AC 130 ms
10,752 KB
testcase_05 AC 61 ms
11,520 KB
testcase_06 AC 141 ms
11,904 KB
testcase_07 AC 105 ms
10,880 KB
testcase_08 AC 28 ms
11,520 KB
testcase_09 AC 26 ms
11,520 KB
testcase_10 AC 29 ms
11,648 KB
testcase_11 AC 28 ms
11,520 KB
testcase_12 AC 32 ms
11,392 KB
testcase_13 AC 18 ms
9,600 KB
testcase_14 AC 24 ms
11,904 KB
testcase_15 AC 21 ms
11,264 KB
testcase_16 AC 18 ms
9,600 KB
testcase_17 AC 22 ms
11,008 KB
testcase_18 AC 931 ms
11,392 KB
testcase_19 AC 1,006 ms
11,520 KB
testcase_20 AC 1,062 ms
11,520 KB
testcase_21 AC 1,107 ms
12,032 KB
testcase_22 AC 1,200 ms
12,160 KB
testcase_23 AC 1,289 ms
12,544 KB
testcase_24 AC 1,385 ms
12,416 KB
testcase_25 AC 1,428 ms
12,544 KB
testcase_26 AC 1,415 ms
12,416 KB
testcase_27 AC 1,431 ms
12,416 KB
testcase_28 AC 1,482 ms
12,416 KB
testcase_29 AC 1,467 ms
12,416 KB
testcase_30 AC 1,420 ms
12,544 KB
testcase_31 AC 963 ms
11,392 KB
testcase_32 AC 991 ms
11,520 KB
testcase_33 AC 1,064 ms
11,648 KB
testcase_34 AC 1,119 ms
11,904 KB
testcase_35 AC 1,231 ms
12,288 KB
testcase_36 AC 1,353 ms
12,416 KB
testcase_37 AC 1,372 ms
12,288 KB
testcase_38 AC 1,450 ms
12,672 KB
testcase_39 AC 1,455 ms
12,544 KB
testcase_40 AC 1,420 ms
12,544 KB
testcase_41 AC 14 ms
12,800 KB
testcase_42 AC 14 ms
12,928 KB
testcase_43 AC 30 ms
12,672 KB
testcase_44 AC 31 ms
12,544 KB
testcase_45 AC 1 ms
6,940 KB
testcase_46 AC 2 ms
6,944 KB
testcase_47 AC 2 ms
6,944 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
// created [2019/12/20] 21:56:21
#pragma GCC diagnostic ignored "-Wsign-compare"
#pragma GCC diagnostic ignored "-Wsign-conversion"

using i32   = int32_t;
using i64   = int64_t;
using u32   = uint32_t;
using u64   = uint64_t;
using uint  = unsigned int;
using usize = std::size_t;
using ll    = long long;
using ull   = unsigned long long;
using ld    = long double;
template<typename T, usize n>
using arr = T (&)[n];
template<typename T, usize n>
using c_arr = const T (&)[n];
template<typename T> constexpr T popcount(const T u) { return u ? static_cast<T>(__builtin_popcountll(static_cast<u64>(u))) : static_cast<T>(0); }
template<typename T> constexpr T log2p1(const T u) { return u ? static_cast<T>(64 - __builtin_clzll(static_cast<u64>(u))) : static_cast<T>(0); }
template<typename T> constexpr T msbp1(const T u) { return log2p1(u); }
template<typename T> constexpr T lsbp1(const T u) { return __builtin_ffsll(u); }
template<typename T> constexpr T clog(const T u) { return u ? log2p1(u - 1) : static_cast<T>(u); }
template<typename T> constexpr bool ispow2(const T u) { return u and (static_cast<u64>(u) & static_cast<u64>(u - 1)) == 0; }
template<typename T> constexpr T ceil2(const T u) { return static_cast<T>(1) << clog(u); }
template<typename T> constexpr T floor2(const T u) { return u == 0 ? static_cast<T>(0) : static_cast<T>(1) << (log2p1(u) - 1); }
template<typename T> constexpr bool btest(const T mask, const usize ind) { return static_cast<bool>((static_cast<u64>(mask) >> ind) & static_cast<u64>(1)); }
template<typename T> void bset(T& mask, const usize ind) { mask |= (static_cast<T>(1) << ind); }
template<typename T> void breset(T& mask, const usize ind) { mask &= ~(static_cast<T>(1) << ind); }
template<typename T> void bflip(T& mask, const usize ind) { mask ^= (static_cast<T>(1) << ind); }
template<typename T> void bset(T& mask, const usize ind, const bool b) { (b ? bset(mask, ind) : breset(mask, ind)); }
template<typename T> constexpr T bcut(const T mask, const usize ind) { return ind == 0 ? static_cast<T>(0) : static_cast<T>((static_cast<u64>(mask) << (64 - ind)) >> (64 - ind)); }
template<typename T> bool chmin(T& a, const T& b) { return (a > b ? a = b, true : false); }
template<typename T> bool chmax(T& a, const T& b) { return (a < b ? a = b, true : false); }
constexpr unsigned int mod                  = 1000000007;
template<typename T> constexpr T inf_v      = std::numeric_limits<T>::max() / 4;
template<typename Real> constexpr Real pi_v = Real{3.141592653589793238462643383279502884};
auto mfp = [](auto&& f) { return [=](auto&&... args) { return f(f, std::forward<decltype(args)>(args)...); }; };

template<typename T>
T in()
{
    T v;
    return std::cin >> v, v;
}
template<typename T, typename Uint, usize n, usize i>
T in_v(typename std::enable_if<(i == n), c_arr<Uint, n>>::type) { return in<T>(); }
template<typename T, typename Uint, usize n, usize i>
auto in_v(typename std::enable_if<(i < n), c_arr<Uint, n>>::type& szs)
{
    const usize s = (usize)szs[i];
    std::vector<decltype(in_v<T, Uint, n, i + 1>(szs))> ans(s);
    for (usize j = 0; j < s; j++) { ans[j] = in_v<T, Uint, n, i + 1>(szs); }
    return ans;
}
template<typename T, typename Uint, usize n>
auto in_v(c_arr<Uint, n> szs) { return in_v<T, Uint, n, 0>(szs); }
template<typename... Types>
auto in_t() { return std::tuple<std::decay_t<Types>...>{in<Types>()...}; }
struct io_init
{
    io_init()
    {
        std::cin.tie(nullptr), std::ios::sync_with_stdio(false);
        std::cout << std::fixed << std::setprecision(20);
    }
    void clear()
    {
        std::cin.tie(), std::ios::sync_with_stdio(true);
    }
} io_setting;

template<typename T>
int out(const T& v) { return std::cout << v, 0; }
template<typename T>
int out(const std::vector<T>& v)
{
    for (usize i = 0; i < v.size(); i++) {
        if (i > 0) { std::cout << ' '; }
        out(v[i]);
    }
    return std::cout << "\n", 0;
}
template<typename T1, typename T2>
int out(const std::pair<T1, T2>& v) { return out(v.first), std::cout << ' ', out(v.second), 0; }
template<typename T, typename... Args>
int out(const T& v, const Args... args) { return out(v), std::cout << ' ', out(args...), 0; }
template<typename... Args>
int outln(const Args... args) { return out(args...), std::cout << '\n', 0; }
template<typename... Args>
void outel(const Args... args) { return out(args...), std::cout << std::endl, 0; }
#    define SHOW(...) static_cast<void>(0)
constexpr ull TEN(const usize n) { return n == 0 ? 1ULL : TEN(n - 1) * 10ULL; }

template<typename T, typename Uint, usize n, usize i>
auto make_v(typename std::enable_if<(i == n), c_arr<Uint, n>>::type, const T& v = T{}) { return v; }
template<typename T, typename Uint, usize n, usize i>
auto make_v(typename std::enable_if<(i < n), c_arr<Uint, n>>::type szs, const T& v = T{})
{
    const usize s = (usize)szs[i];
    return std::vector<decltype(make_v<T, Uint, n, i + 1>(szs, v))>(s, make_v<T, Uint, n, i + 1>(szs, v));
}
template<typename T, typename Uint, usize n>
auto make_v(c_arr<Uint, n> szs, const T& t = T{}) { return make_v<T, Uint, n, 0>(szs, t); }


template<typename Cap>
struct flow
{
    using capacity_type = Cap;
    flow(const usize v) : sz(v), edges(v) {}
    void add_edge(const usize from, const usize to, const capacity_type cap) { edges[from].push_back({to, edges[to].size(), cap, false}), edges[to].push_back({from, edges[from].size() - 1, 0, true}); }
    struct edge
    {
        usize to, rev;
        capacity_type cap;
        const bool is_rev;
    };
    const std::vector<edge>& operator[](const usize i) const { return edges[i]; }
    std::vector<edge>& operator[](const usize i) { return edges[i]; }
    friend std::ostream& operator<<(std::ostream& os, const flow& f)
    {
        os << "[\n";
        for (usize i = 0; i < f.size(); i++) {
            for (const auto& e : f[i]) {
                if (not e.is_rev) { os << i << "->" << e.to << ":"
                                       << e.cap << "\n"; }
            }
        }
        return (os << "]\n");
    }
    usize size() const { return sz; }

private:
    const usize sz;
    std::vector<std::vector<edge>> edges;
};
template<typename Cap>
Cap dinic(flow<Cap>& flow, const usize s, const usize t, const Cap max_cap = inf_v<Cap>)
{
    const usize sz = flow.size();
    std::vector<usize> level(sz, sz), iter(sz, 0);
    auto bfs = [&, sz](const Cap base) {
        std::fill(level.begin(), level.end(), sz);
        std::queue<usize> q;
        q.push(s), level[s] = 0;
        while (not q.empty()) {
            const usize s = q.front();
            q.pop();
            for (const auto& e : flow[s]) {
                if (e.cap < base or level[e.to] != sz) { continue; }
                level[e.to] = level[s] + 1, q.push(e.to);
            }
        }
    };
    auto dfs = [&, sz](auto&& self, const usize s, const Cap base, const Cap f) -> Cap {
        if (s == t) { return f; }
        Cap ans = 0;
        for (usize& i = iter[s]; i < flow[s].size(); i++) {
            auto& e = flow[s][i];
            if (e.cap < base or level[e.to] == sz or level[s] >= level[e.to]) { continue; }
            const Cap d = self(self, e.to, base, std::min(f - ans, e.cap));
            if (d == 0) { continue; }
            e.cap -= d, flow[e.to][e.rev].cap += d, ans += d;
            if (f - ans < base) { break; }
        }
        return ans;
    };
    Cap f = 0;
    for (Cap base = floor2(max_cap); base >= 1;) {
        bfs(base);
        if (level[t] == sz) {
            base >>= 1;
            continue;
        }
        std::fill(iter.begin(), iter.end(), 0);
        const Cap df = dfs(dfs, s, base, inf_v<Cap>);
        f += df;
    }
    return f;
}
int main()
{
    const auto [H, W] = in_t<int, int>();
    const auto G      = in_v<ll>({H, W});
    const auto R      = in_v<ll>({H});
    const auto C      = in_v<ll>({W});
    ll plus           = 0;
    flow<ll> f(H + W + 2);
    const int s = H + W;
    const int t = H + W + 1;
    for (int i = 0; i < H; i++) {
        ll g = 0;
        for (int j = 0; j < W; j++) { g += G[i][j]; }
        if (R[i] >= g) {
            plus += R[i] - g;
        } else {
            f.add_edge(s, i, g - R[i]);
        }
    }
    for (int i = 0; i < H; i++) {
        for (int j = 0; j < W; j++) { f.add_edge(i, j + H, G[i][j]); }
    }
    ll F = 0;
    for (int j = 0; j < W; j++) {
        F -= C[j];
        f.add_edge(j + H, t, C[j]);
    }
    F += dinic(f, s, t, inf_v<ll>);
    outln(plus - F);
    return 0;
}
0