#718823 (C++17) No.1775 Love Triangle 2

提出ソース

結果

問題	No.1775 Love Triangle 2
ユーザー	👑 hitonanode
提出日時	2021-11-21 14:29:30
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	WA (最新) AC (最初)
実行時間	-
コード長	3,737 bytes
コンパイル時間	1,760 ms
コンパイル使用メモリ	111,768 KB
実行使用メモリ	4,648 KB
最終ジャッジ日時	2023-09-16 22:12:30
合計ジャッジ時間	5,108 ms
ジャッジサーバーID （参考情報）	judge12 / judge13

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	1 ms 4,380 KB
testcase_01	AC	1 ms 4,380 KB
testcase_02	AC	2 ms 4,380 KB
testcase_03	AC	2 ms 4,376 KB
testcase_04	AC	15 ms 4,376 KB
testcase_05	AC	16 ms 4,648 KB
testcase_06	AC	16 ms 4,380 KB
testcase_07	AC	7 ms 4,384 KB
testcase_08	AC	18 ms 4,380 KB
testcase_09	AC	3 ms 4,380 KB
testcase_10	AC	3 ms 4,376 KB
testcase_11	AC	3 ms 4,380 KB
testcase_12	AC	3 ms 4,376 KB
testcase_13	AC	3 ms 4,376 KB
testcase_14	AC	4 ms 4,380 KB
testcase_15	AC	7 ms 4,376 KB
testcase_16	AC	7 ms 4,380 KB
testcase_17	AC	8 ms 4,380 KB
testcase_18	AC	13 ms 4,376 KB
testcase_19	AC	3 ms 4,380 KB
testcase_20	AC	3 ms 4,380 KB
testcase_21	AC	3 ms 4,380 KB
testcase_22	AC	3 ms 4,376 KB
testcase_23	AC	3 ms 4,380 KB
testcase_24	AC	3 ms 4,380 KB
testcase_25	AC	3 ms 4,380 KB
testcase_26	AC	3 ms 4,380 KB
testcase_27	AC	3 ms 4,376 KB
testcase_28	AC	4 ms 4,380 KB
testcase_29	AC	16 ms 4,480 KB
testcase_30	AC	18 ms 4,380 KB
testcase_31	AC	11 ms 4,380 KB
testcase_32	AC	10 ms 4,380 KB
testcase_33	AC	7 ms 4,380 KB
testcase_34	AC	7 ms 4,376 KB
testcase_35	AC	6 ms 4,376 KB
testcase_36	AC	4 ms 4,380 KB
testcase_37	AC	3 ms 4,376 KB
testcase_38	AC	3 ms 4,384 KB
testcase_39	AC	18 ms 4,380 KB
testcase_40	AC	12 ms 4,380 KB
testcase_41	AC	10 ms 4,384 KB
testcase_42	AC	9 ms 4,384 KB
testcase_43	AC	8 ms 4,384 KB
testcase_44	AC	7 ms 4,380 KB
testcase_45	WA	-
testcase_46	AC	4 ms 4,376 KB
testcase_47	AC	4 ms 4,376 KB
testcase_48	AC	3 ms 4,380 KB
testcase_49	AC	15 ms 4,376 KB
testcase_50	AC	17 ms 4,384 KB
testcase_51	AC	10 ms 4,380 KB
testcase_52	AC	10 ms 4,380 KB
testcase_53	AC	7 ms 4,384 KB
testcase_54	WA	-
testcase_55	AC	6 ms 4,380 KB
testcase_56	AC	4 ms 4,376 KB
testcase_57	AC	3 ms 4,380 KB
testcase_58	AC	3 ms 4,380 KB
testcase_59	AC	18 ms 4,380 KB
testcase_60	AC	13 ms 4,376 KB
testcase_61	AC	10 ms 4,380 KB
testcase_62	AC	9 ms 4,376 KB
testcase_63	AC	7 ms 4,376 KB
testcase_64	AC	6 ms 4,376 KB
testcase_65	AC	4 ms 4,376 KB
testcase_66	AC	3 ms 4,504 KB
testcase_67	AC	3 ms 4,380 KB
testcase_68	AC	3 ms 4,380 KB
testcase_69	WA	-
testcase_70	WA	-
testcase_71	WA	-
testcase_72	WA	-
testcase_73	WA	-
testcase_74	WA	-
testcase_75	WA	-
testcase_76	WA	-
testcase_77	WA	-
testcase_78	WA	-
testcase_79	WA	-
testcase_80	WA	-
testcase_81	WA	-
testcase_82	WA	-
testcase_83	WA	-
testcase_84	WA	-
testcase_85	WA	-
testcase_86	WA	-
testcase_87	WA	-
testcase_88	WA	-
testcase_89	AC	3 ms 4,380 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

