1. 程式人生 > >hdoj 1518 Square 【dfs】

hdoj 1518 Square 【dfs】

efi string.h pre lan .cn rac article 長度 trac

題意:給出n個(不同長度的)棍子,問能不能將他們構成一個正方形。

策略:深搜。

hdoj 1455的簡化版

代碼:

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define M 25
using namespace std;
int s[M], n, ans;//ans就是答案
bool vis[M];
int dfs(int cou, int cur, int pos){  //cou是已分配的木棍數,cur是當前的長度, pos是當前的序號
	if(cou == n){
		return 1;
	}
	int i;
	for(i = pos; i < n; i ++){
		if(vis[i]) continue;
		if(cur+s[i] < ans){
			vis[i] = 1;
			if(dfs(cou+1, cur+s[i], i+1)) return 1;
			vis[i] = 0;
			if(cur == 0) return 0;
			while(s[i] == s[i+1]&&i+1<n) ++i;
		}
		else if(cur+s[i] == ans){
			vis[i] = 1;
			if(dfs(cou+1, 0, 0)) return 1;
			vis[i] = 0;
			return 0;
		}
	}
	return 0;
}
int main(){
	int t, i;
	scanf("%d", &t);
	while(t --){
		int sum = 0;
		scanf("%d", &n);
		for(i = 0; i < n; i ++){
			scanf("%d", &s[i]);
			sum+=s[i];
		}
		if(sum%4){  //假設能,肯定能整除4
			printf("no\n");
			continue;
		}
		ans = sum/4;
		memset(vis, 0, sizeof(vis));
		int flag = dfs(0, 0, 0);
		printf("%s\n", flag?

"yes":"no"); } return 0; }


題目鏈接:http://acm.hdu.edu.cn/showproblem.php?pid=1058


hdoj 1518 Square 【dfs】