結果
問題 | No.2238 Rock and Hole |
ユーザー |
![]() |
提出日時 | 2023-02-11 13:17:59 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 41 ms / 3,000 ms |
コード長 | 20,483 bytes |
コンパイル時間 | 2,562 ms |
コンパイル使用メモリ | 217,280 KB |
最終ジャッジ日時 | 2025-02-10 14:19:40 |
ジャッジサーバーID (参考情報) |
judge1 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 24 |
コンパイルメッセージ
/home/nok0/documents/programming/library/template/input.hpp: In function ‘void scanner::scan(char*)’: /home/nok0/documents/programming/library/template/input.hpp:29:33: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
ソースコード
#line 1 "/home/nok0/documents/programming/library/atcoder/maxflow.hpp"#include <algorithm>#include <cassert>#include <limits>#include <queue>#include <vector>#line 1 "/home/nok0/documents/programming/library/atcoder/internal_queue.hpp"#line 5 "/home/nok0/documents/programming/library/atcoder/internal_queue.hpp"namespace atcoder {namespace internal {template <class T> struct simple_queue {std::vector<T> payload;int pos = 0;void reserve(int n) { payload.reserve(n); }int size() const { return int(payload.size()) - pos; }bool empty() const { return pos == int(payload.size()); }void push(const T& t) { payload.push_back(t); }T& front() { return payload[pos]; }void clear() {payload.clear();pos = 0;}void pop() { pos++; }};} // namespace internal} // namespace atcoder#line 11 "/home/nok0/documents/programming/library/atcoder/maxflow.hpp"namespace atcoder {template <class Cap> struct mf_graph {public:mf_graph() : _n(0) {}explicit mf_graph(int n) : _n(n), g(n) {}int add_edge(int from, int to, Cap cap) {assert(0 <= from && from < _n);assert(0 <= to && to < _n);assert(0 <= cap);int m = int(pos.size());pos.push_back({from, int(g[from].size())});int from_id = int(g[from].size());int to_id = int(g[to].size());if (from == to) to_id++;g[from].push_back(_edge{to, to_id, cap});g[to].push_back(_edge{from, from_id, 0});return m;}struct edge {int from, to;Cap cap, flow;};edge get_edge(int i) {int m = int(pos.size());assert(0 <= i && i < m);auto _e = g[pos[i].first][pos[i].second];auto _re = g[_e.to][_e.rev];return edge{pos[i].first, _e.to, _e.cap + _re.cap, _re.cap};}std::vector<edge> edges() {int m = int(pos.size());std::vector<edge> result;for (int i = 0; i < m; i++) {result.push_back(get_edge(i));}return result;}void change_edge(int i, Cap new_cap, Cap new_flow) {int m = int(pos.size());assert(0 <= i && i < m);assert(0 <= new_flow && new_flow <= new_cap);auto& _e = g[pos[i].first][pos[i].second];auto& _re = g[_e.to][_e.rev];_e.cap = new_cap - new_flow;_re.cap = new_flow;}Cap flow(int s, int t) {return flow(s, t, std::numeric_limits<Cap>::max());}Cap flow(int s, int t, Cap flow_limit) {assert(0 <= s && s < _n);assert(0 <= t && t < _n);assert(s != t);std::vector<int> level(_n), iter(_n);internal::simple_queue<int> que;auto bfs = [&]() {std::fill(level.begin(), level.end(), -1);level[s] = 0;que.clear();que.push(s);while (!que.empty()) {int v = que.front();que.pop();for (auto e : g[v]) {if (e.cap == 0 || level[e.to] >= 0) continue;level[e.to] = level[v] + 1;if (e.to == t) return;que.push(e.to);}}};auto dfs = [&](auto self, int v, Cap up) {if (v == s) return up;Cap res = 0;int level_v = level[v];for (int& i = iter[v]; i < int(g[v].size()); i++) {_edge& e = g[v][i];if (level_v <= level[e.to] || g[e.to][e.rev].cap == 0) continue;Cap d =self(self, e.to, std::min(up - res, g[e.to][e.rev].cap));if (d <= 0) continue;g[v][i].cap += d;g[e.to][e.rev].cap -= d;res += d;if (res == up) return res;}level[v] = _n;return res;};Cap flow = 0;while (flow < flow_limit) {bfs();if (level[t] == -1) break;std::fill(iter.begin(), iter.end(), 0);Cap f = dfs(dfs, t, flow_limit - flow);if (!f) break;flow += f;}return flow;}std::vector<bool> min_cut(int s) {std::vector<bool> visited(_n);internal::simple_queue<int> que;que.push(s);while (!que.empty()) {int p = que.front();que.pop();visited[p] = true;for (auto e : g[p]) {if (e.cap && !visited[e.to]) {visited[e.to] = true;que.push(e.to);}}}return visited;}private:int _n;struct _edge {int to, rev;Cap cap;};std::vector<std::pair<int, int>> pos;std::vector<std::vector<_edge>> g;};} // namespace atcoder#line 2 "/home/nok0/documents/programming/library/template/header.hpp"#include <bits/stdc++.h>#line 3 "/home/nok0/documents/programming/library/template/def_const.hpp"const int inf = 1000000000;const long long INF = 1000000000000000000ll;#line 4 "/home/nok0/documents/programming/library/template/debug.hpp"namespace viewer {void view(const long long &e) {if(e == INF)std::cerr << "INF";else if(e == -INF)std::cerr << "-INF";elsestd::cerr << e;}void view(const int &e) {if(e == inf)std::cerr << "inf";else if(e == -inf)std::cerr << "-inf";elsestd::cerr << e;}template <typename T>void view(const T &e) {std::cerr << e;}template <typename T, typename U>void view(const std::pair<T, U> &p) {std::cerr << "(";view(p.first);std::cerr << ", ";view(p.second);std::cerr << ")";}template <class T0, class T1, class T2>void view(const std::tuple<T0, T1, T2> &p) {std::cerr << "(";view(std::get<0>(p));std::cerr << ", ";view(std::get<1>(p));std::cerr << ", ";view(std::get<2>(p));std::cerr << ")";}template <class T0, class T1, class T2, class T3>void view(const std::tuple<T0, T1, T2, T3> &p) {std::cerr << "(";view(std::get<0>(p));std::cerr << ", ";view(std::get<1>(p));std::cerr << ", ";view(std::get<2>(p));std::cerr << ", ";view(std::get<3>(p));std::cerr << ")";}template <typename T>void view(const std::set<T> &s) {if(s.empty()) {std::cerr << "{ }";return;}std::cerr << "{ ";for(auto &t : s) {view(t);std::cerr << ", ";}std::cerr << "\b\b }";}template <typename T>void view(const std::unordered_set<T> &s) {if(s.empty()) {std::cerr << "{ }";return;}std::cerr << "{ ";for(auto &t : s) {view(t);std::cerr << ", ";}std::cerr << "\b\b }";}template <typename T>void view(const std::multiset<T> &s) {if(s.empty()) {std::cerr << "{ }";return;}std::cerr << "{ ";for(auto &t : s) {view(t);std::cerr << ", ";}std::cerr << "\b\b }";}template <typename T>void view(const std::unordered_multiset<T> &s) {if(s.empty()) {std::cerr << "{ }";return;}std::cerr << "{ ";for(auto &t : s) {view(t);std::cerr << ", ";}std::cerr << "\b\b }";}template <typename T>void view(const std::vector<T> &v) {if(v.empty()) {std::cerr << "{ }";return;}std::cerr << "{ ";for(const auto &e : v) {view(e);std::cerr << ", ";}std::cerr << "\b\b }";}template <typename T, std::size_t ary_size>void view(const std::array<T, ary_size> &v) {if(v.empty()) {std::cerr << "{ }";return;}std::cerr << "{ ";for(const auto &e : v) {view(e);std::cerr << ", ";}std::cerr << "\b\b }";}template <typename T>void view(const std::vector<std::vector<T>> &vv) {std::cerr << "{\n";for(const auto &v : vv) {std::cerr << "\t";view(v);std::cerr << '\n';}std::cerr << "}";}template <typename T, typename U>void view(const std::vector<std::pair<T, U>> &v) {std::cerr << "{\n";for(const auto &c : v) {std::cerr << "\t(";view(c.first);std::cerr << ", ";view(c.second);std::cerr << ")\n";}std::cerr << "}";}template <class T0, class T1, class T2>void view(const std::vector<std::tuple<T0, T1, T2>> &v) {if(v.empty()) {std::cerr << "{ }";return;}std::cerr << '{';for(const auto &t : v) {std::cerr << "\n\t";view(t);std::cerr << ",";}std::cerr << "\n}";}template <class T0, class T1, class T2, class T3>void view(const std::vector<std::tuple<T0, T1, T2, T3>> &v) {if(v.empty()) {std::cerr << "{ }";return;}std::cerr << '{';for(const auto &t : v) {std::cerr << "\n\t";view(t);std::cerr << ",";}std::cerr << "\n}";}template <typename T, typename U>void view(const std::map<T, U> &m) {std::cerr << "{\n";for(const auto &t : m) {std::cerr << "\t[";view(t.first);std::cerr << "] : ";view(t.second);std::cerr << '\n';}std::cerr << "}";}template <typename T, typename U>void view(const std::unordered_map<T, U> &m) {std::cerr << "{\n";for(const auto &t : m) {std::cerr << "\t[";view(t.first);std::cerr << "] : ";view(t.second);std::cerr << '\n';}std::cerr << "}";}} // namespace viewer// when compiling : g++ foo.cpp -DLOCAL#ifdef LOCALvoid debug_out() {}template <typename Head, typename... Tail>void debug_out(Head H, Tail... T) {viewer::view(H);std::cerr << ", ";debug_out(T...);}#define debug(...) \do { \std::cerr << __LINE__ << " [" << #__VA_ARGS__ << "] : ["; \debug_out(__VA_ARGS__); \std::cerr << "\b\b]\n"; \} while(0)#define dump(x) \do { \std::cerr << __LINE__ << " " << #x << " : "; \viewer::view(x); \std::cerr << '\n'; \} while(0)#else#define debug(...) (void(0))#define dump(x) (void(0))#endif#line 3 "/home/nok0/documents/programming/library/template/def_name.hpp"#define pb push_back#define eb emplace_back#define SZ(x) ((int)(x).size())#define all(x) (x).begin(), (x).end()#define rall(x) (x).rbegin(), (x).rend()#define popcnt(x) __builtin_popcountll(x)template<class T = int>using V = std::vector<T>;template<class T = int>using VV = std::vector<std::vector<T>>;template<class T>using pqup = std::priority_queue<T, std::vector<T>, std::greater<T>>;using ll = long long;using ld = long double;using int128 = __int128_t;using pii = std::pair<int, int>;using pll = std::pair<long long, long long>;#line 3 "/home/nok0/documents/programming/library/template/fast_io.hpp"struct fast_io {fast_io() {std::ios::sync_with_stdio(false);std::cin.tie(nullptr);std::cout << std::fixed << std::setprecision(15);}} fast_io_;#line 3 "/home/nok0/documents/programming/library/template/input.hpp"template<class T, class U>std::istream &operator>>(std::istream &is, std::pair<T, U> &p) {is >> p.first >> p.second;return is;}template<class T>std::istream &operator>>(std::istream &is, std::vector<T> &v) {for (T &i : v) is >> i;return is;}std::istream &operator>>(std::istream &is, __int128_t &a) {std::string s;is >> s;__int128_t ret = 0;for (int i = 0; i < (int)s.length(); i++)if ('0' <= s[i] and s[i] <= '9')ret = 10 * ret + s[i] - '0';a = ret * (s[0] == '-' ? -1 : 1);return is;}namespace scanner {void scan(int &a) { std::cin >> a; }void scan(long long &a) { std::cin >> a; }void scan(std::string &a) { std::cin >> a; }void scan(char &a) { std::cin >> a; }void scan(char a[]) { std::scanf("%s", a); }void scan(double &a) { std::cin >> a; }void scan(long double &a) { std::cin >> a; }template<class T, class U>void scan(std::pair<T, U> &p) { std::cin >> p; }template<class T>void scan(std::vector<T> &a) { std::cin >> a; }void INPUT() {}template<class Head, class... Tail>void INPUT(Head &head, Tail &...tail) {scan(head);INPUT(tail...);}} // namespace scanner#define VEC(type, name, size) \std::vector<type> name(size); \scanner::INPUT(name)#define VVEC(type, name, h, w) \std::vector<std::vector<type>> name(h, std::vector<type>(w)); \scanner::INPUT(name)#define INT(...) \int __VA_ARGS__; \scanner::INPUT(__VA_ARGS__)#define LL(...) \long long __VA_ARGS__; \scanner::INPUT(__VA_ARGS__)#define STR(...) \std::string __VA_ARGS__; \scanner::INPUT(__VA_ARGS__)#define CHAR(...) \char __VA_ARGS__; \scanner::INPUT(__VA_ARGS__)#define DOUBLE(...) \double __VA_ARGS__; \scanner::INPUT(__VA_ARGS__)#define LD(...) \long double __VA_ARGS__; \scanner::INPUT(__VA_ARGS__)#line 3 "/home/nok0/documents/programming/library/template/math.hpp"template <class T, class U>inline bool chmin(T &a, const U &b) { return a > b ? a = b, true : false; }template <class T, class U>inline bool chmax(T &a, const U &b) { return a < b ? a = b, true : false; }template <class T>T divup(T x, T y) { return (x + y - 1) / y; }template <class T>T POW(T a, long long n) {T ret = 1;while(n) {if(n & 1) ret *= a;a *= a;n >>= 1;}return ret;}long long POW(long long a, long long n, const int mod) {long long ret = 1;a = (a % mod + mod) % mod;while(n) {if(n & 1) (ret *= a) %= mod;(a *= a) %= mod;n >>= 1;}return ret;}template <class T, class F>T bin_search(T ok, T ng, const F &f) {while(abs(ok - ng) > 1) {T mid = (ok + ng) >> 1;(f(mid) ? ok : ng) = mid;}return ok;}template <class T, class F>T bin_search(T ok, T ng, const F &f, int loop) {for(int i = 0; i < loop; i++) {T mid = (ok + ng) / 2;(f(mid) ? ok : ng) = mid;}return ok;}#line 3 "/home/nok0/documents/programming/library/template/output.hpp"template<class T, class U>std::ostream &operator<<(std::ostream &os, const std::pair<T, U> &p) {os << p.first << " " << p.second;return os;}template<class T>std::ostream &operator<<(std::ostream &os, const std::vector<T> &a) {for (int i = 0; i < int(a.size()); ++i) {if (i) os << " ";os << a[i];}return os;}std::ostream &operator<<(std::ostream &dest, __int128_t &value) {std::ostream::sentry s(dest);if (s) {__uint128_t tmp = value < 0 ? -value : value;char buffer[128];char *d = std::end(buffer);do {--d;*d = "0123456789"[tmp % 10];tmp /= 10;} while (tmp != 0);if (value < 0) {--d;*d = '-';}int len = std::end(buffer) - d;if (dest.rdbuf()->sputn(d, len) != len) {dest.setstate(std::ios_base::badbit);}}return dest;}template<class T>void print(const T a) { std::cout << a << '\n'; }template<class Head, class... Tail>void print(Head H, Tail... T) {std::cout << H << ' ';print(T...);}template<class T>void printel(const T a) { std::cout << a << '\n'; }template<class T>void printel(const std::vector<T> &a) {for (const auto &v : a)std::cout << v << '\n';}template<class Head, class... Tail>void printel(Head H, Tail... T) {std::cout << H << '\n';printel(T...);}void Yes(const bool b = true) { std::cout << (b ? "Yes\n" : "No\n"); }void No() { std::cout << "No\n"; }void YES(const bool b = true) { std::cout << (b ? "YES\n" : "NO\n"); }void NO() { std::cout << "NO\n"; }#line 2 "/home/nok0/documents/programming/library/template/rep.hpp"#define foa(v, a) for (auto &v : a)#define repname(a, b, c, d, e, ...) e#define rep(...) repname(__VA_ARGS__, rep3, rep2, rep1, rep0)(__VA_ARGS__)#define rep0(x) for (int rep_counter = 0; rep_counter < (x); ++rep_counter)#define rep1(i, x) for (int i = 0; i < (x); ++i)#define rep2(i, l, r) for (int i = (l); i < (r); ++i)#define rep3(i, l, r, c) for (int i = (l); i < (r); i += (c))#define repsname(a, b, c, ...) c#define reps(...) repsname(__VA_ARGS__, reps1, reps0)(__VA_ARGS__)#define reps0(x) for (int reps_counter = 1; reps_counter <= (x); ++reps_counter)#define reps1(i, x) for (int i = 1; i <= (x); ++i)#define rrepname(a, b, c, ...) c#define rrep(...) rrepname(__VA_ARGS__, rrep1, rrep0)(__VA_ARGS__)#define rrep0(x) for (int rrep_counter = (x)-1; rrep_counter >= 0; --rrep_counter)#define rrep1(i, x) for (int i = (x)-1; i >= 0; --i)#line 3 "/home/nok0/documents/programming/library/template/vector.hpp"template <class T>int lb(const std::vector<T> &a, const T x) { return std::distance((a).begin(), std::lower_bound((a).begin(), (a).end(), (x))); }template <class T>int ub(const std::vector<T> &a, const T x) { return std::distance((a).begin(), std::upper_bound((a).begin(), (a).end(), (x))); }template <class T>void UNIQUE(std::vector<T> &a) {std::sort(a.begin(), a.end());a.erase(std::unique(a.begin(), a.end()), a.end());}template <class T>std::vector<T> press(std::vector<T> &a) {auto res = a;UNIQUE(res);for(auto &v : a)v = lb(res, v);return res;}#define SORTname(a, b, c, ...) c#define SORT(...) SORTname(__VA_ARGS__, SORT1, SORT0, ...)(__VA_ARGS__)#define SORT0(a) std::sort((a).begin(), (a).end())#define SORT1(a, c) std::sort((a).begin(), (a).end(), [](const auto x, const auto y) { return x c y; })template <class T>void ADD(std::vector<T> &a, const T x = 1) {for(auto &v : a) v += x;}template <class T>void SUB(std::vector<T> &a, const T x = 1) {for(auto &v : a) v -= x;}template <class T>struct cum_vector {public:cum_vector() = default;template <class U>cum_vector(const std::vector<U> &vec) : cum((int)vec.size() + 1) {for(int i = 0; i < (int)vec.size(); i++)cum[i + 1] = cum[i] + vec[i];}T prod(int l, int r) {return cum[r] - cum[l];}private:std::vector<T> cum;};std::vector<std::pair<char, int>> rle(const std::string &s) {const int n = s.size();std::vector<std::pair<char, int>> ret;for(int l = 0; l < n;) {int r = l + 1;for(; r < n and s[l] == s[r]; r++) {}ret.emplace_back(s[l], r - l);l = r;}return ret;}template <class T>std::vector<std::pair<T, int>> rle(const std::vector<T> &v) {int n = v.size();std::vector<std::pair<T, int>> ret;for(int l = 0; l < n;) {int r = l + 1;for(; r < n and v[l] == v[r]; r++) {}ret.emplace_back(v[l], r - l);l = r;}return ret;}std::vector<int> iota(int n) {std::vector<int> p(n);std::iota(p.begin(), p.end(), 0);return p;}#line 11 "/home/nok0/documents/programming/library/template/all"using namespace std;#line 3 "a.cpp"void main_();int main() {int t = 1;while(t--) main_();}void main_() {INT(h, w);VEC(string, s, h);auto id = [&](int i, int j) {return i * w + j;};V<> hole, stone;VV<> x(h), y(w);rep(i, h) {rep(j, w) {if(s[i][j] == 'h') {hole.push_back(id(i, j));x[i].push_back(j);y[j].push_back(i);}if(s[i][j] == 'r') {stone.push_back(id(i, j));}}}debug(hole);int H = SZ(hole), S = SZ(stone);atcoder::mf_graph<int> mf(S + H + 2);int st = S + H, en = st + 1;rep(i, S) {mf.add_edge(st, i, 1);}rep(i, H) {mf.add_edge(S + i, en, 1);}rep(i, h) {rep(j, w) {if(s[i][j] != 'r') continue;int u = lb(stone, id(i, j));debug(u);auto itr = lb(x[i], j);if(SZ(x[i]) > itr) {debug(i, x[i][itr]);debug(id(i, x[i][itr]));debug(lb(hole, id(i, x[i][itr])));mf.add_edge(u, S + lb(hole, id(i, x[i][itr])), 1);}--itr;if(itr >= 0) {debug(id(i, x[i][itr]));debug(lb(hole, id(i, x[i][itr])));mf.add_edge(u, S + lb(hole, id(i, x[i][itr])), 1);}itr = lb(y[j], i);if(SZ(y[j]) > itr) {debug(lb(hole, id(y[j][itr], j)));mf.add_edge(u, S + lb(hole, id(y[j][itr], j)), 1);}--itr;if(itr >= 0) {debug(lb(hole, id(y[j][itr], j)));mf.add_edge(u, S + lb(hole, id(y[j][itr], j)), 1);}}}auto flow = mf.flow(st, en);debug(flow);Yes(flow == S);}