HDU 1518 Square

题目链接:Square

解题思路:这个题按照一个一个边来找,找到3个就可以了,排序之后只要最大的变边长大于 sum / 4就无法构成正方形,每次找到一根木棍时,要找他的下一根木棍,避免多次重复寻找,刚开始一直TLE,最后看了个别人的博客才发现可以这样剪枝。(DFS的搜索状态量要作为函数参数)  所以发现DFS的题目时要先分析哪些变量跟着搜索深度在变化,之后根据判定条件和题设要求具体求解。

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define MAX 25

using namespace std;

int v[MAX], stick[MAX], m, flag, sum;

void DFS(int x, int len, int k){
	if(flag){
		return;
	}
	if(k == 3){
		flag = 1;
		return;
	}
	if(len == sum){
		DFS(0, 0, k + 1);
	}
	for(int i = x; i < m; i++){
		if(!v[i]){
			if(stick[i] + len > sum){
				break;
			}
			else{
				v[i] = 1;
				DFS(i + 1, len + stick[i], k);
				v[i] = 0;
			}
		}
	}
}

int main(){
	int i, j, k, t;
	scanf("%d", &t);
	while(t--){
		memset(v, 0, sizeof(v));
		sum = flag = 0;
		scanf("%d", &m);
		for(i = 0; i < m; i++){
			scanf("%d", &stick[i]);
			sum += stick[i];
		}
		if(sum % 4 || stick[m - 1] > sum / 4){
			printf("no\n");
			continue;
		}
		sum /= 4;
		sort(stick, stick + m);
		DFS(0, 0, 0);
		if(flag){
			printf("yes\n");
		}
		else{
			printf("no\n");
		}
	}
	return 0;
}


你可能感兴趣的:(深度优先搜索)