Square HDU - 1518 POJ - 2362

最新推荐文章于 2021-08-15 13:57:51 发布

原创最新推荐文章于 2021-08-15 13:57:51 发布 · 360 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#square #poj #dfs #hdu

dfs 专栏收录该内容

15 篇文章

订阅专栏

本文介绍了一个使用深度优先搜索（DFS）算法解决如何用不同长度的棒子拼成正方形的问题。通过递归地尝试每种可能的组合，并采用剪枝策略来减少搜索空间，最终判断出所有棒子是否能恰好组成一个正方形。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

dfs

题意：给一堆不同长度的棒，问是否可以组成正方形，所有的棒都要用上

#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
int n,m,sum;
int a[25];
bool book[25],f,ff;
void dfs(int b,int l,int k) {     //l表示在第b条边目前积累的长度，k表示在第b条边目前进行排查到的棒的标号 
	if (b==5) {
		ff=true;
		return;
	}
	if (l==sum) {
		dfs(b+1,0,0);
		if (ff==true) return;
	}
	for (int i=k;i<m;i++) {
		if (book[i]==false && l+a[i]<=sum) {
			book[i]=true;
			dfs(b,a[i]+l,i+1);
			if (ff==true) return;
			book[i]=false;
		}
	}
}
int main()
{
	scanf ("%d",&n);
	while (n--) {
		scanf ("%d",&m);
		sum=0;
		f=false;
		ff=false;
		memset(book,false,sizeof(book));
		for (int i=0;i<m;i++) {
			scanf ("%d",&a[i]);
			sum+=a[i];
		}
		if (sum%4!=0) {     //剪枝 
			printf ("no\n");
			continue;
		}
		sum/=4;
		for (int i=0;i<m;i++) {     //剪枝 
			if (a[i]>sum) {
				f=true;
				break;
			}
		}
		if (f==true) {
			printf ("no\n");
			continue;
		}
		dfs(1,0,0);
		if (ff==true) printf ("yes\n");
		else printf ("no\n");
	}
	return 0;
}