线性DP——D1. Hot Start Up (easy version) (easy version){CF854(div1+div2)}

传送门:https://codeforces.com/contest/1799/problem/D1

思路:(根据佬的题解):

这道题目是一个将一条序列分成两个子序列并且要求目标结果最优的问题，有一个通用的状态表示方法，当第i个物品放入其中一条序列后另一条序列的末尾元素是什么？

在本题当中按照题意本应该定义f[i][j][k]表示放完第i个物品并且第一条序列末尾是j并且第二条序列末尾是k时最小的花费，这样会变成O(n^3)的时间复杂度，但是放完第i个物品时就可以知道其中一个的末尾元素了，因此数组变成f[i][j],表示放完第i个物品后其中一条序列后另一条子序列的结尾是j时的最小花费。

由此得到两条状态转移方程，分别是第i个物品放在第i-1个物品后面时，以及第i个物品不放在第i-1个物品后面时.

第一种:放在第i-1个元素的后面

f[i][j] = min(f[i][j], f[i - 1][j] + (a[i] == a[i - 1] ? hot[a[i]] : cold[a[i]]));

第二种:不放在第i-1个元素的后面，此时有其中一条子序列的末尾肯定是a[i-1]
f[i][a[i - 1]] = min(f[i][a[i - 1]], f[i - 1][j] + (a[i] == j ? hot[a[i]] : cold[a[i]]));

代码:

// Problem: D1. Hot Start Up (easy version)
// Contest: Codeforces - Codeforces Round #854 by cybercats (Div. 1 + Div. 2)
// URL: https://codeforces.com/contest/1799/problem/D1
// Memory Limit: 512 MB
// Time Limit: 1000 ms
//
// Powered by CP Editor (https://cpeditor.org)

#include <bits/stdc++.h>
using namespace std;
#define int long long
typedef pair<int, int> PII;
const int N = 2e5 + 10, mod = 1e18;
int a[N];
int n, t, s, m, k;
int last[4];
int first[N];
int cold[N];
int hot[N];
int f[5010][5010];

signed main() {
  int t;
  scanf("%lld", &t);
  while (t--) {
    scanf("%lld%lld", &n, &m);
    for (int i = 1; i <= n; i++) {
      scanf("%lld", &a[i]);
    }
    for (int i = 0; i <= n; i++)
      for (int j = 0; j <= m; j++) f[i][j] = mod;

    for (int i = 1; i <= m; i++) scanf("%lld", &cold[i]);
    for (int i = 1; i <= m; i++) scanf("%lld", &hot[i]);
    f[1][0] = cold[a[1]];

    for (int i = 2; i <= n; i++)
      for (int j = 0; j <= m; j++) {
        f[i][j] = min(
            f[i][j], f[i - 1][j] + (a[i] == a[i - 1] ? hot[a[i]] : cold[a[i]]));
        f[i][a[i - 1]] = min(
            f[i][a[i - 1]], f[i - 1][j] + (a[i] == j ? hot[a[i]] : cold[a[i]]));
      }
    int ans = mod;
    for (int i = 0; i <= m; i++) ans = min(ans, f[n][i]);
    printf("%lld\n", ans);
  }
  return 0;
}