【LUOGU P1220】关路灯(区间dp)
一开始直接搜索+剪枝,拿了50分,发现原来是搜索搜复杂了..因为搜索是可以过的
注意到题目中有一句''再回过头来关掉另一边的路灯,而事实并非如此,因为在关的过程中适当地调头有可能会更省一些",也就是说老王是在沿着某个方向走着走着突然转向,而不是毫无逻辑的,我之前写的搜索枚举就是毫无逻辑的
for(int i=1;i<=n;i++)
{
if(vis[i]) continue;
vis[i]=true;
dfs(...);
vis[i]=false;
}
而应该策略是:对于每个位置,去寻找左边没关掉的灯或者右边没有关掉的灯。之后dfs ,寻找最优情况
不过话说回来,这道题本质上是一个很好的区间DP
我们发现老王关过的灯是属于一个区间的,而不是跳着跳着的关,因此设dp[i][j]表示老王已关闭区间[i,j]的灯,但是老王既可以在左边也可以在右边,所以再加一维[0/1]表示老王在左还是右
则有方程:
f[i][j][0] = min ( f[i+1][j][0] + ( a[i+1] - a[i] ) * ( sum[i] + sum[n] - sum[j] ) , f[i+1][j][1] + ( a[j]-a[i] ) * ( sum[i]+sum[n]-sum[j]) );
f[i][j][1] = min ( f[i][j-1][0] + ( a[j] - a[i] ) * ( sum[i-1] + sum[n] - sum[j-1] ) , f[i][j-1][1] + ( a[j]-a[j-1] ) * ( sum[i-1] + sum[n] - sum[j-1] ) );
其中sum是前缀和,要注意的是每次乘上的总功耗是包含了当前点i(j)的
#include<bits/stdc++.h>
#define N 55
using namespace std;
int n,c;
int dp[N][N][2],dis[N],w[N],sum[N];
int main()
{
cin>>n>>c;
for(int i=1;i<=n;i++)
{
cin>>dis[i]>>w[i];
sum[i]=sum[i-1]+w[i];
}
memset(dp,63,sizeof(dp));
dp[c][c][0]=0,dp[c][c][1]=0;
for(int len=2;len<=n;len++)
{
for(int st=1;st+len-1<=n;st++)
{
int ed=st+len-1;
dp[st][ed][0]=min(dp[st+1][ed][0]+(sum[n]-sum[ed]+sum[st])*(dis[st+1]-dis[st]),dp[st+1][ed][1]+(sum[n]-sum[ed]+sum[st])*(dis[ed]-dis[st]));
dp[st][ed][1]=min(dp[st][ed-1][1]+(sum[n]-sum[ed-1]+sum[st-1])*(dis[ed]-dis[ed-1]),dp[st][ed-1][0]+(sum[n]-sum[ed-1]+sum[st-1])*(dis[ed]-dis[st]));
}
}
cout<<min(dp[1][n][0],dp[1][n][1])<<endl;
return 0;
}
QQ40523591~欢迎一起学习交流~