oj.uz

답안 #258922

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
258922 2020-08-06T19:33:06 Z Vimmer Domino (COCI15_domino) C++14
90 / 160
 3045 ms 54756 KB 
#include <bits/stdc++.h>
//#include <ext/pb_ds/assoc_container.hpp>
//#include <ext/pb_ds/tree_policy.hpp>

//#pragma GCC optimize("unroll-loops")
//#pragma GCC optimize("-O3")
//#pragma GCC optimize("Ofast")
//#pragma GCC optimize("fast-math")
//#pragma GCC optimize("no-stack-protector")

#define F first
#define S second
#define sz(x) int(x.size())
#define pb push_back
#define pf push_front
#define N 100050
#define M ll(1e9 + 7)
#define inf 1e9 + 1e9

using namespace std;
//using namespace __gnu_pbds;

typedef long double ld;
typedef long long ll;
typedef unsigned long long ull;
typedef short int si;
typedef array <ll, 5> a5;

//typedef tree <ll, null_type, less_equal<ll>, rb_tree_tag, tree_order_statistics_node_update> ordered_set;


bool mk[2005][2005];

int a[2005][2005], id = 2, idr[2005][2005];

ll sum, ans, dp[2005], pr[2005];

vector <a5> edges;

bool cmp(pair <int, int> x, pair <int, int> y) {return a[x.F][x.S] > a[y.F][y.S];}

vector <pair <int, int> > gr;

set <pair <int, int> > wh, bl;

void add(pair <int, int> t) {if (mk[t.F][t.S]) bl.insert({t.F, t.S}); else wh.insert({t.F, t.S});}

void add_edge(int from, int to, int cost)
{
    edges.pb({from, to, cost, 0, 1});

    edges.pb({to, from, -cost, 0, 0});
}

int main()
{
    //freopen("input.txt", "r", stdin); //freopen("output4.txt", "w", stdout);

    ios_base::sync_with_stdio(0); istream::sync_with_stdio(0); cin.tie(0); cout.tie(0);

    int n, k;

    cin >> n >> k;

    for (int i = 0; i < n; i++)
      for (int j = 0; j < n; j++)
      {
          cin >> a[i][j];

          sum += ll(a[i][j]);

          gr.pb({i, j});

          if (!mk[i][j])
          {
                if (i + 1 != n) mk[i + 1][j] = 1;

                if (j + 1 != n) mk[i][j + 1] = 1;
          }
      }

    sort(gr.begin(), gr.end(), cmp);

    for (int i = 0; i < min(300, sz(gr)); i++)
    {
        add(gr[i]);

        if (gr[i].F != 0) add({gr[i].F - 1, gr[i].S});

        if (gr[i].S != 0) add({gr[i].F, gr[i].S - 1});

        if (gr[i].F + 1 != n) add({gr[i].F + 1, gr[i].S});

        if (gr[i].S + 1 != n) add({gr[i].F, gr[i].S + 1});
    }

    for (auto it : wh)
    {
        idr[it.F][it.S] = id++;

        add_edge(0, idr[it.F][it.S], 0);
    }

    for (auto it : bl)
    {
        idr[it.F][it.S] = id++;

        add_edge(idr[it.F][it.S], 1, 0);
    }

    for (auto it : bl)
      for (auto itr : wh)
        if (abs(itr.F - it.F) + abs(itr.S - it.S) == 1)
          add_edge(idr[itr.F][itr.S], idr[it.F][it.S], -(a[itr.F][itr.S] + a[it.F][it.S]));

    for (; k > 0; k--)
    {
        for (int i = 0; i < 2005; i++) dp[i] = 1e18;

        dp[0] = 0;

        for (int it = 0; it < 30000; it++)
            for (int i = 0; i < sz(edges); i++)
            {
                int to = edges[i][1], from = edges[i][0], cost = edges[i][2], flow = edges[i][3], cap = edges[i][4];

                if (flow < cap && dp[from] < 1e18 && dp[from] + cost < dp[to])
                {
                    dp[to] = dp[from] + cost;

                    pr[to] = i;
                }
            }

        ans += dp[1];

        int v = 1;

        while (v != 0)
        {
            int id = pr[v];

            edges[id][3]++;

            edges[id ^ 1][3]--;

            v = edges[id][0];
        }
    }

    cout << sum + ans << '\n';
}

Subtask #1
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 300 ms 8940 KB Output is correct
2 Correct 246 ms 7016 KB Output is correct

Subtask #2
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 448 ms 1376 KB Output is correct
2 Correct 210 ms 1024 KB Output is correct

Subtask #3
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1672 ms 54576 KB Output is correct
2 Correct 1044 ms 52124 KB Output is correct

Subtask #4
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 138 ms 760 KB Output is correct
2 Correct 133 ms 640 KB Output is correct

Subtask #5
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 1360 ms 47296 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #6
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 1169 ms 21396 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #7
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 2364 ms 54228 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #8
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1338 ms 1376 KB Output is correct
2 Correct 629 ms 1248 KB Output is correct

Subtask #9
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 2546 ms 54560 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #10
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1640 ms 2076 KB Output is correct
2 Correct 875 ms 1656 KB Output is correct

Subtask #11
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 28 ms 384 KB Output is correct
2 Correct 34 ms 384 KB Output is correct

Subtask #12
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 2828 ms 54056 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #13
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1846 ms 1636 KB Output is correct
2 Correct 955 ms 1400 KB Output is correct

Subtask #14
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 2074 ms 20816 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #15
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 60 ms 384 KB Output is correct
2 Correct 62 ms 492 KB Output is correct

Subtask #16
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 3045 ms 54756 KB Output isn't correct
2 Halted 0 ms 0 KB -