結果

問題 No.1479 Matrix Eraser
ユーザー 👑 emthrmemthrm
提出日時 2021-04-16 20:43:35
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 432 ms / 3,000 ms
コード長 4,755 bytes
コンパイル時間 2,938 ms
コンパイル使用メモリ 230,616 KB
実行使用メモリ 30,848 KB
最終ジャッジ日時 2024-07-02 23:15:07
合計ジャッジ時間 10,682 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 1 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 38 ms
6,528 KB
testcase_08 AC 70 ms
8,576 KB
testcase_09 AC 163 ms
14,208 KB
testcase_10 AC 327 ms
22,016 KB
testcase_11 AC 189 ms
15,872 KB
testcase_12 AC 51 ms
7,296 KB
testcase_13 AC 68 ms
8,704 KB
testcase_14 AC 52 ms
7,552 KB
testcase_15 AC 12 ms
5,376 KB
testcase_16 AC 59 ms
7,936 KB
testcase_17 AC 432 ms
25,088 KB
testcase_18 AC 408 ms
25,088 KB
testcase_19 AC 414 ms
25,088 KB
testcase_20 AC 397 ms
25,088 KB
testcase_21 AC 411 ms
25,088 KB
testcase_22 AC 398 ms
24,960 KB
testcase_23 AC 408 ms
24,960 KB
testcase_24 AC 414 ms
25,088 KB
testcase_25 AC 420 ms
25,088 KB
testcase_26 AC 430 ms
25,216 KB
testcase_27 AC 149 ms
6,400 KB
testcase_28 AC 145 ms
6,272 KB
testcase_29 AC 150 ms
6,400 KB
testcase_30 AC 150 ms
6,400 KB
testcase_31 AC 156 ms
6,400 KB
testcase_32 AC 64 ms
13,108 KB
testcase_33 AC 64 ms
13,076 KB
testcase_34 AC 64 ms
12,868 KB
testcase_35 AC 65 ms
13,164 KB
testcase_36 AC 62 ms
13,008 KB
testcase_37 AC 13 ms
5,376 KB
testcase_38 AC 127 ms
5,376 KB
testcase_39 AC 354 ms
30,848 KB
testcase_40 AC 2 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define _USE_MATH_DEFINES
#include <bits/stdc++.h>
using namespace std;
#define FOR(i,m,n) for(int i=(m);i<(n);++i)
#define REP(i,n) FOR(i,0,n)
#define ALL(v) (v).begin(),(v).end()
using ll = long long;
constexpr int INF = 0x3f3f3f3f;
constexpr long long LINF = 0x3f3f3f3f3f3f3f3fLL;
constexpr double EPS = 1e-8;
constexpr int MOD = 1000000007;
// constexpr int MOD = 998244353;
constexpr int dy[] = {1, 0, -1, 0}, dx[] = {0, -1, 0, 1};
constexpr int dy8[] = {1, 1, 0, -1, -1, -1, 0, 1}, dx8[] = {0, -1, -1, -1, 0, 1, 1, 1};
template <typename T, typename U> inline bool chmax(T &a, U b) { return a < b ? (a = b, true) : false; }
template <typename T, typename U> inline bool chmin(T &a, U b) { return a > b ? (a = b, true) : false; }
struct IOSetup {
  IOSetup() {
    std::cin.tie(nullptr);
    std::ios_base::sync_with_stdio(false);
    std::cout << fixed << setprecision(20);
  }
} iosetup;

template <typename T>
struct Dinic {
  struct Edge {
    int dst, rev;
    T cap;
    Edge(int dst, T cap, int rev) : dst(dst), cap(cap), rev(rev) {}
  };

  std::vector<std::vector<Edge>> graph;

  Dinic(int n) : graph(n), level(n), itr(n) {}

  void add_edge(int src, int dst, T cap) {
    graph[src].emplace_back(dst, cap, graph[dst].size());
    graph[dst].emplace_back(src, 0, graph[src].size() - 1);
  }

