题目描述
某一村庄在一条路线上安装了 nn 盏路灯,每盏灯的功率有大有小(即同一段时间内消耗的电量有多有少)。老张就住在这条路中间某一路灯旁,他有一项工作就是每天早上天亮时一盏一盏地关掉这些路灯。
为了给村里节省电费,老张记录下了每盏路灯的位置和功率,他每次关灯时也都是尽快地去关,但是老张不知道怎样去关灯才能够最节省电。他每天都是在天亮时首先关掉自己所处位置的路灯,然后可以向左也可以向右去关灯。开始他以为先算一下左边路灯的总功率再算一下右边路灯的总功率,然后选择先关掉功率大的一边,再回过头来关掉另一边的路灯,而事实并非如此,因为在关的过程中适当地调头有可能会更省一些。
现在已知老张走的速度为 1m/s1m/s,每个路灯的位置(是一个整数,即距路线起点的距离,单位:mm)、功率(WW),老张关灯所用的时间很短而可以忽略不计。
请你为老张编一程序来安排关灯的顺序,使从老张开始关灯时刻算起所有灯消耗电最少(灯关掉后便不再消耗电了)。
输入格式
第一行是两个数字 nn(表示路灯的总数)和 cc(老张所处位置的路灯号);
接下来 nn 行,每行两个数据,表示第 11 盏到第 nn 盏路灯的位置和功率。数据保证路灯位置单调递增。
输出格式
一个数据,即最少的功耗(单位:JJ,1J=1W\times s1J=1W×s)。
输入输出样例
输入 #1复制
5 3 2 10 3 20 5 20 6 30 8 10
输出 #1复制
270
说明/提示
样例解释
此时关灯顺序为 3 4 2 1 5
。
数据范围
1\le n\le501≤n≤50,1\le c\le n1≤c≤n。
方法一:深搜 +剪枝
#include<bits/stdc++.h>
using namespace std;
int a[55],s[55],n,w[55],x,ans=0x3f3f3f3f,m,b[55],h,ss,sum,vis[55];
void dfs(int k,int dep, int ss,int sum){//k 当前位置, dep搜索的位置, ss计算使用,sum:当前用的时间
if(k<1 || k>n || sum>ans) return;//假如当前的时间已经比ans还大,停止搜索
if(dep>n){
ans=min(ans,sum);
return ;
}
b[dep-1]=k;//b记录路径坐标
for(int i=k-1; i>=1; i--){//往左边找一个
if(vis[i]==0){
vis[i]=1;
b[dep]=i;
dfs(i,dep+1,ss-w[b[dep-1]],sum+(ss-w[b[dep-1]])*abs(a[b[dep]]-a[b[dep-1]]));
vis[i]=0;
break;
}
}
for(int i=k+1; i<=n; i++){//往右边找一个
if(vis[i]==0){
vis[i]=1;
b[dep]=i;
dfs(i,dep+1,ss-w[b[dep-1]],sum+(ss-w[b[dep-1]])*abs(a[b[dep]]-a[b[dep-1]]));
vis[i]=0;
break;
}
}
}
int main()
{
cin>>n>>x;
for(int i=1; i<=n; i++){
cin>>a[i]>>w[i];
h+=w[i];
}
vis[x]=1;
dfs(x,2,h,0);
cout<<ans<<endl;
return 0;
}
//搜索方法
方法二:区间dp
状态:
刚开始想:dp[i][j]表示关掉区间[i,j]所花费的最小功耗。
但是每一个状态有两种情况:
1、此时站在左端点(i)
2、此时站在右端点(j)
所以我们再加上一维信息,dp[i][j][0]关掉区间[i,j],此时站在i点所花费的最小功耗,dp[i][j][1]关掉区间[i,j],此时站在j点所花费的最小功耗.
dp[i][j][0]=min(dp[i+1][j]+ (a[i+1]-a[i])*sum(i+1,j),dp[i+1][j][1]+(a[j]-a[i])*sum(i+1,j) ); 其中sum(i,j)表示i到j的和,使用前缀和。
dp[i][j][1]=min(dp[i][j-1][0]+(a[j]-a[i]))*sum(i,j-1), dp[i][j-1][1]+(a[j]-a[j-1])*sum(i,j-1)).
边界值: dp[x][x][0]=dp[x][x][1]=0; x为老张最初站的位置。
答案:min(dp[1][n][0],dp[1][n][1]);
#include<bits/stdc++.h>
using namespace std;
int a[55],n,w[55],sum[55],dp[55][55][2],x;
int f(int x, int y,int l, int r){
return (a[x]-a[y])*(sum[n]-sum[r]+sum[l-1]);
}
int main()
{
cin>>n>>x;
for(int i=1; i<=n; i++){
cin>>a[i]>>w[i];
sum[i]=sum[i-1]+w[i];
}
memset(dp,0x3f,sizeof(dp));
dp[x][x][1]=dp[x][x][0]=0;//区间长度为1,边界值
for(int l=2; l<=n; l++){//区间长度为2
for(int i=1; i+l-1<=n; i++){//区间左端点
int j=i+l-1;//区间右端
dp[i][j][0]=min(dp[i+1][j][0]+f(i+1,i,i+1,j),dp[i+1][j][1]+f(j,i,i+1,j));
dp[i][j][1]=min(dp[i][j-1][0]+f(j,i,i,j-1),dp[i][j-1][1]+f(j,j-1,i,j-1));
}
}
cout<<min(dp[1][n][0],dp[1][n][1])<<endl;
return 0;
}
//搜索方法
总结:区间dp,根据题目需要,可以加一维数组记录相关信息。