洛谷P1220关路灯题解
此题是一个状态转移方程还算比较多的一个区间DP,这个题也能启示我们如果某个状态不能够很好地解决问题,那么不妨试试再加一维,而且如果转移顺序不确定的话,可以试试记忆化搜索,说不定就可以比较容易的写出状态转移方程和状态转移。
状态
如果我们要关掉\(i\)到\(j\)的灯,则最后一个关掉的灯就是\(i\)或者\(j\),我们设\(dp[i][j][0/1]\)分别表示最后一个关掉的灯是\(i\)还是\(j\),\(dp[i][j][0]\)一定是由\(dp[i+1][j][0/1]\)来转移过来的,所以我们就可以分类讨论,所以很容易得出
状态转移方程
dp[i][j][0] = min(dp[i][j][0], dp[i + 1][j][0] + (data[i + 1].pos - data[i].pos) * (data[n].sum - data[j].sum + data[i].sum));
dp[i][j][0] = min(dp[i][j][0], dp[i + 1][j][1] + (data[j].pos - data[i].pos) * (data[n].sum - data[j].sum + data[i].sum));
然后就可以按照区间DP的基本思路来求解了。
\(Code\)
#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <cmath>
#include <cstdlib>
using namespace std;
int n, c;
struct light {
int pow, pos, sum;
}data[100100];
int dp[1010][1010][3];
int main()
{
scanf("%d%d", &n, &c);
for (int i = 1; i <= n; i++)
scanf("%d%d", &data[i].pos, &data[i].pow), data[i].sum = data[i - 1].sum + data[i].pow;
memset(dp, 124, sizeof(dp));
dp[c][c][0] = dp[c][c][1] = 0;
for (int l = 2; l <= n; l++)
for (int i = 1; i + l - 1 <= n; i++)
{
int j = i + l - 1;
dp[i][j][1] = min(dp[i][j][1], dp[i][j - 1][0] + (data[j].pos - data[i].pos) * (data[n].sum - data[j - 1].sum + data[i - 1].sum));
dp[i][j][1] = min(dp[i][j][1], dp[i][j - 1][1] + (data[j].pos - data[j - 1].pos) * (data[n].sum - data[j - 1].sum + data[i - 1].sum)) ;
}
/*
for (int i = 1; i <= n; i++)
for (int j = i; j <= n; j++)
printf("%d %d\n", dp[i][j][0], dp[i][j][1]);
*/
printf("%d", min(dp[1][n][0], dp[1][n][1]));
return 0;
}