题目:
小Q和牛博士合唱一首歌曲,这首歌曲由n个音调组成,每个音调由一个正整数表示。
对于每个音调要么由小Q演唱要么由牛博士演唱,对于一系列音调演唱的难度等于所有相邻音调变化幅度之和, 例如一个音调序列是8, 8, 13, 12, 那么它的难度等于|8 - 8| + |13 - 8| + |12 - 13| = 6(其中||表示绝对值)。
现在要对把这n个音调分配给小Q或牛博士,让他们演唱的难度之和最小,请你算算最小的难度和是多少。
如样例所示: 小Q选择演唱{5, 6}难度为1, 牛博士选择演唱{1, 2, 1}难度为2,难度之和为3,这一个是最小难度和的方案了。
输入描述:
输入包括两行,第一行一个正整数n(1 ≤ n ≤ 2000) 第二行n个整数v[i](1 ≤ v[i] ≤ 10^6), 表示每个音调。
输出描述:
输出一个整数,表示小Q和牛博士演唱最小的难度和是多少。
输入例子1:
5
1 5 6 2 1
输出例子1:
3
程序:
c版本
#include <stdio.h>
#include <stdlib.h>
typedef long long llong;
inline void getMin(llong& n, llong x) {
n > x && (n = x);
}
#define MAXN 2020
int n;
int v[MAXN], cost[MAXN];
void read() {
scanf_s("%d%d", &n, v);
for (int i = 1; i < n; ++i) {
scanf_s("%d", v + i);
cost[i] = abs(v[i] - v[i - 1]);
}
}
llong dp[MAXN][MAXN];
void work() {
llong res = (1ll << 63) - 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];
getMin(dp[i][i - 1], dp[i - 1][j] + abs(v[i] - v[j]));
}
}
for (int i = 0; i < n - 1; ++i) {
getMin(res, dp[n - 1][i]);
}
printf("%lld\n", res);
}
int main() {
read();
work();
return 0;
}
c++版本(此版本只能AC70%,未找到原因)
#include<iostream>
#include<vector>
#include<cmath>
using namespace std;
int main()
{
int n;
cin >>n ;
vector<long> v(n, 0);
for (int i = 0; i < n; i++)
cin >> v[i];
vector<vector<long>> dp(n, vector<long>(n, 0));
for (int i = 2; i < n; i++)
{
//dp[i][0] = dp[i - 1][0] + abs(v[i] - v[i - 1]);
dp[i][i - 1] = dp[i - 1][i - 2] + abs(v[i-1] - v[i - 2]);
}
for (int i = 2; i < n; i++)
{
long min = dp[i][i - 1];
for (int j = 0; j < i - 1; j++)
{
dp[i][j] = dp[i - 1][j] + abs(v[i] - v[i - 1]);
if (dp[i - 1][j] + abs(v[i] - v[j]) < min)
min = dp[i - 1][j] + abs(v[i] - v[j]);
}
dp[i][i - 1] = min;
}
long res =1000000;
for (int i = 0; i < n-1; i++)
if (dp[n - 1][i] < res)
res = dp[n - 1][i];
cout<<res<<endl;
return 0;
}
点评:
此题采用动态规划
容易想到一个人继续演唱或换人演唱的时候发生状态转移:
可先设 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