深度搜索和宽度搜索对立,宽度搜索是横向搜索(队列实现),而深度搜索是纵向搜索(递归实现);
看下面这个例子:
现在需要驾车穿越一片沙漠,总的行驶路程为L。小胖的吉普装满油能行驶X距离,同时其后备箱最多能放下四桶油。在起点有N种汽油,每种汽油都有无限桶,一桶能行驶距离Ai。现在小胖想知道:能不能恰好带四桶油,再加上出发前装满的油,使得恰好能行驶L距离。
两个 恰好 使得这道题的难度增加;
如何才能做到恰好? 动态规划 还是 搜索?
先看看搜索行不行
限制条件是:“恰好四桶油”,“恰好行驶距离为L”
那就一个一个的试试,深搜一把试试看
bool flag = false ;
void dfs( int tong , int s ) {
if(tong > 4 || s > L)
return ;
if(tong == 4 && s == L) {
flag = true ;
return ;
}
for(int i = 1 ; i <= n ; i++ ) // n 为汽油种类
for(int j = 1 ; j <= 4 ; j++)
dfs(tong + j , s + a[i] * j ) ;
}
如果能够找到 flag 就为 true ;
判断 flag 即可得出结果;
具体实现代码如下:
#include<iostream>
#include<string.h>
#include<string>
#include<algorithm>
using namespace std ;
int a[1005] ;
int L , X , N ;
int ans ;
void dfs(int a1 , int sum) {
if(a1 > 4 || sum > L - X)
return ;
if(a1 == 4 && sum == L - X) {
ans = 1 ;
return ;
}
for(int i = 1 ; i <= N ; i++)
for(int j = 1 ; j <= 4 ; j++)
dfs(a1+j , sum + a[i] * j ) ;
}
int main() {
int n ;
cin >> n ;
while(n--) {
cin >> L >> X >> N ;
memset(a,0,sizeof(a[0])) ;
for(int i = 1 ; i <= N ; i++)
cin >> a[i] ;
sort(a+1,a+1+n) ;
ans = 0 ;
dfs(0,0) ;
if(ans)
cout << "Yes" << endl ;
else
cout << "No" << endl ;
}
return 0 ;
}
很高兴总算找到答案了,但是不要忘了这个数据可能会很大,如果还用这个方法肯定会超时,如果一个算法速度达不到,这个算法是没有多大意义的。
那我们下面应该想想能不能用对程序进行优化,看到上面是不是有很多地方都是重复计算了,一般可以用多消耗一些内存换取程序运行的时间;
观察发现,当带四桶相同的油,多算了好多,既然是深度搜索,应该一直往深度探索,所以以上程序可修改为:
#include<iostream>
#include<string.h>
#include<string>
#include<algorithm>
using namespace std ;
int a[1005] ;
int L , X , N ;
int ans ;
void dfs(int i ,int a1 , int sum) {
if(a1 > 4 || sum > L - X)
return ;
if(a1 == 4 && sum == L - X) {
ans = 1 ;
return ;
}
for( ; i <= N ; i++)
for(int j = 1 ; j <= 4 ; j++)
dfs(i+1,a1+j , sum + a[i] * j ) ;
}
int main() {
int n ;
cin >> n ;
while(n--) {
cin >> L >> X >> N ;
memset(a,0,sizeof(a[0])) ;
for(int i = 1 ; i <= N ; i++)
cin >> a[i] ;
sort(a+1,a+1+n) ;
ans = 0 ;
dfs(1,0,0) ;
if(ans)
cout << "Yes" << endl ;
else
cout << "No" << endl ;
}
return 0 ;
}
就算如此优化,还是避免不了超时的结果,那还应该如何优化才能达到想要的结果呢,结果发现,还存在可以优化的地方,如果一桶汽油就可以跑完全程,那样的汽油就应该抛弃,通过先前对汽油种类排序,最后留下来的都是一桶跑不完全程的,当全程不能够分成四份时,那么同一种汽油最多取三桶;
通过分析再次对上面程序就行优化得到下面程序:
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int a[1002];
int l,x,n,m;
int flag;
void dfs(int i,int count,int sum)
{
int j;
if(count>4||sum>l||flag)
return;
if(count==4&&sum==l)
{
flag=1;
return ;
}
for(;i<n;i++)
for(j=1;j<=m;j++)
dfs(i+1,count+j,sum+j*a[i]);
}
int main()
{
int t;
int i,j;
scanf("%d",&t);
while(t--)
{
scanf("%d%d%d",&l,&x,&n);
l=l-x;
for(i=0;i<n;i++)
scanf("%d",&a[i]);
sort(a,a+n);
for(i=0;i<n;i++)
if(a[i]>=l) //若第a[i]种汽油,一桶的行驶路程不小于l-x,后面的搜索情况不能满足题意,直接舍弃后面的数据
{
n=i;
break;
}
m=4;
if(l%4!=0) //若l-x不能被4整除,每种汽油最多带3桶,可能满足题意
m=3;
flag=0;
dfs(0,0,0);
if(flag)
printf("Yes\n");
else
printf("No\n");
}
return 0;
}
下面介绍一种用动态规划解答方案:
用 dp[ i ][ j ] 代表 带 i 桶汽油,恰好能行驶 j 公里,如果 dp[ 4 ][ L ] = true 则代表存在 ;
#include<iostream>
#include<string>
#include<string.h>
using namespace std ;
int main() {
int n ;
cin >> n ;
while(n--) {
bool dp[6][1005] ; // dp[i][j] 带 i 桶汽油,恰好能行驶 j 公里
memset(dp,false,sizeof(dp)) ;
int L , X , N ;
cin >> L >> X >> N ;
int a[1005];
for(int ii = 1 ; ii <= N ; ii++)
cin >> a[ii] ;
L = L - X ;
dp[0][0] = true ; // 带 0 桶汽油 , 恰好行驶 0 公里
for(int i = 0 ; i <= L ; i++) // 控制条件有两个时,主要判断哪个在前,哪个在后
for(int j = 0 ; j <= 4 ; j++) {
if(dp[j][i])
for(int k = 1 ; k <= N ; k++) {
if(i + a[k] <= L )
dp[j+1][i+a[k]] = true ; // 为动态变量赋值
}
}
if(dp[4][L])
cout << "Yes" << endl ;
else
cout << "No" << endl ;
}
return 0 ;
}