先贴一个没有优化的代码:
#include<cstdio>
#include<cstring>
#define MAX(x,y) ((x)>(y)?(x):(y))
#define MIN(x,y) ((x)>(y)?(y):(x))
#define INF 0x3f3f3f3f;
int L[6],C[6];
int d[10010];
int dp[10011];
int cal(int d1,int d2)
{
for(int i=1;i<=3;i++)
{
if((d2-d1)<=L[i])
return C[i];
}
return -1;
}
int main()
{
int n,a,b;
scanf("%d%d%d%d%d%d",&L[1],&L[2],&L[3],&C[1],&C[2],&C[3]);
scanf("%d%d%d",&n,&a,&b);
if(a>b)
{
int tem=a;
a=b;
b=tem;
}
d[1]=0;
for(int i=2;i<=n;i++)
scanf("%d",&d[i]);
dp[a]=0;
for(int i=a+1;i<=b;i++)
{
dp[i]=INF;
for(int j=i-1;d[i]-d[j]<=L[3]&&j>=a;j--)
{
dp[i]=MIN(dp[i],dp[j]+cal(d[j],d[i]));
}
}
printf("%d\n",dp[b]);
return 0;
}
单调队列优化:
#include<cstdio>
#include<cstring>
#define MAX(x,y) ((x)>(y)?(x):(y))
#define MIN(x,y) ((x)>(y)?(y):(x))
#define INF 0x3f3f3f3f;
using namespace std;
int L[6],C[6];
int d[10010];
int dp[10011];
int cal(int d1,int d2)
{
for(int i=1;i<=3;i++)
{
if((d2-d1)<=L[i])
return C[i];
}
return -1;
}
int Q[4][10000];
int l[4],r[4];
int main()
{
int n,a,b;
scanf("%d%d%d%d%d%d",&L[1],&L[2],&L[3],&C[1],&C[2],&C[3]);
scanf("%d%d%d",&n,&a,&b);
if(a>b)
{
int tem=a;
a=b;
b=tem;
}
d[1]=0;
for(int i=2;i<=n;i++)
scanf("%d",&d[i]);
dp[a]=0;
for(int i=1;i<=3;i++)
{
Q[i][0]=a;
l[i]=0;
r[i]=0;
}
for(int i=a+1;i<=b;i++)
{
dp[i]=INF;
for(int j=1;j<=3;j++)
{
while(l[j]<=r[j]&&d[i]-d[Q[j][l[j]]]>L[j])
l[j]++;
if(l[j]<=r[j])
dp[i]=MIN(dp[i],dp[Q[j][l[j]]]+C[j]);
}
for(int j=1;j<=3;j++)
{
while(l[j]<=r[j]&&dp[Q[j][r[j]]]>=dp[i])
r[j]--;
Q[j][++r[j]]=i;
}
}
printf("%d\n",dp[b]);
return 0;
}