Luogu P1220 关路灯 （区间dp）

From：JZOJ,Luogu

Problem

某一村庄在一条路线上安装了 $n$ 盏路灯，每盏灯的功率有大有小（即同一段时间内消耗的电量有多有少）。老张就住在这条路中间某一路灯旁，他有一项工作就是每天早上天亮时一盏一盏地关掉这些路灯。

为了给村里节省电费，老张记录下了每盏路灯的位置和功率，他每次关灯时也都是尽快地去关，但是老张不知道怎样去关灯才能够最节省电。他每天都是在天亮时首先关掉自己所处位置的路灯，然后可以向左也可以向右去关灯。开始他以为先算一下左边路灯的总功率再算一下右边路灯的总功率，然后选择先关掉功率大的一边，再回过头来关掉另一边的路灯，而事实并非如此，因为在关的过程中适当地调头有可能会更省一些。

现在已知老张走的速度为 $1m/s$，每个路灯的位置（是一个整数，即距路线起点的距离，单位：$m$）、功率（ $W$ ），老张关灯所用的时间很短而可以忽略不计。

请你为老张编一程序来安排关灯的顺序，使从老张开始关灯时刻算起所有灯消耗电最少（灯关掉后便不再消耗电了）。

Input

文件第一行是两个数字 $n(1<=n<=50$，表示路灯的总数$)$和$c(1<＝c<=n$老张所处位置的路灯号$)$；

接下来 $n$ 行，每行两个数据，表示第 $1$ 盏到第 $n$ 盏路灯的位置和功率。数据保证路灯位置单调递增。

Output

一个数据，即最少的功耗(单位：$J，1J＝1W\cdot s$)。

Solution

• 经典的区间 dp；
• 首先，我们维护一个前缀和，$s[i]$ 记录前 $i$ 盏灯的功率和。然后为了方便，用 $s[i]$ 去处理出一个 $p[i][j]$ 表示当把 $i$~$j$ 的灯关掉后其他灯在单位时间内的能耗；
• 区间 dp 紧随其后，二重循环分别枚举区间长度与左端点，右端点通过计算得出，先来设置状态：$f[i][j][0]$ 表示把 $i$~$j$ 的灯关掉，人站在最左边需要的最少能耗，自然，$f[i][j][1]$ 就表示把 $i$~$j$ 的灯关掉，人站在最右边需要的最少能耗。
• 然后就是状态转移方程了，锵锵锵！

	f[i][j][0]=min(f[i][j][0],f[i+1][j][0]+(l[i+1]-l[i])*p[i+1][j]);  //从i+1走到i加上走这段距离时其他灯需要的能耗是否更优
	f[i][j][0]=min(f[i][j][0],f[i+1][j][1]+(l[j]-l[i])*p[i+1][j]);    //从j走到i加上走这段距离时其他灯需要的能耗是否更优
	f[i][j][1]=min(f[i][j][1],f[i][j-1][1]+(d[j]-d[j-1])*p[i][j-1]);  //从j-1走到j加上走这段距离时其他灯需要的能耗是否更优
	f[i][j][1]=min(f[i][j][1],f[i][j-1][0]+(d[j]-d[i])*p[i][j-1]);   //从i走到j加上走这段距离时其他灯需要的能耗是否更优

Code

#include <bits/stdc++.h>
using namespace std;
int n,m;
int d[1005],w[1005];
int f[1005][1005][2];
int p[1005][1005];
int main()
{
	scanf("%d %d",&n,&m);
	for (int i=1;i<=n;i++)
	{
		scanf("%d %d",&d[i],&w[i]);
		w[i]+=w[i-1];    //前缀和
	}
	for (int i=1;i<=n;i++)
	 for (int j=i;j<=n;j++)
	  p[i][j]=w[n]-(w[j]-w[i-1]);  //预处理
	memset(f,10,sizeof(f));
	f[m][m][0]=0;f[m][m][1]=0;
	for (int len=2;len<=n;len++)
	 for (int i=1;i<=n-len+1;i++)
	 {
	 	int j=i+len-1;
	 	f[i][j][0]=min(f[i][j][0],f[i+1][j][0]+(d[i+1]-d[i])*p[i+1][j]);
	 	f[i][j][0]=min(f[i][j][0],f[i+1][j][1]+(d[j]-d[i])*p[i+1][j]);
	 	f[i][j][1]=min(f[i][j][1],f[i][j-1][1]+(d[j]-d[j-1])*p[i][j-1]);
	 	f[i][j][1]=min(f[i][j][1],f[i][j-1][0]+(d[j]-d[i])*p[i][j-1]);
	 }
	 cout << min(f[1][n][0],f[1][n][1]);  //关完n盏灯后，人可以在最左边也可以在最右边，取最小功耗
	 return 0;
}