// 各辺にコスト1を割り当て，x->y, x->z に流量 2 の最小費用流を流し，x->y のパスを固定して
// 残りの頂点と辺で z->x, z->y に流量 2 の最小費用流を流す
// 正当性なし（反例構成可能）
#include <algorithm>
#include <iostream>
#include <utility>
#include <vector>
using namespace std;

#include <atcoder/mincostflow>

// used_vs に含まれる頂点は使わずに，from -> to1 と from->to2 の点素なパスを構成する．
// 両方のパスが構築できなければ empty vector の組を返す．
pair<vector<int>, vector<int>> twopaths(const vector<vector<int>> &to, const vector<int> &used_vs, int from, int to1, int to2) {
    const int N = to.size();
    const int gt = N * 2;
    atcoder::mcf_graph<int, int> graph(gt + 1);

    vector<int> valid_v(N, 1);
    for (auto i : used_vs) valid_v[i] = 0;


    valid_v[to1] = valid_v[to2] = 1;

    for (int i = 0; i < N; ++i) {
        graph.add_edge(i, i + N, valid_v[i], 0);
    }

    for (int i = 0; i < N; ++i) {
        for (auto j : to[i]) {
            int cost = 1;
            graph.add_edge(i + N, j, 1, cost);
        }
    }
    graph.add_edge(to1 + N, gt, 1, 0);
    graph.add_edge(to2 + N, gt, 1, 0);
    auto f = graph.flow(from + N, gt, 2);
    if (f.first < 2) return {{}, {}};

    vector<int> conn(N);
    for (auto e : graph.edges()) {
        if (e.flow) {
            if (e.to == gt) continue;
            int s = e.from % N, t = e.to % N;
            conn[s] ^= t;
            conn[t] ^= s;  // ループがないので生えている辺の xor だけ持っておけば後で解が復元できる
        }
    }

    vector<int> ret1, ret2;
    while (to1 != from) {
        ret1.push_back(to1);
        to1 = conn[to1];
        conn[to1] ^= ret1.back();
    }
    while (to2 != from) {
        ret2.push_back(to2);
        to2 = conn[to2];
        conn[to2] ^= ret2.back();
    }
    ret1.push_back(from);
    ret2.push_back(from);
    reverse(ret1.begin(), ret1.end());
    reverse(ret2.begin(), ret2.end());
    return {ret1, ret2};
}

constexpr int inf = 1 << 20;

int solve_fake_flow_sub(const vector<vector<int>> &to, const vector<int> &banxy, int z, int x, int y) {
    auto [p1, p2] = twopaths(to, banxy, z, x, y);
    if (p1.empty()) return inf;
    return p1.size() + p2.size() + banxy.size() - 1;
}

int solve_by_flow_twice(const vector<vector<int>> &to, int x, int y, int z) {

    int N = to.size();
    vector<int> vs{x, y, z};
    sort(vs.begin(), vs.end());
    int ret = inf;

    do {
        auto [p01, p02] = twopaths(to, {}, vs[0], vs[1], vs[2]);
        if (p01.empty() or p02.empty()) continue;
        auto ban01 = p01, ban02 = p02;
        for (int t = 0; t < 2; ++t) {
            ban01.pop_back(), ban02.pop_back();
            reverse(ban01.begin(), ban01.end()), reverse(ban02.begin(), ban02.end());
        }
        ret = min(ret, solve_fake_flow_sub(to, ban01, vs[2], vs[0], vs[1]));
        ret = min(ret, solve_fake_flow_sub(to, ban02, vs[1], vs[0], vs[2]));
    } while (next_permutation(vs.begin(), vs.end()));
    return ret <= N ? ret : -1;
}

int main() {
    cin.tie(nullptr), ios::sync_with_stdio(false);

    int N, M;
    int x, y, z;
    cin >> N >> M;
    cin >> x >> y >> z;

    --x, --y, --z;
    vector conn(N, vector<int>(N, 1));
    for (int i = 0; i < N; ++i) conn[i][i] = 0;
    while (M--) {
        int a, b;
        cin >> a >> b;
        --a, --b;
        conn[a][b] = conn[b][a] = 0;
    }
    vector<vector<int>> to(N);
    for (int i = 0; i < N; ++i) {
        for (int j = 0; j < N; ++j) {
            if (conn[i][j]) to[i].push_back(j);
        }
    }

    int ret = solve_by_flow_twice(to, x, y, z);
    cout << ret << '\n';
}

yukicoder

結果

テストケース

ソースコード