結果

問題 No.957 植林
ユーザー a5ob7ra5ob7r
提出日時 2019-12-20 01:20:37
言語 C++11
(gcc 11.4.0)
結果
WA  
実行時間 -
コード長 1,694 bytes
コンパイル時間 1,303 ms
コンパイル使用メモリ 77,596 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-07-07 02:22:27
合計ジャッジ時間 5,229 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,812 KB
testcase_01 AC 1 ms
6,944 KB
testcase_02 AC 1 ms
6,944 KB
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 RE -
testcase_09 RE -
testcase_10 WA -
testcase_11 RE -
testcase_12 AC 24 ms
6,940 KB
testcase_13 RE -
testcase_14 RE -
testcase_15 WA -
testcase_16 RE -
testcase_17 RE -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
testcase_34 WA -
testcase_35 WA -
testcase_36 WA -
testcase_37 WA -
testcase_38 WA -
testcase_39 WA -
testcase_40 WA -
testcase_41 RE -
testcase_42 RE -
testcase_43 AC 39 ms
6,940 KB
testcase_44 AC 39 ms
6,940 KB
testcase_45 RE -
testcase_46 WA -
testcase_47 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <set>
#include <map>
#include <cmath>
#include <algorithm>
#include <climits>

using namespace std;

typedef long long ll;

/* {{{ Variables */
ll H, W;
vector<vector<ll>> G;
vector<ll> R, C;
/* }}} */

int argmax(vector<ll> v) {
  int len = v.size();
  ll max_val = LLONG_MIN;
  int max_arg = -1;

  for (int i = 0; i < len; ++i) {
    if (max_val < v[i]) {
      max_val = v[i];
      max_arg = i;
    }
  }

  return max_arg;
}

int main(int argc, char *argv[])
{
  /* {{{ Input */
  cin >> H >> W;
  for (int i = 0; i < H; ++i) {
    ll g;
    G.emplace_back(vector<ll>());
    for (int j = 0; j < W; ++j) {
      cin >> g;
      G[i].emplace_back(g);
    }
  }
  for (int i = 0; i < H; ++i) {
    ll r;
    cin >> r;
    R.emplace_back(r);
  }
  for (int i = 0; i < W; ++i) {
    ll c;
    cin >> c;
    C.emplace_back(c);
  }
  /* }}} */

  /* {{{ Output */
  vector<ll> ROW(R), COL(C);
  for (int i = 0; i < H; ++i) {
    for (int j = 0; j < W; ++j) {
      ROW[i] -= G[i][j];
      COL[j] -= G[i][j];
    }
  }

  ll cost_sum = 0;

  while (1) {
    ll r_argmax = argmax(ROW);
    ll c_argmax = argmax(COL);
    ll cost;

    if (ROW[r_argmax] > COL[c_argmax]) {
      cost = ROW[r_argmax];
      ROW[r_argmax] = LLONG_MIN;
      for (int i = 0; i < W; ++i) {
        COL[i] += G[r_argmax][i];
        G[r_argmax][i] = 0;
      }
    } else {
      cost = COL[c_argmax];
      COL[c_argmax] = LLONG_MIN;
      for (int i = 0; i < H; ++i) {
        ROW[i] += G[i][c_argmax];
        G[i][c_argmax] = 0;
      }
    }

    if (cost < 0) {
      break;
    }

    cost_sum += cost;
  }

  cout << cost_sum << endl;
  /* }}} */

  return 0;
}
0