  T maximum_flow(int s, int t, T limit) {
    T res = 0;
    while (true) {
      std::fill(level.begin(), level.end(), -1);
      std::queue<int> que;
      level[s] = 0;
      que.emplace(s);
      while (!que.empty()) {
        int ver = que.front(); que.pop();
        for (const Edge &e : graph[ver]) {
          if (level[e.dst] == -1 && e.cap > 0) {
            level[e.dst] = level[ver] + 1;
            que.emplace(e.dst);
          }
        }
      }
      if (level[t] == -1) return res;
      std::fill(itr.begin(), itr.end(), 0);
      T f;
      while ((f = dfs(s, t, limit)) > 0) res += f;
    }
  }

private:
  std::vector<int> level, itr;

  T dfs(int ver, int t, T flow) {
    if (ver == t) return flow;
    for (; itr[ver] < graph[ver].size(); ++itr[ver]) {
      Edge &e = graph[ver][itr[ver]];
      if (level[ver] < level[e.dst] && e.cap > 0) {
        T tmp = dfs(e.dst, t, std::min(flow, e.cap));
        if (tmp > 0) {
          e.cap -= tmp;
          graph[e.dst][e.rev].cap += tmp;
          return tmp;
        }
      }
    }
    return 0;
  }
};

template <template <typename> class C, typename T>
struct ProjectSelectionProblem {
  ProjectSelectionProblem(int n) : n(n) {}

  void add_neq(int u, int v, T cost) {
    assert(cost >= 0);
    us.emplace_back(u);
    vs.emplace_back(v);
    costs.emplace_back(cost);
  }

  void add(int v, bool group, T cost) {
    if (cost < 0) {
      cost = -cost;
      res += cost;
      group = !group;
    }
    if (group) {
      add_neq(-2, v, cost);  // -2 represents S.
    } else {
      add_neq(v, -1, cost);  // -1 represents T.
    }
  }

  void add_or(const std::vector<int> &v, bool group, T cost) {
    assert(cost >= 0);
    add(n, group, cost);
    if (group) {
      for (int vi : v) add_neq(n, vi, inf);
    } else {
      for (int vi : v) add_neq(vi, n, inf);
    }
    ++n;
  }

  void add_or(int u, int v, bool group, T cost) { add_or({u, v}, group, cost); }

  void add_eq(const std::vector<int> &v, bool group, T cost) {
    assert(cost <= 0);
    cost = -cost;
    res += cost;
    add_or(v, !group, cost);
  }

  void add_eq(int u, int v, bool group, T cost) { add_eq({u, v}, group, cost); }

  T solve() {
    C<T> flow(n + 2);
    int neq_sz = costs.size();
    for (int i = 0; i < neq_sz; ++i) {
      flow.add_edge(us[i] < 0 ? us[i] + n + 2 : us[i], vs[i] < 0 ? vs[i] + n + 2 : vs[i], costs[i]);
    }
    return flow.maximum_flow(n, n + 1, inf) - res;
  }

private:
  const T inf = std::numeric_limits<T>::max();
  int n;
  T res = 0;
  std::vector<int> us, vs;
  std::vector<T> costs;
};

int main() {
  int h, w; cin >> h >> w;
  map<int, vector<pair<int, int>>> mp;
  REP(i, h) REP(j, w) {
    int a; cin >> a;
    if (a > 0) mp[a].emplace_back(i, j);
  }
  int ans = 0;
  for (auto [_, pos] : mp) {
    vector<int> row, col;
    for (auto [i, j] : pos) {
      row.emplace_back(i);
      col.emplace_back(j);
    }
    sort(ALL(row));
    row.erase(unique(ALL(row)), row.end());
    int r = row.size();
    sort(ALL(col));
    col.erase(unique(ALL(col)), col.end());
    int c = col.size();
    ProjectSelectionProblem<Dinic, int> psp(r + c);
    REP(i, r) psp.add(i, 1, 1);
    REP(j, c) psp.add(r + j, 0, 1);
    for (auto [i, j] : pos) {
      i = lower_bound(ALL(row), i) - row.begin();
      j = lower_bound(ALL(col), j) - col.begin();
      psp.add_neq(i, r + j, INF);
    }
    ans += psp.solve();
  }
  cout << ans << '\n';
  return 0;
}
0