正方形

蒜头君手上有一些小木棍,它们长短不一,蒜头君想用这些木棍拼出一个正方形,并且每根木棍都要用到。 例如,蒜头君手上有长度为 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";
}






  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值