正方形
蒜头君手上有一些小木棍,它们长短不一,蒜头君想用这些木棍拼出一个正方形,并且每根木棍都要用到。 例如,蒜头君手上有长度为 1,2,3,3, 3的 55 根木棍,他可以让长度为11,22 的木棍组成一条边,另外三根分别组成 33 条边,拼成一个边长为 33 的正方形。蒜头君希望你提前告诉他能不能拼出来,免得白费功夫。
输入格式
首先输入一个整数 n(4≤n≤20)n(4≤n≤20),表示木棍数量,接下来输入 nn 根木棍的长度 pi(1≤pi≤10000)pi(1≤pi≤10000)。
输出格式
如果蒜头君能拼出正方形,输出"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";
}