結果

問題 No.1479 Matrix Eraser
コンテスト
ユーザー trineutron
提出日時 2021-04-17 01:03:42
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 630 ms / 3,000 ms
コード長 1,665 bytes
コンパイル時間 3,128 ms
コンパイル使用メモリ 218,800 KB
最終ジャッジ日時 2025-01-20 20:52:47
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 39
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/maxflow>

using namespace std;
using namespace atcoder;

int main()
{
    int h, w;
    cin >> h >> w;
    vector a(h, vector<int>(w));
    for (int i = 0; i < h; i++)
    {
        for (int j = 0; j < w; j++)
        {
            cin >> a.at(i).at(j);
        }
    }
    unordered_map<int, vector<pair<int, int>>> m;
    for (int i = 0; i < h; i++)
    {
        for (int j = 0; j < w; j++)
        {
            if (a.at(i).at(j))
            {
                m[a.at(i).at(j)].emplace_back(i, h + j);
            }
        }
    }
    int ans = 0;
    for (auto &&[i, v] : m)
    {
        unordered_map<int, int> newindex;
        for (auto &&k : v)
        {
            if (newindex[k.first] == 0)
            {
                newindex[k.first] = newindex.size();
            }
            if (newindex[k.second] == 0)
            {
                newindex[k.second] = newindex.size();
            }
        }
        const int n = newindex.size();
        int s = 0, t = n + 1;
        mf_graph<int> flow(n + 2);
        vector<bool> used(n + 2);
        for (auto &&k : v)
        {
            flow.add_edge(newindex[k.first], newindex[k.second], 1);
            if (not used.at(newindex[k.first]))
            {
                flow.add_edge(s, newindex[k.first], 1);
                used.at(newindex[k.first]) = true;
            }
            if (not used.at(newindex[k.second]))
            {
                flow.add_edge(newindex[k.second], t, 1);
                used.at(newindex[k.second]) = true;
            }
        }
        ans += flow.flow(s, t);
    }
    cout << ans << endl;
    return 0;
}
0