結果

問題 No.2759 Take Pictures, Elements?
ユーザー テナガザルテナガザル
提出日時 2024-05-17 21:47:18
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 27 ms / 2,000 ms
コード長 1,161 bytes
コンパイル時間 838 ms
コンパイル使用メモリ 86,752 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-05-17 21:47:20
合計ジャッジ時間 2,357 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 19 ms
6,812 KB
testcase_01 AC 27 ms
6,940 KB
testcase_02 AC 2 ms
6,944 KB
testcase_03 AC 1 ms
6,944 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 2 ms
6,944 KB
testcase_06 AC 2 ms
6,944 KB
testcase_07 AC 1 ms
6,940 KB
testcase_08 AC 1 ms
6,940 KB
testcase_09 AC 8 ms
6,944 KB
testcase_10 AC 8 ms
6,940 KB
testcase_11 AC 8 ms
6,944 KB
testcase_12 AC 8 ms
6,944 KB
testcase_13 AC 9 ms
6,944 KB
testcase_14 AC 10 ms
6,940 KB
testcase_15 AC 9 ms
6,940 KB
testcase_16 AC 10 ms
6,940 KB
testcase_17 AC 10 ms
6,940 KB
testcase_18 AC 10 ms
6,944 KB
testcase_19 AC 9 ms
6,940 KB
testcase_20 AC 10 ms
6,948 KB
testcase_21 AC 2 ms
6,944 KB
testcase_22 AC 1 ms
6,944 KB
testcase_23 AC 1 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int main()
{
  int n, q;
  cin >> n >> q;
  vector<int> a(n);
  for (int i = 0; i < n; ++i) cin >> a[i];
  auto all = a;
  sort(all.begin(), all.end());
  all.erase(unique(all.begin(), all.end()), all.end());
  int siz = all.size();
  vector<vector<int>> id(siz);
  for (int i = 0; i < n; ++i)
  {
    int idx = lower_bound(all.begin(), all.end(), a[i]) - all.begin();
    id[idx].push_back(i);
  }
  vector<int> dp(n, 1e9);
  dp[0] = 0;
  while (q--)
  {
    int b;
    cin >> b;
    int idx = lower_bound(all.begin(), all.end(), b) - all.begin();
    vector<int> ndp(n, 1e9);
    for (int i = 0; i < n; ++i)
    {
      int id2 = lower_bound(id[idx].begin(), id[idx].end(), i) - id[idx].begin();
      int li = -1, ri = -1;
      if (id2 < id[idx].size()) ri = id[idx][id2];
      --id2;
      if (id2 >= 0) li = id[idx][id2];
      if (li >= 0) ndp[li] = min(ndp[li], dp[i] + abs(li - i));
      if (ri < n) ndp[ri] = min(ndp[ri], dp[i] + abs(ri - i));
    }
    swap(ndp, dp);
  }
  int ans = 1e9;
  for (int i = 0; i < n; ++i) ans = min(ans, dp[i]);
  cout << ans << endl;
}
0