結果

問題 No.957 植林
ユーザー jelljell
提出日時 2019-12-21 00:00:08
言語 C++14
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 482 ms / 2,000 ms
コード長 13,814 bytes
コンパイル時間 1,836 ms
コンパイル使用メモリ 138,892 KB
実行使用メモリ 22,104 KB
最終ジャッジ日時 2023-09-25 04:07:50
合計ジャッジ時間 14,343 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,376 KB
testcase_01 AC 1 ms
4,380 KB
testcase_02 AC 1 ms
4,376 KB
testcase_03 AC 40 ms
18,396 KB
testcase_04 AC 36 ms
17,880 KB
testcase_05 AC 43 ms
18,760 KB
testcase_06 AC 42 ms
19,912 KB
testcase_07 AC 40 ms
18,192 KB
testcase_08 AC 31 ms
18,280 KB
testcase_09 AC 32 ms
19,372 KB
testcase_10 AC 35 ms
19,932 KB
testcase_11 AC 33 ms
18,792 KB
testcase_12 AC 31 ms
18,412 KB
testcase_13 AC 28 ms
17,984 KB
testcase_14 AC 34 ms
20,192 KB
testcase_15 AC 30 ms
17,788 KB
testcase_16 AC 28 ms
18,260 KB
testcase_17 AC 30 ms
18,428 KB
testcase_18 AC 340 ms
19,000 KB
testcase_19 AC 352 ms
18,384 KB
testcase_20 AC 360 ms
18,908 KB
testcase_21 AC 375 ms
20,268 KB
testcase_22 AC 406 ms
19,556 KB
testcase_23 AC 446 ms
20,412 KB
testcase_24 AC 449 ms
20,048 KB
testcase_25 AC 474 ms
21,928 KB
testcase_26 AC 460 ms
20,588 KB
testcase_27 AC 470 ms
22,104 KB
testcase_28 AC 470 ms
21,096 KB
testcase_29 AC 467 ms
20,772 KB
testcase_30 AC 459 ms
21,560 KB
testcase_31 AC 330 ms
19,112 KB
testcase_32 AC 356 ms
18,660 KB
testcase_33 AC 377 ms
18,956 KB
testcase_34 AC 389 ms
19,744 KB
testcase_35 AC 396 ms
20,772 KB
testcase_36 AC 424 ms
20,220 KB
testcase_37 AC 456 ms
19,956 KB
testcase_38 AC 482 ms
22,016 KB
testcase_39 AC 467 ms
20,884 KB
testcase_40 AC 463 ms
20,920 KB
testcase_41 AC 27 ms
20,852 KB
testcase_42 AC 28 ms
20,720 KB
testcase_43 AC 32 ms
20,820 KB
testcase_44 AC 35 ms
21,720 KB
testcase_45 AC 2 ms
4,376 KB
testcase_46 AC 2 ms
4,376 KB
testcase_47 AC 1 ms
4,376 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:373:1: 警告: ISO C++ では型の無い ‘main’ の宣言を禁止しています [-Wreturn-type]
  373 | main()
      | ^~~~

ソースコード

diff # 

#ifdef LOCAL
    #define _GLIBCXX_DEBUG
    #define __clock__
#else
    #pragma GCC optimize("Ofast")
    // #define NDEBUG
#endif
// #define __buffer_check__
#define __precision__ 10
#define iostream_untie true
#define debug_stream std::cerr

#include <algorithm>
#include <bitset>
#include <cassert>
#include <chrono>
#include <complex>
#include <cstring>
#include <deque>
#include <functional>
#include <iomanip>
#include <iostream>
#include <list>
#include <map>
#include <queue>
#include <random>
#include <set>
#include <stack>
#include <unordered_map>
#include <unordered_set>

#define all(v) std::begin(v), std::end(v)
#define rall(v) std::rbegin(v), std::rend(v)
#define odd(n) ((n) & 1)
#define even(n) (not __odd(n))
#define __popcount(n) __builtin_popcountll(n)
#define __clz32(n) __builtin_clz(n)
#define __clz64(n) __builtin_clzll(n)
#define __ctz32(n) __builtin_ctz(n)
#define __ctz64(n) __builtin_ctzll(n)

using i32 = int_least32_t; using i64 = int_least64_t; using u32 = uint_least32_t; using u64 = uint_least64_t;
using pii = std::pair<i32, i32>; using pll = std::pair<i64, i64>;
template <class T> using heap = std::priority_queue<T>;
template <class T> using rheap = std::priority_queue<T, std::vector<T>, std::greater<T>>;
template <class T> using hashset = std::unordered_set<T>;
template <class Key, class Value> using hashmap = std::unordered_map<Key, Value>;

