洛谷P1220 关路灯（区间dp）

题目描述

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

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

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

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

输入格式

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

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

输出格式

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

输入输出样例

输入 #1复制

5 3
2 10
3 20
5 20
6 30
8 10

输出 #1复制

270  

说明/提示

输出解释：

{此时关灯顺序为3 4 2 1 5，不必输出这个关灯顺序}

 题意：一共有n盏路灯和起点的编号c（即第c个路灯）；接下来n行，第i行给出编号为i的路灯在数轴上的位置，路灯位置单调递增；以为老张的速度为1m/s，所以每个路灯在此过程中的功耗就是其功率*等待时间（即：老张在关掉它之前所走的路程）求：按一定顺序关掉所有路灯，在所有方案中，关掉所有路灯的过程中的功耗总和最小的方案的功耗值，样例解释如下：

30*1+20*4+10*5+10*11=270

思路：区间dp，这里所谓区间dp，就是让每个子区间达到最优，进而推出所求区间最优解。根据这个题的题意，已知最初的起点，我们可以理解为从后往前推，每次取最优，因为每次都是选择继续向前走关掉i+1还是从j返回关掉i，所以每次取这两种方案的最小值，因为每个点都要经过，当经过第一个和最后一个时，其中间的所有点肯定都经过了（就顺便关了），所以最终状态一定是在第一个或最后一个，所以最后取这两者的最小值即可，具体地说，就是：

用dp[i][j][0]表示i到j这个区间功耗的最小值（走完后老张在此区间左端点）

用dp[i][j][1]表示i到j这个区间功耗的最小值（走完后老张在此区间右端点）

状态转移方程：

dp[i][j][0]=min(dp[i+1][j][0]+(a[i+1]-a[i])*(sum[n]-(sum[j]-sum[i])),dp[i+1][j][1]+(a[j]-a[i])*(sum[n]-(sum[j]-sum[i])));//i到j这个区间当走完后老张的状态在左端点的功耗的最小值即为min（其右子区间[i+1,j]的终点在左端点的最优解（当前在右子区间的左端点）+从其右子区间的左端点(a[i+1])向左到当前区间左端点(a[i])的距离(a[i+1]-a[i]) * 在此走动的过程中还亮着的灯的总的功率（sum[n]-(sum[j]-sum[i])）这两者的乘积即为该过程的总的功耗)，其右子区间[i+1,j]的终点在右端点的最优解（当前在右子区间的右端点）+从其右子区间的右端点(a[j])向左到当前区间左端点(a[i])的距离（a[j]-a[i]）* 在此走动的过程中还亮着的灯的总的功率（sum[n]-(sum[j]-sum[i])））即：从上一个的左向左到当前左/从上一个的右向左到当前左

dp[i][j][1]=min(dp[i][j-1][0]+(a[j]-a[i])*(sum[n]-(sum[j-1]-sum[i-1])),dp[i][j-1][1]+(a[j]-a[j-1])*(sum[n]-(sum[j-1]-sum[i-1])));//i到j这个区间当走完后老张的状态在右端点的功耗的最小值即为min（其左子区间[i,j-1]的终点在左端点的最优解（当前在左子区间的左端点）+从其左子区间的左端点(a[i])向右到当前区间右端点(a[j])的距离(a[j]-a[i]) * 在此走动的过程中还亮着的灯的总的功率（sum[n]-(sum[j-1]-sum[i-1])）这两者的乘积即为该过程的总的功耗)，其左子区间[i,j-1]的终点在右端点的最优解（当前在左子区间的右端点）+从其左子区间的右端点(a[j-1])向右到当前区间右端点(a[j])的距离(a[j]-a[j-1]) * 在此走动的过程中还亮着的灯的总的功率（sum[n]-(sum[j-1]-sum[i-1]) )）即：从上一个的左向右到当前右/从上一个的右向右到当前右

完整代码：

#include <bits/stdc++.h>
#define int long long
using namespace std;
int n,c,a[55],p[55],sum[55],dp[55][55][2];
signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin>>n>>c;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i]>>p[i];
        sum[i]=sum[i-1]+p[i];//sum[i]记录i这个点（包括i）的功率的点缀和
    }
    memset(dp,127,sizeof(dp));//初始值设为正无穷，因为只能选择从给定的点开始走的路线，所以设正无穷就可以把别的路线就在取min时舍去（memset设127就相当于正无穷）
    dp[c][c][0]=dp[c][c][1]=0;//一定是先关起点处的，所以设为0
    for(int k=2;k<=n;k++)//枚举所有区间长度（区间长度为j-i+1）
    {
        for(int i=1;i+k-1<=n;i++)//枚举所有区间左端点i
        {
            int j=i+k-1;//当前区间右端点j
            //接下来的状态转移方程参考思路中的详解
            dp[i][j][0]=min(dp[i+1][j][0]+(a[i+1]-a[i])*(sum[n]-(sum[j]-sum[i])),dp[i+1][j][1]+(a[j]-a[i])*(sum[n]-(sum[j]-sum[i])));
            dp[i][j][1]=min(dp[i][j-1][0]+(a[j]-a[i])*(sum[n]-(sum[j-1]-sum[i-1])),dp[i][j-1][1]+(a[j]-a[j-1])*(sum[n]-(sum[j-1]-sum[i-1])));
        }
    }
    int ans=min(dp[1][n][0],dp[1][n][1]);//取[1,n]这个区间的最优解（位于左端点的最优和右端点的最优这两种方案的最小值）
    cout<<ans<<endl;
    return 0;
}