【深度优先搜索】洛谷:母亲的牛奶

深度优先搜索:

dfs(a,b,c):当前ABC桶牛奶的体积

if(a==0)记录c的值

以a->b为例:当a不空且b不满时

if(b+a>B)

dfs(a-(B-b),B,c)

if(a+b<=B)dfs(0,b+a,c);

注意去重:

如果C->A后,A->C后,C->A,......,会一直递归,然后stackoverflow

可以用状态来唯一表示一次选择,即开一个三维数组,vis[a][b][c],每次做 选择之前做标记

#include<iostream>
using namespace std;
int res[21];
int A,B,C;
bool vis[21][21][21]; 
void dfs(int a,int b,int c){
	if(vis[a][b][c]==true)return;
	if(a==0&&res[c]==0){
		res[c]=1;
		return;
	}
	vis[a][b][c]=true;
	if(c!=0&&b<B){
	    if(b+c>B){
			dfs(a,B,c-(B-b));
		}
		else{
			dfs(a,b+c,0);
		}
	}
	if(c!=0&&a<A){
		if(a+c>A){
			dfs(A,b,c-(A-a));
		}
		else{
			dfs(a+c,b,0);
		}
	}
	if(a!=0&&b<B){
		if(a+b>B){
			dfs(a-(B-b),B,c);
		}
		else{
			dfs(0,b+a,c);
		}
	}

	if(a!=0&&c<C){
		if(a+c>C){
			dfs(a-(C-c),b,C);
		}
		else{
			dfs(0,b,a+c);
		}
	}

	if(b!=0&&c<C){
		if(b+c>C){
			dfs(a,b-(C-c),C);
		}
		else{
			dfs(a,0,b+c);
		}
	}
	if(b!=0&&a<A){
		if(a+b>A){
			dfs(A,a+b-A,c);
		}
		else{
			dfs(a+b,0,c);
		}
	}
}

int main(){
	cin>>A>>B>>C;
	res[C]=1; 
	dfs(0,0,C);
	for(int i=0;i<=20;i++){
		if(res[i]==1){
			cout<<i<<" ";
		}
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/m0_52043808/article/details/123929998
今日推荐