計蒜客 等邊三角形 dfs

注意:
如果某條邊的長度過大,停止對這個分支的搜索,這一個過程稱爲剪枝。

#include<iostream>
#include<stdio.h>
#include<string.h>
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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章