poj 2355 DP 单调队列优化

最新推荐文章于 2020-12-12 01:50:14 发布

sky_zdk

最新推荐文章于 2020-12-12 01:50:14 发布

阅读量322

点赞数

分类专栏： ACM poj 动态规划(DP) 文章标签： poj 优化 dp 动态规划

本文链接：https://blog.csdn.net/sky_zdk/article/details/71629781

版权

ACM poj 同时被 2 个专栏收录

201 篇文章 0 订阅

订阅专栏

动态规划(DP)

104 篇文章 1 订阅

订阅专栏

先贴一个没有优化的代码：

#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;
}