namespace setting
{
    using namespace std::chrono;
    system_clock::time_point start_time, end_time;
    long long get_elapsed_time() { end_time = system_clock::now(); return duration_cast<milliseconds>(end_time - start_time).count(); }
    void print_elapsed_time() { debug_stream << "\n----- Exec time : " << get_elapsed_time() << " ms -----\n"; }
    void buffer_check()
    {
        char bufc;
        if(std::cin >> bufc) debug_stream << "\n\033[1;35mwarning\033[0m: buffer not empty.\n";
    }
    struct setupper
    {
        setupper()
        {
            if(iostream_untie) std::ios::sync_with_stdio(false), std::cin.tie(nullptr);
            std::cout << std::fixed << std::setprecision(__precision__);
    #ifdef stderr_path
            if(freopen(stderr_path, "a", stderr))
            {
                std::cerr << std::fixed << std::setprecision(__precision__);
            }
    #endif
    #ifdef stdout_path
            if(not freopen(stdout_path, "w", stdout))
            {
                freopen("CON", "w", stdout);
                debug_stream << "\n\033[1;35mwarning\033[0m: failed to open stdout file.\n";
            }
            std::cout << "";
    #endif
    #ifdef stdin_path
            if(not freopen(stdin_path, "r", stdin))
            {
                freopen("CON", "r", stdin);
                debug_stream << "\n\033[1;35mwarning\033[0m: failed to open stdin file.\n";
            }
    #endif
    #ifdef LOCAL
            debug_stream << "----- stderr at LOCAL -----\n\n";
            atexit(print_elapsed_time);
    #endif
    #ifdef __buffer_check__
            atexit(buffer_check);
    #endif
    #if defined(__clock__) || defined(LOCAL)
            start_time = system_clock::now();
    #endif
        }
    } __setupper; // struct setupper
} // namespace setting
#ifdef __clock__
class
{
    std::chrono::system_clock::time_point built_pt, last_pt; int built_ln, last_ln;
    std::string built_func, last_func; bool is_built = false;
  public:
    void build(int crt_ln, const std::string &crt_func)
    {
        is_built = true, last_pt = built_pt = std::chrono::system_clock::now(),  last_ln = built_ln = crt_ln, last_func = built_func = crt_func;
    }
    void set(int crt_ln, const std::string &crt_func)
    {
        if(is_built) last_pt = std::chrono::system_clock::now(), last_ln = crt_ln, last_func = crt_func;
        else debug_stream << "[ " << crt_ln << " : " << crt_func << " ] " << "myclock_t::set failed (yet to be built!)\n";
    }
    void get(int crt_ln, const std::string &crt_func)
    {
        if(is_built)
        {
            std::chrono::system_clock::time_point crt_pt(std::chrono::system_clock::now());
            long long diff = std::chrono::duration_cast<std::chrono::milliseconds>(crt_pt - last_pt).count();
            debug_stream << diff << " ms elapsed from" << " [ " << last_ln << " : " << last_func << " ]";
            if(last_ln == built_ln) debug_stream << " (when built)";
            debug_stream << " to" << " [ " << crt_ln << " : " << crt_func << " ]" << "\n";
            last_pt = built_pt, last_ln = built_ln, last_func = built_func;
        }
        else
        {
            debug_stream << "[ " << crt_ln << " : " << crt_func << " ] " << "myclock_t::get failed (yet to be built!)\n";
        }
    }
} myclock; // unnamed class
    #define build_clock() myclock.build(__LINE__, __func__)
    #define set_clock() myclock.set(__LINE__, __func__)
    #define get_clock() myclock.get(__LINE__, __func__)
#else
    #define build_clock() ((void)0)
    #define set_clock() ((void)0)
    #define get_clock() ((void)0)
#endif

