poj2362 Square（DFS）

题目链接

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

题意

输入n根棍子的长度，求这n根棍子是否能组成一个正方形。

思路

假设能组成正方形，则正方形的周长为sum，sum/4为正方形的边长，问题转化为这n根棍子能否组成4根长度为side的棍子。由于棍子的长度越长，组合的灵活性就越差，所以将n根棍子按从长到短排序后dfs。

代码

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

const int N = 30;
vector<int> stick;
int visit[N];
int n;
int side;

bool cmp(int a, int b)
{
    return a>b;
}

bool dfs(int num, int len, int cur)
{
    if(num==4)
        return true;

    for(int i=cur; i<n; i++)
    {
        if(visit[i])
            continue;
        visit[i] = 1;
        if(len+stick[i]==side)
        {
            if(dfs(num+1, 0, 0))
                return true;
        }
        else if(len+stick[i]<side)
        {
            if(dfs(num, len+stick[i], i+1))
                return true;
        }
        visit[i] = 0;
    }
    return false;
}

int main()
{
    //freopen("poj2362.txt", "r", stdin);
    int t;
    cin>>t;
    while(t--)
    {
        int sum = 0;
        stick.clear();
        cin>>n;
        for(int i=0; i<n; i++)
        {
            int len;
            cin>>len;
            sum += len;
            stick.push_back(len);
        }
        side = sum / 4;
        sort(stick.begin(), stick.end(), cmp);
        if(sum%4!=0 || side<stick[0])  
        {
            cout<<"no"<<endl;
            continue;
        }
        memset(visit, 0, sizeof(visit));
        bool ans = dfs(1, 0, 0);
        if(ans)
            cout<<"yes"<<endl;
        else cout<<"no"<<endl;
    }
    return 0;
}

相似题目

1、poj1011：该题的增强版。

参考

1、http://blog.csdn.net/xindoo/article/details/8867919