题目描述
小Q和牛博士合唱一首歌曲,这首歌曲由n个音调组成,每个音调由一个正整数表示。
对于每个音调要么由小Q演唱要么由牛博士演唱,对于一系列音调演唱的难度等于所有相邻音调变化幅度之和, 例如一个音调序列是8, 8, 13, 12, 那么它的难度等于|8 - 8| + |13 - 8| + |12 - 13| = 6(其中||表示绝对值)。
现在要对把这n个音调分配给小Q或牛博士,让他们演唱的难度之和最小,请你算算最小的难度和是多少。
输入描述
输入包括两行,第一行一个正整数n(1 ≤ n ≤ 2000) 第二行n个整数v[i](1 ≤ v[i] ≤ 10^6), 表示每个音调。
输出描述
输出一个整数,表示小Q和牛博士演唱最小的难度和是多少。
测试用例
input:
5
1 5 6 2 1
output:
3
解题思路
题目太难了,参考了别人的思路,序列dp问题。
- 可先设 dp[i][j] 表示当前小Q唱到第 i 个音调,牛博士唱到第 j 个音调的难度和;
不妨设当前 i > j ;若 i - 1 == j 则发生换人,由于不知道上一次 i 唱到哪里,状态由 min{ dp[k][j] + abs(v[i] - v[k]) }, k < j 转移来;
若 i - 1 > j 则表示当前是从 i - 1 唱到 i 的,没有换人,状态由 dp[i-1][j] + abs(v[i] - v[i-1]) 累加;
不妨设 dp[i][j] 表示当前演唱到第 i 个,上一个人演唱到第 j 个,则状态转移方程为
dp[i][j] = dp[i-1][j] + abs(v[i] - v[i-1]), j < i - 1
dp[i][i -1] = min{ dp[i-1][k] + abs(v[i] - v[k]) }, k < i - 1
- 初始情况是若当前有 i 个音调,可以让一个人只唱第一个或最后一个音调,剩下的音调都由另一个人唱:
dp[i][0] = dp[i-1][0] + abs(v[i] - v[i-1]), i ≥ 2
dp[i][i-1] = dp[i-1][i-2] + abs(v[i-1] - v[i-2]), i ≥ 2
代码
#include <iostream>
#include <cmath>
#include <limits.h>
#include <vector>
#include <algorithm>
using namespace std;
const int MAXN = 2010;
int n;
int a[MAXN],cost[MAXN];
int dp[MAXN][MAXN];
int main()
{
int n;
cin >> n;
for (int i = 0; i < n; i++) {
cin >> a[i];
cost[i] = abs(a[i] - a[i - 1]);
}
for (int i = 2; i < n; ++i){
dp[i][0] = dp[i - 1][0] + cost[i],
dp[i][i - 1] = dp[i - 1][i - 2] + cost[i - 1];
}
for (int i = 2; i<n; ++i)
for (int j = 0; j<i - 1; ++j) {
dp[i][j] = dp[i - 1][j] + cost[i];
dp[i][i - 1] = min(dp[i][i - 1], dp[i - 1][j] + abs(a[i] - a[j]));
}
int ans = INT_MAX;
for (int i = 0; i < n - 1; i++) ans = min(ans, dp[n - 1][i]);
cout << ans;
return 0;
}