【POJ】Square(DFS + 剪枝)

来源：互联网发布：linux虚拟机home扩容编辑：程序博客网时间：2024/05/24 04:33

记录一个菜逼的成长。。

题目大意：
给你一些棍子的长度，问这些棍子能否组成正方形。
剪枝：
1.既然是正方形，长度总和肯定能被4整除。
2.最长的棍子的长度一定小于等于正方形边长。
3.满足上述两个条件，只要判断三边是否符合就可。

看到网上一些题解上有对边长的排序，其实不排序也能过。

#include <cstdio>#include <iostream>#include <cstring>#include <string>#include <algorithm>#include <cstdlib>#include <vector>#include <set>#include <map>#include <queue>#include <stack>#include <list>#include <deque>#include <cctype>#include <bitset>#include <cmath>using namespace std;#define ALL(v) (v).begin(),(v).end()#define cl(a) memset(a,0,sizeof(a))#define bp __builtin_popcount#define pb push_back#define fin freopen("D://in.txt","r",stdin)#define fout freopen("D://out.txt","w",stdout)#define lson t<<1,l,mid#define rson t<<1|1,mid+1,r#define seglen (node[t].r-node[t].l+1)typedef long long LL;typedef unsigned long long ULL;typedef pair<int,int> PII;typedef pair<LL,LL> PLL;typedef vector<PII> VPII;const int INF = 0x3f3f3f3f;const int MOD = 1e9 + 7;template <typename T>inline void read(T &x){    T ans=0;    char last=' ',ch=getchar();    while(ch<'0' || ch>'9')last=ch,ch=getchar();    while(ch>='0' && ch<='9')ans=ans*10+ch-'0',ch=getchar();    if(last=='-')ans=-ans;    x = ans;}inline bool DBread(double &num){    char in;double Dec=0.1;    bool IsN=false,IsD=false;    in=getchar();    if(in==EOF) return false;    while(in!='-'&&in!='.'&&(in<'0'||in>'9'))        in=getchar();    if(in=='-'){IsN=true;num=0;}    else if(in=='.'){IsD=true;num=0;}    else num=in-'0';    if(!IsD){        while(in=getchar(),in>='0'&&in<='9'){            num*=10;num+=in-'0';}    }    if(in!='.'){        if(IsN) num=-num;            return true;    }else{        while(in=getchar(),in>='0'&&in<='9'){                num+=Dec*(in-'0');Dec*=0.1;        }    }    if(IsN) num=-num;    return true;}template <typename T>inline void write(T a) {    if(a < 0) { putchar('-'); a = -a; }    if(a >= 10) write(a / 10);    putchar(a % 10 + '0');}/******************head***********************/const int maxn = 20 + 10;int a[maxn],vis[maxn],edge,n;bool dfs(int x,int sum,int cnt)//下标，边长，计数{    if(cnt == 3)return true;//满足三边    for( int i = x; i <= n; i++ ){        if(!vis[i]){            vis[i] = 1;            if(sum + a[i] < edge){                if(dfs(i,sum+a[i],cnt))return true;            }            else if(sum + a[i] == edge){                //**若等于边长，则需要从头开始搜索，才不会漏掉**                if(dfs(1,0,cnt+1))return true;            }            vis[i] = 0;        }    }    return false;}int main(){    int T;read(T);    while(T--){        cl(vis);        int sum = 0,mx = 0;read(n);        for( int i = 1; i <= n; i++ ){            read(a[i]);            mx = max(mx,a[i]);            sum += a[i];        }        edge = sum / 4;        if(sum % 4 || mx > edge){            printf("no\n");            continue;        }        printf("%s\n",dfs(1,0,0) ? "yes":"no");    }    return 0;}

0 0