結果

問題 No.2759 Take Pictures, Elements?
ユーザー kekenx
提出日時 2024-05-25 23:44:19
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 94 ms / 2,000 ms
コード長 793 bytes
コンパイル時間 874 ms
コンパイル使用メモリ 105,084 KB
実行使用メモリ 7,844 KB
最終ジャッジ日時 2025-06-20 13:21:19
合計ジャッジ時間 1,771 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <cassert>
#include <vector>
#include <map>
#include <cmath>

using namespace std;

int main() {
  int N, Q; cin >> N >> Q;
  vector<int> A(N);
  map<int, vector<int>> mp;
  vector<vector<int>> dp(Q + 1, vector<int>(N, 2e9));
  for (int i = 0; i < N; i++) {
    cin >> A[i];
    mp[A[i]].push_back(i);
  }

  dp[0][0] = 0;
  int prv = -1;
  for (int i = 0; i < Q; i++) {
    int b; cin >> b;
    if (i == 0) {
      for (int j: mp[b]) {
	dp[i + 1][j] = min(dp[i + 1][j], dp[i][0] + j);
      }
    } else {
      for (int j: mp[prv]) {
	for (int k: mp[b]) {
	  dp[i + 1][k] = min(dp[i + 1][k], dp[i][j] + (abs(j - k)));
	}
      }
    }
    prv = b;
  }

  int mi = 2e9;
  for (int i: mp[prv]) {
    mi = min(mi, dp[Q][i]);
  }
  cout << mi << endl;
  return 0;
}

0