63-正方形

问题描述：

蒜头君手上有一些小木棍，它们长短不一，蒜头君想用这些木棍拼出一个正方形，并且每根木棍都要用到。 例如，蒜头君手上有长度为 $1$，$2$，$3$，$3$, $3$ 的 $5$ 根木棍，他可以让长度为$1$，$2$ 的木棍组成一条边，另外三根分别组成 $3$ 条边，拼成一个边长为 $3$ 的正方形。蒜头君希望你提前告诉他能不能拼出来，免得白费功夫。

输入格式

首先输入一个整数 $n(4\le n\le 20)$，表示木棍数量，接下来输入 $n$ 根木棍的长度 pi​(1≤pi​≤10000)。

输出格式

如果蒜头君能拼出正方形，输出"Yes"，否则输出"No"。

样例输入1

4
1 1 1 1

样例输出1

Yes

样例输入2

5
10 20 30 40 50

样例输出2

No

代码解析：

//
//  main.cpp
//  zhengfangxing
//
//  Created by apple on 2018/3/25.
//  Copyright © 2018年 apple. All rights reserved.
//

#include<iostream>
#include<algorithm>
using namespace std;
int l[30];//木棍长度
bool mark[30];//用于标记是否选取
int n;//木棍数量
int sum;//用于计算正方形周长
int one;//记录边长
int flag=0;//用于标记是否有答案
//判断a是否大于b,用于sort排序，降序排列
bool cmp(int a,int b)
{
    return a>b;
}
//dfs：分别传入边长，拼好的木棍的数量，木棍的编号。
void dfs(int onel,int num,int pos)
{
    //出口
    if(flag==1) return ;
    //如果有三条边都符合要求，由周长知，最后一条边必定也符合要求。
    if(num==3){
        flag=1;
        return ;
    }
    //如果onel==0,说明已经凑好一条边了
    //num++
    if(onel == 0){
        num++;
        onel=one;
        pos=0;
    }
    
    for(int i=pos;i<n;i++){
        if(!mark[i]){
            if(onel>=l[i]){
                mark[i]=true;//标记为已访问
                dfs(onel-l[i],num,i);
                mark[i]=false;//未访问
            }
        }
    }
}

int main()
{
    fill(mark,mark+30,false);//标记数组全部初始化为未标记
    cin>>n;//木棍个数
    int sum=0;//求和初始化
    //记录每个木棍的长度，并计算周长
    for(int i=0;i<n;i++){
        cin>>l[i];
        sum+=l[i];
    }
    //记录边长
    one = sum/4;
    //对木棍长度进行排序
    sort(l,l+n,cmp);
    //如果第一根最长木棍长度大于边长或者周长不是4的倍数
    if(l[0]>one||sum%4) cout<<"No"<<endl;
    else{
        dfs(one,0,0);//dfs
        if(flag==1) cout<<"Yes"<<endl;
        else cout<<"No"<<endl;
    }
}