結果

問題 No.1606 Stuffed Animals Keeper
ユーザー simansiman
提出日時 2023-05-02 11:34:43
言語 C++17(clang)
(17.0.6 + boost 1.83.0)
結果
AC  
実行時間 35 ms / 3,000 ms
コード長 1,581 bytes
コンパイル時間 1,203 ms
コンパイル使用メモリ 105,472 KB
実行使用メモリ 35,856 KB
最終ジャッジ日時 2023-08-13 09:18:06
合計ジャッジ時間 2,867 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,384 KB
testcase_01 AC 2 ms
4,380 KB
testcase_02 AC 2 ms
4,380 KB
testcase_03 AC 16 ms
16,572 KB
testcase_04 AC 19 ms
20,012 KB
testcase_05 AC 4 ms
6,024 KB
testcase_06 AC 7 ms
8,612 KB
testcase_07 AC 13 ms
14,468 KB
testcase_08 AC 12 ms
13,392 KB
testcase_09 AC 6 ms
7,596 KB
testcase_10 AC 4 ms
5,444 KB
testcase_11 AC 3 ms
4,380 KB
testcase_12 AC 7 ms
8,660 KB
testcase_13 AC 3 ms
4,384 KB
testcase_14 AC 2 ms
4,384 KB
testcase_15 AC 3 ms
4,384 KB
testcase_16 AC 25 ms
24,740 KB
testcase_17 AC 25 ms
25,000 KB
testcase_18 AC 26 ms
25,024 KB
testcase_19 AC 25 ms
24,976 KB
testcase_20 AC 25 ms
24,716 KB
testcase_21 AC 6 ms
6,548 KB
testcase_22 AC 5 ms
5,972 KB
testcase_23 AC 5 ms
6,076 KB
testcase_24 AC 6 ms
6,504 KB
testcase_25 AC 6 ms
6,232 KB
testcase_26 AC 3 ms
4,380 KB
testcase_27 AC 3 ms
4,380 KB
testcase_28 AC 3 ms
4,384 KB
testcase_29 AC 4 ms
4,384 KB
testcase_30 AC 4 ms
4,384 KB
testcase_31 AC 3 ms
4,380 KB
testcase_32 AC 3 ms
4,380 KB
testcase_33 AC 3 ms
4,384 KB
testcase_34 AC 3 ms
4,384 KB
testcase_35 AC 3 ms
4,380 KB
testcase_36 AC 3 ms
4,380 KB
testcase_37 AC 3 ms
4,384 KB
testcase_38 AC 3 ms
4,384 KB
testcase_39 AC 3 ms
4,380 KB
testcase_40 AC 3 ms
4,384 KB
testcase_41 AC 2 ms
4,380 KB
testcase_42 AC 3 ms
4,380 KB
testcase_43 AC 35 ms
35,816 KB
testcase_44 AC 35 ms
35,844 KB
testcase_45 AC 35 ms
35,816 KB
testcase_46 AC 35 ms
35,824 KB
testcase_47 AC 34 ms
35,856 KB
testcase_48 AC 2 ms
4,380 KB
testcase_49 AC 1 ms
4,384 KB
testcase_50 AC 1 ms
4,384 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cassert>
#include <cmath>
#include <algorithm>
#include <iostream>
#include <iomanip>
#include <climits>
#include <map>
#include <queue>
#include <set>
#include <cstring>
#include <vector>

using namespace std;
typedef long long ll;

int main() {
  int N;
  cin >> N;

  int A[N];
  for (int i = 0; i < N; ++i) {
    cin >> A[i];
  }

  int zero_cnt = 0;
  int one_cnt = 0;
  int cnt = 0;
  int div_cnt = 0;
  vector<int> swap_cnt;
  vector<int> div_size;

  for (int i = 0; i < N; ++i) {
    int a = A[i];

    if (a == 2) {
      if (cnt > 0) {
        swap_cnt.push_back(one_cnt);
        div_size.push_back(cnt);
        div_cnt++;
        cnt = 0;
        one_cnt = 0;
      }
    } else {
      cnt++;
      if (a == 1) {
        one_cnt++;
      } else {
        zero_cnt++;
      }
    }
  }

  if (cnt > 0) {
    div_cnt++;
    swap_cnt.push_back(one_cnt);
    div_size.push_back(cnt);
  }

  vector<vector<int>> dp(div_cnt + 1, vector<int>(N + 1, INT_MAX));
  dp[0][0] = 0;
  int ans = dp[0][zero_cnt];

  for (int i = 1; i <= div_cnt; ++i) {
    int size = div_size[i - 1];
    int sc = swap_cnt[i - 1];

    dp[i] = dp[i - 1];
    // fprintf(stderr, "i: %d, size: %d, sc: %d\n", i, size, sc);

    for (int len = 0; len < zero_cnt; ++len) {
      if (dp[i - 1][len] == INT_MAX) continue;

      int n_size = len + size;
      int n_sc = dp[i - 1][len] + sc;

      dp[i][n_size] = min(dp[i][n_size], n_sc);
    }

    ans = min(ans, dp[i][zero_cnt]);
  }

  if (ans == INT_MAX) {
    cout << -1 << endl;
  } else {
    cout << ans << endl;
  }

  return 0;
}
0