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

相似题目

1、poj1011：该题的增强版。

参考

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