結果
問題 | No.2713 Just Solitaire |
ユーザー | 👑 emthrm |
提出日時 | 2024-03-31 14:49:36 |
言語 | C++23 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 3 ms / 2,000 ms |
コード長 | 4,973 bytes |
コンパイル時間 | 2,881 ms |
コンパイル使用メモリ | 264,488 KB |
実行使用メモリ | 6,820 KB |
最終ジャッジ日時 | 2024-09-30 19:59:53 |
合計ジャッジ時間 | 3,794 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
6,816 KB |
testcase_01 | AC | 2 ms
6,816 KB |
testcase_02 | AC | 2 ms
6,816 KB |
testcase_03 | AC | 2 ms
6,816 KB |
testcase_04 | AC | 2 ms
6,816 KB |
testcase_05 | AC | 1 ms
6,816 KB |
testcase_06 | AC | 2 ms
6,820 KB |
testcase_07 | AC | 1 ms
6,820 KB |
testcase_08 | AC | 1 ms
6,820 KB |
testcase_09 | AC | 2 ms
6,820 KB |
testcase_10 | AC | 2 ms
6,820 KB |
testcase_11 | AC | 2 ms
6,816 KB |
testcase_12 | AC | 1 ms
6,816 KB |
testcase_13 | AC | 2 ms
6,816 KB |
testcase_14 | AC | 2 ms
6,820 KB |
testcase_15 | AC | 1 ms
6,816 KB |
testcase_16 | AC | 1 ms
6,816 KB |
testcase_17 | AC | 2 ms
6,820 KB |
testcase_18 | AC | 1 ms
6,816 KB |
testcase_19 | AC | 2 ms
6,816 KB |
testcase_20 | AC | 1 ms
6,816 KB |
testcase_21 | AC | 2 ms
6,816 KB |
testcase_22 | AC | 1 ms
6,820 KB |
testcase_23 | AC | 2 ms
6,816 KB |
testcase_24 | AC | 3 ms
6,816 KB |
testcase_25 | AC | 3 ms
6,816 KB |
testcase_26 | AC | 2 ms
6,816 KB |
testcase_27 | AC | 2 ms
6,820 KB |
testcase_28 | AC | 2 ms
6,820 KB |
testcase_29 | AC | 2 ms
6,816 KB |
testcase_30 | AC | 3 ms
6,816 KB |
testcase_31 | AC | 2 ms
6,816 KB |
testcase_32 | AC | 2 ms
6,816 KB |
testcase_33 | AC | 3 ms
6,820 KB |
ソースコード
#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) using ll = long long; constexpr int INF = 0x3f3f3f3f; constexpr long long LINF = 0x3f3f3f3f3f3f3f3fLL; constexpr double EPS = 1e-8; constexpr int MOD = 998244353; // constexpr int MOD = 1000000007; constexpr int DY4[]{1, 0, -1, 0}, DX4[]{0, -1, 0, 1}; constexpr int DY8[]{1, 1, 0, -1, -1, -1, 0, 1}; constexpr int 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 <template <typename> class C, typename T> concept MaximumFlow = requires (C<T> mf) { {mf.add_edge(std::declval<int>(), std::declval<int>(), std::declval<T>())} -> std::same_as<void>; {mf.maximum_flow(std::declval<int>(), std::declval<int>())} -> std::same_as<T>; }; template <typename T> struct Dinic { struct Edge { int dst, rev; T cap; explicit Edge(const int dst, const T cap, const int rev) : dst(dst), rev(rev), cap(cap) {} }; std::vector<std::vector<Edge>> graph; explicit Dinic(const int n) : graph(n), level(n), itr(n) {} void add_edge(const int src, const int dst, const T cap) { graph[src].emplace_back(dst, cap, graph[dst].size()); graph[dst].emplace_back(src, 0, graph[src].size() - 1); } T maximum_flow(const int s, const int t, T limit = std::numeric_limits<T>::max()) { T res = 0; while (limit > 0) { std::fill(level.begin(), level.end(), -1); level[s] = 0; std::queue<int> que; que.emplace(s); while (!que.empty()) { const 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) break; std::fill(itr.begin(), itr.end(), 0); while (limit > 0) { const T f = dfs(s, t, limit); if (f == 0) break; limit -= f; res += f; } } return res; } private: std::vector<int> level, itr; T dfs(const int ver, const int t, const T flow) { if (ver == t) return flow; for (; std::cmp_less(itr[ver], graph[ver].size()); ++itr[ver]) { Edge& e = graph[ver][itr[ver]]; if (level[ver] < level[e.dst] && e.cap > 0) { const 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> requires MaximumFlow<C, T> struct ProjectSelectionProblem { explicit ProjectSelectionProblem(const int n) : inf(std::numeric_limits<T>::max()), n(n), res(0) {} void add_neq(const int u, const int v, const T cost) { assert(cost >= 0); us.emplace_back(u); vs.emplace_back(v); costs.emplace_back(cost); } void add(const 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, const bool group, const T cost) { assert(cost >= 0); add(n, group, cost); if (group) { for (const int e : v) add_neq(n, e, inf); } else { for (const int e : v) add_neq(e, n, inf); } ++n; } void add_or(const int u, const int v, const bool group, const T cost) { add_or({u, v}, group, cost); } void add_eq(const std::vector<int>& v, const bool group, T cost) { assert(cost <= 0); cost = -cost; res += cost; add_or(v, !group, cost); } void add_eq(const int u, const int v, const bool group, const T cost) { add_eq({u, v}, group, cost); } T solve() { C<T> mf(n + 2); const int neq_size = costs.size(); for (int i = 0; i < neq_size; ++i) { mf.add_edge(us[i] < 0 ? us[i] + n + 2 : us[i], vs[i] < 0 ? vs[i] + n + 2 : vs[i], costs[i]); } return mf.maximum_flow(n, n + 1, inf) - res; } private: const T inf; int n; T res; std::vector<int> us, vs; std::vector<T> costs; }; int main() { int n, m; cin >> n >> m; ProjectSelectionProblem<Dinic, ll> psp(n); REP(i, n) { int a; cin >> a; psp.add(i, true, a); } vector<int> b(m); for (int& b_i : b) cin >> b_i; REP(i, m) { int k; cin >> k; vector<int> c(k); for (int& c_i : c) cin >> c_i, --c_i; psp.add_eq(c, true, -b[i]); } cout << -psp.solve() << '\n'; return 0; }