codeforces 1221D DP

個人博客鏈接:https://blog.nuoyanli.com/2020/03/25/cf1221d/

鏈接

https://codeforces.com/problemset/problem/1221/D

題意

你有一個長度爲n的序列,每次你可以令aia_i的值加11,但需要消耗bib_i的代價。現在,你希望花費儘可能少的代價修改你的序列,使序列中任意相鄰兩項不相等。

思路

不難想到,對於一個數來說,它要麼不變,要麼加11,要麼加22,所以可以dp[n][3]:dp[n][3]:

  • dp[n][0]dp[n][0]表示aia_i不變符合條件的最小代價
  • dp[n][1]dp[n][1]表示aia_i+1符合條件的最小代價
  • dp[n][2]dp[n][2]表示aia_i+2符合條件的最小代價

狀態轉移不難想,詳細見代碼:

參考代碼

#include <bits/stdc++.h>
using namespace std;
#define IOS ios::sync_with_stdio(false), cin.tie(0)
#define endl '\n'
#define PB push_back
#define FI first
#define SE second
#define m_p(a, b) make_pair(a, b)
const double pi = acos(-1.0);
const double eps = 1e-9;
typedef long long LL;
const int N = 1e6 + 10;
const int M = 1e5 + 10;
const int mod = 1e9 + 7;
const int inf = 0x3f3f3f3f;
const double f = 2.32349;
LL dp[N][4], a[N], b[N];
LL min3(LL a, LL b, LL c) { return min(min(a, b), c); }
void solve() {
  IOS;
  int t, n;
  cin >> t;
  while (t--) {
    cin >> n;
    for (int i = 1; i <= n; i++) {
      cin >> a[i] >> b[i];
      dp[i][0] = dp[i][1] = dp[i][2] = 0;
    }
    for (int i = 1; i <= n; i++) {
      if (a[i - 1] == a[i]) {
        dp[i][0] = min(dp[i - 1][1], dp[i - 1][2]);
        dp[i][1] = min(dp[i - 1][0], dp[i - 1][2]) + b[i];
        dp[i][2] = min(dp[i - 1][0], dp[i - 1][1]) + 2 * b[i];
      } else if (a[i - 1] + 1 == a[i]) {
        dp[i][0] = min(dp[i - 1][0], dp[i - 1][2]);
        dp[i][1] = min(dp[i - 1][1], dp[i - 1][0]) + b[i];
        dp[i][2] = min3(dp[i - 1][0], dp[i - 1][1], dp[i - 1][2]) + 2 * b[i];
      } else if (a[i - 1] - 1 == a[i]) {
        dp[i][0] = min3(dp[i - 1][0], dp[i - 1][1], dp[i - 1][2]);
        dp[i][1] = min(dp[i - 1][1], dp[i - 1][2]) + b[i];
        dp[i][2] = min(dp[i - 1][0], dp[i - 1][2]) + 2 * b[i];
      } else if (a[i - 1] + 2 == a[i]) {
        dp[i][0] = min(dp[i - 1][0], dp[i - 1][1]);
        dp[i][1] = min3(dp[i - 1][0], dp[i - 1][1], dp[i - 1][2]) + b[i];
        dp[i][2] = min3(dp[i - 1][0], dp[i - 1][1], dp[i - 1][2]) + 2 * b[i];
      } else {
        dp[i][0] = min3(dp[i - 1][0], dp[i - 1][1], dp[i - 1][2]);
        dp[i][1] = min3(dp[i - 1][0], dp[i - 1][1], dp[i - 1][2]) + b[i];
        dp[i][2] = min(dp[i - 1][1], dp[i - 1][2]) + 2 * b[i];
      }
    }
    cout << min3(dp[n][0], dp[n][1], dp[n][2]) << endl;
  }
}
signed main() {
  solve();
  return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章