結果

問題 No.2288 Somen Sliders
コンテスト
ユーザー みここ
提出日時 2023-03-08 02:10:31
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
(最新)
AC  
(最初)
実行時間 -
コード長 3,921 bytes
コンパイル時間 1,223 ms
コンパイル使用メモリ 93,092 KB
最終ジャッジ日時 2025-02-11 06:40:12
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 1 RE * 2
other AC * 1 WA * 10 RE * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <iostream>
#include <cassert>
#include <queue>
using namespace std;
typedef long long ll;

template <typename T = int>
struct stable_matching
{
public:
    struct Edge
    {
        int a, b;
        int preference_a, preference_b; // The smaller the value, the higher the priority.
        bool is_matched;
        T cost;
        int id;

        bool operator<(const Edge &e) const
        {
            return preference_a < e.preference_a;
        }
    };

    stable_matching() {}

    stable_matching(int p, int q) : p(p), q(q), m(0)
    {
        g.resize(p);
    }

    void add_directed_edge(int a, int b, int preference_a, int preference_b, T cost = 1)
    {
        assert(0 <= a && a < p);
        assert(0 <= b && b < q);
        epos.push_back(std::pair<int, int>(a, (int)g[a].size()));
        g[a].push_back((Edge){a, b, preference_a, preference_b, false, cost, m++});
    }

    void get_stable_matching()
    {
        for (int a = 0; a < p; a++)
        {
            if (!std::is_sorted(g[a].begin(), g[a].end()))
            {
                std::sort(g[a].begin(), g[a].end());
            }
        }
        std::vector<std::pair<int, int>> match(q, std::pair<int, int>(-1, -1));
        std::vector<int> nx(p, 0);
        std::queue<int> que;
        for (int a = 0; a < p; a++)
        {
            que.push(a);
        }
        while (que.size())
        {
            int a = que.front();
            que.pop();
            for (int &i = nx[a]; i < (int)g[a].size(); i++)
            {
                int b = g[a][i].b;
                int x = match[b].first;
                int j = match[b].second;
                if (x >= 0 && g[a][i].preference_b < g[x][j].preference_b)
                {
                    g[x][j].is_matched = false;
                    nx[x]++;
                    que.push(x);
                    x = -1;
                }
                if (x == -1)
                {
                    g[a][i].is_matched = true;
                    match[b] = std::pair<int, int>(a, i);
                    break;
                }
            }
        }
    }

    std::vector<Edge> get_matching_edges()
    {
        std::vector<Edge> res;
        for(int i = 0; i < m; i++)
        {
            Edge e = g[epos[i].first][epos[i].second];
            if(e.is_matched)
            {
                res.push_back(e);
            }
        }
        return res;
    }

    Edge &get_edge(int id)
    {
        assert(id < m);
        return g[epos[id].first][epos[id].second];
    }

    int get_size()
    {
        return p + q;
    }

    int get_left_size()
    {
        return p;
    }

    int get_right_size()
    {
        return q;
    }

    int get_edge_size()
    {
        return m;
    }

private:
    int p, q, m;
    std::vector<std::vector<Edge>> g;
    std::vector<std::pair<int, int>> epos;
};

void solve()
{
    int n, x, y;
    cin >> n >> x >> y;
    int p[200005], d[200005];
    for (int u = 0; u < n; u++)
    {
        p[u] = -1;
    }
    int k[200005];
    for (int i = 0; i < x; i++)
    {
        cin >> k[i];
        for (int j = 0; j <= k[i]; j++)
        {
            int u;
            cin >> u;
            u--;
            p[u] = i;
            d[u] = j;
        }
    }
    stable_matching<int> g(y, x);
    int m = 0;
    for (int i = 0; i < y; i++)
    {
        int l;
        cin >> l;
        for (int j = 0; j <= l; j++)
        {
            int v;
            cin >> v;
            v--;
            if (p[v] >= 0)
            {
                g.add_directed_edge(i, p[v], -j, d[v]);
            }
        }
    }
    g.get_stable_matching();
    int ans = 0;
    for (int i = 0; i < x; i++)
    {
        ans += k[i];
    }
    auto M = g.get_matching_edges();
    for (auto &e : M)
    {
        ans -= k[e.b] - e.preference_b;
    }
    cout << ans << endl;
}

int main()
{
    solve();
}
0