正方形

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

输入格式

首先输入一个整数 n(4≤n≤20)n(4n20),表示木棍数量,接下来输入 nn 根木棍的长度 pi(1≤pi≤10000)pi(1pi10000)

输出格式

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

样例输入1

4
1 1 1 1

样例输出1

Yes

样例输入2

5
10 20 30 40 50

样例输出2

No


解题思路:

抽象的DFS剪枝+优化。

先找出一条边,再在这条边的基础上找另外的一条边,再在两条边的基础上找第三条边。

DFS就不多说了,主要说一下剪枝与优化。

剪枝:

1.当前小木棍的和>规定的边长后,直接返回

2.如果该木棍已经被选过,则跳过。

优化:

1.当这些小木棍的数值之和不是4的整数倍时,直接no

2.仅需找到三条边即可,第四条边无需找。

3.(最重要)举例说明:

1,2,3,4,5个小木棍。

如果你选了 1 和 3 ,那么下次找边就从4开始。为什么不从2开始?

为什么能选到3?在选3之前已经选过了2,已经判断过集合{1,2....}无法成为一条边,从2开始就重复了之前的动作。

即,从4开始。


#include<iostream>
#include<cstdlib>
using namespace std;
int v[21];//保存小木棍的数值
int state[21];//保存小木棍是否被取走
int n;
int length;//正方形的边长
//tmp:记录上次所选的木棍下标,sum:当前边的总长度,edge:查询当前的边
void dfs(int tmp,int sum,int edge){
    if(sum>length)   return;//如果当前边大于正方形边长,剪枝
    if(sum==length&&edge<3) dfs(1,0,edge+1);//发现一个可行边,在此基础上找其他边
    if(sum==length&&edge==3){//如果已经找到==length的三个边,第四个边可忽略
        cout<<"Yes";
        exit(0);
    }
    
    for(int i=tmp;i<=n;++i){//一个优化
        if(state[i]==1)     continue;
        state[i]=1;//将选择的木棍状态置1
        dfs(i+1,sum+v[i],edge);
        state[i]=0;//回溯
    }
}
int main(){
    cin>>n;
    int sum=0;
    for(int i=1;i<=n;++i){cin>>v[i];sum+=v[i];}
    
    if(sum%4!=0){cout<<"No";return 0;}
    length=sum/4;
    dfs(1,0,1);
    cout<<"No";
}






阅读更多
文章标签: dfs 剪枝优化
上一篇报数
下一篇蒜头君运送宝藏
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭
关闭