算法区间DP- 机器人关灯

题意：

在X轴上有N个灯泡，让机器人去关灯，要求耗费的电量最少，机器人速度恒定1cm/s。例如case1， 机器人在位置5，两盏灯分别在3和7位置。

step1: 机器人从5出发向右走两步到7，此时有两盏灯亮着，耗费 （7-5）*2=4w

step2：机器人从7向左走，（7-3）*1 = 4，

共耗费4+4 = 8

[输入] 第一行是case数，

每个case，第一个是灯泡的数量N (1<=N<=1000)，以及机器人开始所在的位置S (1<=S<=1000000)。从第三行开始是每个灯泡的位置

[输出] 最小耗电量

输入：

2
2 5
3
7
4 10
1
9
11
19

Output
8
44

/*



*/
#include <stdio.h>
#include <algorithm>

#define MAX_N 1005
#define INF 0x7FFFFFFF

int n;			// x轴上共有多少个位置
int X;			// 机器人初始所在位置
int a[MAX_N];	// 存储灯泡的位置信息
unsigned int dp[MAX_N][MAX_N][2];

void DP()
{
	int iRobot;
	int i, j;
	for (i = 1; i <= n; i++)
	{
		for (j = 1; j <= n; j++)
		{
			dp[i][j][0] = dp[i][j][1] = INF;
		}
	}

	for (i = 1; i <= n; i++)
	{
		if (a[i] == X)
		{
			iRobot = i;
			break;
		}
	}

	dp[iRobot][iRobot][0] = dp[iRobot][iRobot][1] = 0;
	for (i = iRobot; i >= 1; i--)
	{
		for (j = iRobot; j <= n; j++)
		{
			int delay = n - (j - i);
			if (i == j) continue;
			dp[i][j][0] = std::min(dp[i][j][0], dp[i + 1][j][0] + delay*(a[i + 1] - a[i]));
			dp[i][j][0] = std::min(dp[i][j][0], dp[i + 1][j][1] + delay*(a[j] - a[i]));
			dp[i][j][1] = std::min(dp[i][j][1], dp[i][j - 1][0] + delay*(a[j] - a[i]));
			dp[i][j][1] = std::min(dp[i][j][1], dp[i][j - 1][1] + delay*(a[j] - a[j-1]));
		}
	}
}

int main()
{
	//freopen("switchoff_input.txt", "r", stdin);
	int T;
	scanf("%d", &T);
	int i,tc;
	for (tc = 0; tc < T; tc++)
	{
		scanf("%d", &n);
		scanf("%d", &X);
		// 读入灯泡的位置
		for (i = 1; i <= n; i++)
		{
			scanf("%d", &(a[i]));
		}
		// 加上机器人的位置
		a[n + 1] = X;
		n++;

		// 将灯泡按照x轴的位置信息从小到大排序
		std::sort(a + 1, a + n + 1);

		DP();
		int ans = std::min(dp[1][n][0], dp[1][n][1]);
		printf("%d\n", ans);
	}
}