結果

問題 No.1479 Matrix Eraser
ユーザー hir355hir355
提出日時 2021-04-16 22:30:13
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 273 ms / 3,000 ms
コード長 2,978 bytes
コンパイル時間 2,879 ms
コンパイル使用メモリ 224,876 KB
実行使用メモリ 23,856 KB
最終ジャッジ日時 2024-07-03 02:36:25
合計ジャッジ時間 8,704 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 7 ms
15,016 KB
testcase_01 AC 7 ms
14,972 KB
testcase_02 AC 8 ms
15,008 KB
testcase_03 AC 6 ms
14,980 KB
testcase_04 AC 7 ms
14,880 KB
testcase_05 AC 7 ms
14,900 KB
testcase_06 AC 8 ms
15,016 KB
testcase_07 AC 36 ms
15,956 KB
testcase_08 AC 57 ms
16,620 KB
testcase_09 AC 110 ms
18,564 KB
testcase_10 AC 212 ms
21,120 KB
testcase_11 AC 135 ms
19,004 KB
testcase_12 AC 42 ms
16,320 KB
testcase_13 AC 55 ms
16,660 KB
testcase_14 AC 46 ms
16,448 KB
testcase_15 AC 15 ms
15,208 KB
testcase_16 AC 53 ms
16,476 KB
testcase_17 AC 257 ms
22,192 KB
testcase_18 AC 263 ms
22,280 KB
testcase_19 AC 253 ms
22,208 KB
testcase_20 AC 252 ms
22,308 KB
testcase_21 AC 258 ms
22,200 KB
testcase_22 AC 249 ms
22,144 KB
testcase_23 AC 260 ms
22,136 KB
testcase_24 AC 273 ms
22,144 KB
testcase_25 AC 261 ms
22,204 KB
testcase_26 AC 255 ms
22,252 KB
testcase_27 AC 131 ms
18,912 KB
testcase_28 AC 134 ms
19,012 KB
testcase_29 AC 131 ms
19,212 KB
testcase_30 AC 128 ms
19,028 KB
testcase_31 AC 136 ms
19,140 KB
testcase_32 AC 81 ms
23,368 KB
testcase_33 AC 76 ms
23,396 KB
testcase_34 AC 77 ms
23,424 KB
testcase_35 AC 82 ms
23,400 KB
testcase_36 AC 79 ms
23,456 KB
testcase_37 AC 42 ms
15,896 KB
testcase_38 AC 138 ms
18,116 KB
testcase_39 AC 250 ms
23,856 KB
testcase_40 AC 7 ms
15,000 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <atcoder/maxflow>
#include <bits/stdc++.h>
using namespace std;
using namespace atcoder;
const long long INF_LL = 2000000000000000000LL;
const int INF = 2000000000;
const long long MOD = 1000000007;
#define ll long long
#define all(x) x.begin(), x.end()
#define REP(i, a, b) for(int i = a; i < b; i++)
#define rep(i, n) REP(i, 0, n)
// typedef float double;
// typedef priority_queue prique;
typedef pair<ll, ll> P;
typedef vector<int> vi;
typedef vector<vi> vvi;
typedef vector<P> vp;
typedef vector<ll> vl;
typedef vector<vi> matrix;
int dx[4] = {0, -1, 0, 1};
int dy[4] = {1, 0, -1, 0};
int sign[2] = {1, -1};
template <class T> bool chmax(T &a, T b) {
    if(a < b) {
        a = b;
        return 1;
    }
    return 0;
}
template <class T> bool chmin(T &a, T b) {
    if(a > b) {
        a = b;
        return 1;
    }
    return 0;
}

ll modpow(ll a, ll b, ll m) {
    if(b == 0)
        return 1;
    ll t = modpow(a, b / 2, m);
    if(b & 1) {
        return (t * t % m) * a % m;
    } else {
        return t * t % m;
    }
}

struct edge {
    int to;
    ll cost;
    edge(int t, ll c) { to = t, cost = c; }
};

typedef vector<vector<edge>> graph;

// using mint = modint998244353;

int main(){
    int h, w;
    cin >> h >> w;
    vector<vector<int>> a(h, vector<int>(w));
    rep(i, h) rep(j, w) cin >> a[i][j];
    vector<vector<pair<int, int>>> t(500000);
    rep(i, h){
        rep(j, w){
            if(a[i][j] > 0) {
                t[a[i][j] - 1].push_back({i, j});
            }
        }
    }
    int ans = 0;
    for(int i = 499999; i >= 0; i--){
        if(t[i].size() > 0){
            int tmp1;
            sort(all(t[i]));
            int tmp = t[i][0].first;
            t[i][0].first = 0;
            rep(j, t[i].size() - 1){
                if(tmp < t[i][j + 1].first){
                    tmp = t[i][j + 1].first;
                    t[i][j + 1].first = t[i][j].first + 1;
                }else{
                    t[i][j + 1].first = t[i][j].first;
                }
            }
            tmp1 = t[i][t[i].size()-1].first + 1;
            int tmp2;
            sort(all(t[i]), [](auto x, auto y){return x.second < y.second;});
            tmp = t[i][0].second;
            t[i][0].second = 0;
            rep(j, t[i].size() - 1){
                if(tmp < t[i][j + 1].second){
                    tmp = t[i][j + 1].second;
                    t[i][j + 1].second = t[i][j].second + 1;
                }else{
                    t[i][j + 1].second = t[i][j].second;
                }
            }
            tmp2 = t[i][t[i].size()-1].second + 1;
            mf_graph<int> g(tmp1 + tmp2 + 2);
            rep(j, tmp1) g.add_edge(tmp1 + tmp2, j, 1);
            rep(j, tmp2) g.add_edge(j + tmp1, tmp1 + tmp2 + 1, 1);
            rep(j, t[i].size()){
                g.add_edge(t[i][j].first, t[i][j].second + tmp1, 1);
            }
            ans += g.flow(tmp1 + tmp2, tmp1 + tmp2 + 1);
        }
    }
    cout << ans << endl;
}
0