poj2362——Square(深搜dfs)

Address：http://poj.org/problem?id=2362

一道深搜的题目，整体不是太难。今天做的很水。。

#include <iostream>
#include <cstdio>
#include<algorithm>
#include <cstring>
using namespace std;

int N,stick[26];
int ave;
bool visit[26];

bool dfs(int t,int sum,int num)
{
    if(num==4)
        return true;
    int i;
    for(i=t;i<=N;i++)
    {
        if(visit[i])
            continue;
        visit[i]=true;
        if(sum+stick[i]<ave)
        {
            if(dfs(i+1,sum+stick[i],num))
                return true;
        }
        if(sum+stick[i]==ave)
        {
            if(dfs(1,0,num+1))
                return true;
        }    
        visit[i]=false;
    }
    return false;
}

int main()
{
    int T,i,sum;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&N);
        sum = 0;
        for(i = 1; i <= N; i++)
        {
            scanf("%d",&stick[i]);
            sum += stick[i];
        }
        ave = sum/4;
        memset(visit,false,sizeof(visit));
        if((ave*4==sum)&&dfs(1,0,0))
            printf("yes\n");
        else
            printf("no\n");
    }
    return 0;
}

其他同学的，代码精简

#include <iostream>
#include <cstdio>
using namespace std;
int a[25],used[25],n,m,sum,flag;
int dfs(int d,int w,int count)
{
	int i;
	if(d==sum)
	{
		d=0;
		w=0;
		count++;
		if(count==4)
			return 1;
	}
	for(i=w;i<m;i++)
	{
		if(!used[i])
		{
			used[i]=1;
			if(d+a[i]<=sum&&dfs(d+a[i],i+1,count))
				return 1;
			used[i]=0;
		}
	}
	return 0;
}
int main()
{
	int i;
	scanf("%d",&n);
	while (n--)
	{
		sum=0;
		scanf("%d",&m);
		for(i=0;i<m;i++)
		{
			scanf("%d",&a[i]);
			used[i]=0;
			sum+=a[i];
		}
		flag=0;
		if(sum%4==0)
		{
			sum/=4;
			flag=dfs(0,0,0);
		}
		if(flag)
			printf("yes\n");
		else
			printf("no\n");
	}
}