Xwxcy的博客

加油↖(^ω^)↗

沙漠之旅(二维dp)

“小胖要穿越一片沙漠,小胖开着一辆大吉普,小胖的吉普油耗高,吉普能放四桶油。”

这就是人人会唱的沙漠之歌~~体现了小胖拔群的聪明才智。

小胖的问题是这样的:现在需要驾车穿越一片沙漠,总的行驶路程为L。小胖的吉普装满油能行驶X距离,同时其后备箱最多能放下四桶油。在起点有N种汽油,每种汽油都有无限桶,一桶能行驶距离Ai。现在小胖想知道:能不能恰好带四桶油,再加上出发前装满的油,使得恰好能行驶L距离。

Input

第一行一个正整数T(1 <= T <= 50),表示数据的组数。

接下来T组数据,每组数据的第一行是三个整数L(1 <= L <= 1000),X(1 <= X <= L),N(1 <= N <= 1000)。

接下来N行,每行一个正整数Ai(1 <= Ai <= 1000),表示第i种汽油一桶能行驶的距离。

Output

对于每组数据输出一行,若能输出“Yes”,否则输出“No”

Sample Input

1
20 9 2
2
3

Sample Output

Yes

AC代码:

#include<bits/stdc++.h>
using namespace std;
#define INF 0x3f3f3f3f
int dp[1010][5];
int T,L,X,N,a[1010];
int main()
{
	int i,j,k,t;
	while(cin>>T)
	while(T--){
		cin>>L>>X>>N;
		L-=X;
		for(i=0;i<N;i++)
		cin>>a[i];
		memset(dp,-INF,sizeof(dp));
		dp[0][0]=0;
		for(i=0;i<N;i++){
			for(j=a[i];j<=L;j++){
				for(k=1;k<5;k++){
					for(t=1;t<=k&&t*a[i]<=j;t++)
					dp[j][k]=max(dp[j][k],dp[j-a[i]*t][k-t]+t*a[i]);
				}
			}
		}
		if(dp[L][4]==L)
		cout<<"Yes"<<endl;
		else cout<<"No"<<endl;
	}
	return 0;
}



阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Xwxcy/article/details/51538352
个人分类: ACM_动态规划 练习
上一篇湫湫系列故事——消灭兔子(优先队列)
下一篇A strange lift
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