結果

問題 No.918 LISGRID
ユーザー risujirohrisujiroh
提出日時 2019-10-25 23:17:23
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 74 ms / 2,000 ms
コード長 2,005 bytes
コンパイル時間 2,111 ms
コンパイル使用メモリ 186,268 KB
実行使用メモリ 22,524 KB
最終ジャッジ日時 2024-09-13 08:41:58
合計ジャッジ時間 5,648 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using lint = long long;
template<class T = int> using V = vector<T>;
template<class T = int> using VV = V< V<T> >;

int scc(const VV<>& g, V<>& comp) {
  int n = g.size();
  VV<> rg(n);
  for (int v = 0; v < n; v++) for (int w : g[v]) {
    rg[w].push_back(v);
  }
  V<> vs;
  V<bool> vis(n);
  comp.resize(n);
  auto dfs = [&](const auto& dfs, int v) -> void {
    vis[v] = true;
    for (int w : g[v]) if (!vis[w]) {
      dfs(dfs, w);
    }
    vs.push_back(v);
  };
  auto rdfs = [&](const auto& rdfs, int v, int k) -> void {
    vis[v] = true;
    comp[v] = k;
    for (int w : rg[v]) if (!vis[w]) {
      rdfs(rdfs, w, k);
    }
  };
  for (int v = 0; v < n; v++) if (!vis[v]) {
    dfs(dfs, v);
  }
  vis.assign(n, false);
  int k = 0;
  reverse(begin(vs), end(vs));
  for (int v : vs) if (!vis[v]) {
    rdfs(rdfs, v, k++);
  }
  return k;
}

int main() {
  cin.tie(nullptr); ios::sync_with_stdio(false);
  int h, w; cin >> h >> w;
  V<> a(h); for (auto&& e : a) cin >> e;
  V<> b(w); for (auto&& e : b) cin >> e;
  sort(rbegin(a), rend(a));
  sort(rbegin(b), rend(b));
  auto _ = [&](int i, int j) -> int { return i * w + j; };
  VV<> g(h * w);
  for (int i = 0; i < h; ++i) {
    for (int j = w - a[i]; j < w - 1; ++j) {
      g[_(i, j)].push_back(_(i, j + 1));
    }
    if (w - a[i]) {
      g[_(i, w - 1)].push_back(_(i, w - a[i] - 1));
    }
    for (int j = w - a[i] - 1; j >= 1; --j) {
      g[_(i, j)].push_back(_(i, j - 1));
    }
  }
  for (int j = 0; j < w; ++j) {
    for (int i = h - b[j]; i < h - 1; ++i) {
      g[_(i, j)].push_back(_(i + 1, j));
    }
    if (h - b[j]) {
      g[_(h - 1, j)].push_back(_(h - b[j] - 1, j));
    }
    for (int i = h - b[j] - 1; i >= 1; --i) {
      g[_(i, j)].push_back(_(i - 1, j));
    }
  }
  V<> comp;
  int k = scc(g, comp);
  cerr << "k = " << k << " (l." << __LINE__ << ")\n";
  for (int i = 0; i < h; ++i) for (int j = 0; j < w; ++j) {
    cout << comp[_(i, j)] + 1 << " \n"[j == w - 1];
  }
}
0