题目描述
某一村庄在一条路线上安装了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;
}