namespace std
{
    // hash
    template <class T> size_t hash_combine(size_t seed, T const &key) { return seed ^ (hash<T>()(key) + 0x9e3779b9 + (seed << 6) + (seed >> 2)); }
    template <class T, class U> struct hash<pair<T, U>> { size_t operator()(pair<T, U> const &pr) const { return hash_combine(hash_combine(0, pr.first), pr.second); } };
    template <class tuple_t, size_t index = tuple_size<tuple_t>::value - 1> struct tuple_hash_calc { static size_t apply(size_t seed, tuple_t const &t) { return hash_combine(tuple_hash_calc<tuple_t, index - 1>::apply(seed, t), get<index>(t)); } };
    template <class tuple_t> struct tuple_hash_calc<tuple_t, 0> { static size_t apply(size_t seed, tuple_t const &t) { return hash_combine(seed, get<0>(t)); } };
    template <class... T> struct hash<tuple<T...>> { size_t operator()(tuple<T...> const &t) const { return tuple_hash_calc<tuple<T...>>::apply(0, t); } };
    // iostream
    template <class T, class U> istream &operator>>(istream &is, pair<T, U> &p) { return is >> p.first >> p.second; }
    template <class T, class U> ostream &operator<<(ostream &os, const pair<T, U> &p) { return os << p.first << ' ' << p.second; }
    template <class tuple_t, size_t index> struct tupleos { static ostream &apply(ostream &os, const tuple_t &t) { tupleos<tuple_t, index - 1>::apply(os, t); return os << ' ' << get<index>(t); } };
    template <class tuple_t> struct tupleos<tuple_t, 0> { static ostream &apply(ostream &os, const tuple_t &t) { return os << get<0>(t); } };
    template <class... T> ostream &operator<<(ostream &os, const tuple<T...> &t)  { return tupleos<tuple<T...>, tuple_size<tuple<T...>>::value - 1>::apply(os, t); }
    template <> ostream &operator<<(ostream &os, const tuple<> &t) { return os; }
    template <class Container, typename Value = typename Container::value_type, enable_if_t<!is_same<decay_t<Container>, string>::value, nullptr_t> = nullptr>
    istream& operator>>(istream& is, Container &cont) { for(auto&& e : cont) is >> e; return is; }
    template <class Container, typename Value = typename Container::value_type, enable_if_t<!is_same<decay_t<Container>, string>::value, nullptr_t> = nullptr>
    ostream& operator<<(ostream& os, const Container &cont) { bool flag = 1; for(auto&& e : cont) flag ? flag = 0 : (os << ' ', 0), os << e; return os; }
} // namespace std

