結果
問題 | No.177 制作進行の宮森あおいです! |
ユーザー |
|
提出日時 | 2024-07-05 10:52:44 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 11 ms / 2,000 ms |
コード長 | 5,101 bytes |
コンパイル時間 | 1,920 ms |
コンパイル使用メモリ | 186,812 KB |
実行使用メモリ | 6,944 KB |
最終ジャッジ日時 | 2024-07-05 10:52:47 |
合計ジャッジ時間 | 2,552 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 13 |
コンパイルメッセージ
main.cpp:34: warning: "rep" redefined 34 | #define rep(i, a, b) for (int i = a; i < b; i++) | main.cpp:2: note: this is the location of the previous definition 2 | #define rep(i, p, n) for (ll i = p; i < (ll)(n); i++) |
ソースコード
#include <bits/stdc++.h>#define rep(i, p, n) for (ll i = p; i < (ll)(n); i++)#define rep2(i, p, n) for(ll i = p; i >= (ll)(n); i-- )using namespace std;using ll = long long;using ld = long double;double pi=3.141592653589793;const long long inf=2*1e9;const long long linf=4*1e18;template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; }template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return 1; } return 0; }//atcoder#include <atcoder/modint>#include <atcoder/dsu>#include <atcoder/segtree>#include <atcoder/lazysegtree>using namespace atcoder;using mint1 = modint1000000007;using mint2 = modint998244353;// ford_fulkersonのアルゴリズム// 事前準備: ffを定義// 入力: ff.add_edge(a, b, c); // from, to, costで辺を生成// 出力: ff.maxFlow(A, B) // 頂点A -> Bへの最大流// 諸々のdefineがいつものと異なる部分があるので注意(rep等)/*FORDFULKERSON*/#include <iostream>#include <vector>#include <algorithm>using namespace std;#define rep(i, a, b) for (int i = a; i < b; i++)#define INF (1e9 + 1)// 辺の構造体struct Edge{// rev: toから行ける頂点のうち、toから見てfromが何番目に位置するか// G[from].size() == revint rev, from, to, cap;};// フォードファルカーソン法class FordFulkerson{public:vector<vector<Edge>> G;vector<bool> visited;// 頂点数 n の残余グラフを用意int size = 0;void init(int n){G.resize(n);visited.resize(n);size = n;}/*頂点 u -> v について 上限 cost の辺を追加コスト0の逆辺も張る*/void add_edge(int u, int v, int cost){int u_vID = G[u].size(); // 現時点での G[u] の要素数 = uからみたvのindexint v_uID = G[v].size(); // 現時点での G[v] の要素数 = vからみたuのindexG[u].push_back(Edge{v_uID, u, v, cost}); //<u,v>の逆辺<v,u>はG[u][v_uID]G[v].push_back(Edge{u_vID, v, u, 0}); // 逆辺は追加時はコスト0!!}/*深さ優先探索 F はスタートした頂点からposに到達する過程での"残余グラフの辺の容量" の最小値)goalまでの往路は頂点を記録しながらs->tまでに共通して流せる容量= s->tまでの容量の最小値を取得復路はs->tまでの容量の最小値を使って残余ネットワークのコストを更新返り値: 流したフローの量*/int dfs(int pos, int goal, int F){if (pos == goal)return F; // ゴールに到着したら流すvisited[pos] = true; // 訪れた頂点を記録// G[pos]に隣接する頂点を探索for (auto &e : G[pos]){// 容量0の辺や訪問済みの頂点は無視if (e.cap == 0 or visited[e.to])continue;// 再帰で目的地までのパスを探すint flow = dfs(e.to, goal, min(F, e.cap));// 残余ネットワークの更新// フローを流せる場合、残余グラフの容量をflowだけ増減させるif (flow > 0){e.cap -= flow; // u->vの辺を減少G[e.to][e.rev].cap += flow; // v->uの辺を増加return flow;}}return 0;}// 頂点sから頂点tまでの最大フローの総流量を返すint maxFlow(int s, int t){int totalFlow = 0;while (true){// s->tに探索する前に記録した頂点をリセットvisited.assign(size, false);int F = dfs(s, t, INF); // s->tへの流量を取得// フローを流せなくなったら終了if (F == 0)break;totalFlow += F;}return totalFlow;}};// FordFulkersonのインスタンスffFordFulkerson ff;int main() {ll W;cin >> W;ll N;cin >> N;vector<ll> J(N);rep(i, 0, N) {cin >> J.at(i);}ll M;cin >> M;vector<ll> C(M);rep(i, 0, M) {cin >> C.at(i);}ff.init(N+M+2);rep(i, 0, N) {ff.add_edge(0, i+1, J.at(i));//cout << J.at(i) << endl;}rep(i, 0, M) {ff.add_edge(N+1+i, N+M+1, C.at(i));}rep(i, 0, M) {ll Q;cin >> Q;vector<bool> li(N, true);rep(j, 0, Q) {ll X;cin >> X;li.at(X-1)=false;}rep(j, 0, N) {if (li.at(j)) {//cout << j+1 << N+i+1 << endl;ff.add_edge(j+1, N+i+1, inf);}}}ll P=ff.maxFlow(0, N+M+1);//cout << P << endl;if (W<=P) {cout << "SHIROBAKO";}else {cout << "BANSAKUTSUKITA";}}