結果

問題 No.177 制作進行の宮森あおいです!
ユーザー RiRinbaruRiRinbaru
提出日時 2024-07-05 10:52:44
言語 C++14
(gcc 12.3.0 + boost 1.83.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
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 1 ms
6,940 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 2 ms
6,940 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 AC 4 ms
6,940 KB
testcase_07 AC 2 ms
6,944 KB
testcase_08 AC 2 ms
6,940 KB
testcase_09 AC 8 ms
6,940 KB
testcase_10 AC 11 ms
6,944 KB
testcase_11 AC 10 ms
6,940 KB
testcase_12 AC 7 ms
6,944 KB
testcase_13 AC 2 ms
6,944 KB
testcase_14 AC 1 ms
6,940 KB
testcase_15 AC 2 ms
6,940 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
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++)
      | 

ソースコード

diff #

#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() == rev
    int 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のindex
        int v_uID = G[v].size();                 // 現時点での G[v] の要素数 = vからみたuのindex
        G[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のインスタンスff
FordFulkerson 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";
    }
}
0