#ifdef LOCAL
    #define dump(...)                                                              \
        debug_stream << "[ " << __LINE__ << " : " << __FUNCTION__ << " ]\n",       \
            dump_func(#__VA_ARGS__, __VA_ARGS__)
    template <class T> void dump_func(const char *ptr, const T &x)
    {
        debug_stream << '\t';
        for(char c = *ptr; c != '\0'; c = *++ptr) if(c != ' ') debug_stream << c;
        debug_stream << " : " << x << '\n';
    }
    template <class T, class... rest_t> void dump_func(const char *ptr, const T &x, rest_t... rest)
    {
        debug_stream << '\t';
        for(char c = *ptr; c != ','; c = *++ptr) if(c != ' ') debug_stream << c;
        debug_stream << " : " << x << ",\n"; dump_func(++ptr, rest...);
    }
#else
    #define dump(...) ((void)0)
#endif
template <class P> void read_range(P __first, P __second) { for(P i = __first; i != __second; ++i) std::cin >> *i; }
template <class P> void write_range(P __first, P __second) { for(P i = __first; i != __second; std::cout << (++i == __second ? '\n' : ' ')) std::cout << *i; }

// substitue y for x if x > y.
template <class T> inline bool sbmin(T &x, const T &y) { return x > y ? x = y, true : false; }
// substitue y for x if x < y.
template <class T> inline bool sbmax(T &x, const T &y) { return x < y ? x = y, true : false; }
// binary search.
i64 bin(const std::function<bool(i64)> &pred, i64 ok, i64 ng)
{
    while(std::abs(ok - ng) > 1) { i64 mid = (ok + ng) / 2; (pred(mid) ? ok : ng) = mid; }
    return ok;
}
double bin(const std::function<bool(double)> &pred, double ok, double ng, const double eps)
{
    while(std::abs(ok - ng) > eps) { double mid = (ok + ng) / 2; (pred(mid) ? ok : ng) = mid; }
    return ok;
}
// be careful that val is type-sensitive.
template <class T, class A, size_t N> void init(A (&array)[N], const T &val) { std::fill((T *)array, (T *)(array + N), val); }
// reset all bits.
template <class A> void reset(A &array) { memset(array, 0, sizeof(array)); }


/* The main code follows. */

// Base class
template <class cap_t, class cost_t>
struct Flow
{
    struct edge_t
    {
        size_t from, to; cap_t cap; cost_t cost; size_t rev;
        edge_t(size_t _from, size_t _to, cap_t _cap, cost_t _cost, size_t _rev) : from(_from), to(_to), cap(_cap), cost(_cost), rev(_rev) {}
    }; // struct edge_t

  protected:
    size_t V;
    std::vector<std::vector<edge_t>> adj;

  public:
    Flow(size_t _V) : V(_V), adj(_V) {}

    void add_edge(size_t from, size_t to, cap_t cap, cost_t cost = cost_t(0))
    {
        adj[from].emplace_back(from, to, cap, cost, adj[to].size());
        adj[to].emplace_back(to, from, 0, -cost, adj[from].size() - 1);
    }
}; // struct Flow

template <class cap_t>
class Dinic : public Flow<cap_t, cap_t>
{
    using edge_t = typename Flow<cap_t, cap_t>::edge_t;
    using Flow<cap_t, cap_t>::V;
    using Flow<cap_t, cap_t>::adj;

    std::vector<size_t> level, itr;

    bool bfs(size_t s, size_t t)
    {
        fill(level.begin(), level.end(), ~0);
        std::queue<size_t> que;
        que.emplace(s);
        level[s] = 0;
        while(!que.empty())
        {
            size_t v = que.front(); que.pop();
            for(const edge_t &e : adj[v])
            {
                if(e.cap > cap_t(0) && not ~level[e.to])
                {
                    level[e.to] = level[v] + 1;
                    que.emplace(e.to);
                }
            }
        }
        return ~level[t];
    }

    cap_t dfs(size_t v, size_t t, cap_t f)
    {
        if(v == t) return f;
        cap_t res{0};
        while(itr[v] < adj[v].size())
        {
            edge_t &e = adj[v][itr[v]];
            if(e.cap > cap_t{0} && level[v] < level[e.to])
            {
                cap_t d = dfs(e.to, t, std::min(f, e.cap));
                e.cap -= d;
                adj[e.to][e.rev].cap += d;
                res += d;
                if((f -= d) == cap_t{0}) break;
            }
            ++itr[v];
        }
        return res;
    }

  public:
    Dinic(size_t V) : Flow<cap_t, cap_t>::Flow(V), level(V), itr(V) {}

    size_t size() const { return V; }
    std::vector<edge_t> &operator[](size_t v) { return adj[v]; }

    cap_t max_flow(size_t s, size_t t)
    {
        cap_t res = 0, f;
        while(bfs(s, t))
        {
            fill(itr.begin(), itr.end(), 0);
            while((f = dfs(s, t, std::numeric_limits<cap_t>::max())) > cap_t(0)) res += f;
        }
        return res;
    }

    class cut_t
    {
        size_t V; bool* const data;
        friend class Dinic;
      public:
        cut_t(size_t _V) : V(_V), data(new bool[V]()) {}
        ~cut_t() { delete[] data; }

        size_t size() const { return V; }
        bool &operator[](size_t v) const { return data[v]; }
        bool *begin() const { return data; }
        bool *end() const { return data + V; }
        friend std::ostream &operator<<(std::ostream &s, const cut_t &cut)
        {
            bool is_front = true;
            for(bool b : cut)
            {
                if(is_front) is_front = false;
                else s << ' ';
                s << b;
            }
            return s;
        }
    }; // class cut_t

    cut_t min_cut(size_t s, size_t t)
    {
        while(bfs(s, t))
        {
            fill(itr.begin(), itr.end(), 0);
            while(dfs(s, t, std::numeric_limits<cap_t>::max()) > cap_t(0));
        }
        cut_t res(V);
        for(size_t v = 0; v != V; ++v) if(~level[v]) res.data[v] = 1;
        return res;
    }
}; // class Dinic

using namespace std;

class solver
{
  public:
    solver()
    {
        int h,w; cin>>h>>w;
        Dinic<i64> flow(h+w+2);
        const int s=h+w,t=s+1;
        for(int i=0; i<h; ++i)
        {
            for(int j=0; j<w; ++j)
            {
                int g; cin>>g;
                flow.add_edge(i,t,g);
                flow.add_edge(j+h,i,g);
            }
        }
        i64 ans=0;
        for(int i=0; i<h; ++i)
        {
            int r; cin>>r;
            ans+=r;
            flow.add_edge(s,i,r);
        }
        for(int j=0; j<w; ++j)
        {
            int c; cin>>c;
            ans+=c;
            flow.add_edge(s,j+h,c);
        }
        ans-=flow.max_flow(s,t);
        cout << ans << "\n";
    }
};

main()
{
    u32 t = 1;
#ifdef LOCAL
    t=3;
#endif
    // t = -1; // infinite loop
    // cin >> t; // case number given
    while(t--) solver();
}
0