计蒜客 等边三角形 dfs

注意:
如果某条边的长度过大,停止对这个分支的搜索,这一个过程称为剪枝。

#include
#include
#include
using namespace std;
int n;
int sum = 0; 
bool flag = false;
int len[30];
int barrel[4];
void dfs(int index){
	if(flag==true){
		return;
	}
	if(index==n+1){
		if(barrel[1]==barrel[2]&&barrel[2]==barrel[3]){
			flag = true;
		}
		return;
	}
	for(int i =1;i<=3;i++){
		if(barrel[i]+len[index]<=sum/3+1){
			barrel[i] += len[index];
			dfs(index+1);
			barrel[i] -= len[index];	
		}
	}
}
int main(){
	//freopen("in.txt","r",stdin);
	memset(barrel,0,sizeof(barrel));
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>len[i];
		sum +=len[i];
	}
	dfs(1);
	if(flag){
		cout<<"yes\n";
	}
	else{
		cout<<"no\n";
	}
	return 0;
}

你可能感兴趣的:(搜索问题,计蒜客)