poj2362
题意:给出一些木棒,问你是否能将这些木棒组成一个正方形。
思路:
1.当木棒总长度不被4整除则输出no。
2.可以这样理解,将所有木棒分成4组等长的,和为sum/4的木棒,搜索的到则成立。
3.如果失败则回溯到上一个状态,若一直回溯到第一个状态,则找不到。
我用了一个visit[]数组来保存木棒是否已经被取用,
#include <iostream> #include <cstdio> #include <algorithm> #include <string> #include <cmath> #include <cstring> #include <queue> #include <set> #include <vector> #include <stack> #include <map> #include <iomanip> #define PI acos(-1.0) #define Max 2005 #define inf 1<<28 using namespace std; int sticks[Max],sum,n; bool visit[Max]; int length; int dfs(int duan, int start, int left)//left代表剩余的length长度 { int i; if(left == 0)// 如果left为0,证明正好组成一条变长为length的木棒 { //cout<<duan<<endl<<endl; //show(); if(duan == 3) return 1;//如果前面已经搜索到3段了,则证明这是第四段,则返回成功。 for(i=0; i<n; i++)//找出未被取到的木棒的最小的序数,从最小的序数继续DFS if(!visit[i]) break; //cout<<i<<":"<<endl; visit[i] = 1;//标记为找过 if(dfs(duan + 1, i + 1, length - sticks[i]))//如果成功,则返回成功 return 1; visit[i] = 0;//否则不取这一段 return 0;//回溯 } //cout<<start<<endl; if(start >= n) return 0; for(int j = start; j < n; j++)//从start开始搜索 { if(visit[j] || sticks[j] > left) continue; //cout<<j<<endl; visit[j] = 1; if(dfs(duan, j +1, left - sticks[j])) return 1; visit[j] = 0; } return 0; } int main() { int i,j,k,l,m,T; cin>>T; while(T--) { cin>>n; sum=0; memset(visit,0,sizeof(visit)); for(i=0; i<n; i++) { scanf("%d",&sticks[i]); sum+=sticks[i]; } if(sum%4!=0) { cout<<"no"<<endl; continue; } length=sum/4; int flag=dfs(0,0,length);//从第0段,序数0,length剩余长度开始搜索 if(flag) cout<<"yes"<<endl; else cout<<"no"<<endl; } return 0; }
poj1011
其实原理和上面那题一样,就是2362那题木棒的数量已经给定为4,而这题木棒的数量没有限制,只要找出组合相同的最小长度即可。
当然这一题很重要的就是一些特殊情况的舍弃,我就没考虑一些情况TLE了好几次。
#include <iostream> #include <cstdio> #include <algorithm> #include <string> #include <cmath> #include <cstring> #include <queue> #include <set> #include <vector> #include <stack> #include <map> #include <iomanip> #define PI acos(-1.0) #define Max 2005 #define inf 1<<28 using namespace std; int sticks[Max]; bool visit[Max]; int n,m,num; bool cmp(int x,int y)//注意排序 ,从大到小排,自己理解一下,递归的次数,从小到大排就TLE { return x>y; } int dfs(int no,int num1,int zong)//搜索和上面那题差不多,就是return 1 的条件不同而已 { int i,j; if(num1==0) { if(zong==0)return 1; for(i=0; i<n; i++) if(!visit[i]) break; visit[i]=1; if(dfs(i+1,num-sticks[i],zong-sticks[i])) return 1; visit[i]=0; return 0; } if(no>=n)return 0; for(i=no; i<n; i++) { if(visit[i]||sticks[i]>num1) continue; if(i > 0 && !visit[i - 1] && sticks[i] == sticks[i - 1])//如果该木棒和前面木棒的长度相同,则不选 continue; visit[i]=1; if(dfs(i+1,num1-sticks[i],zong-sticks[i])) return 1; visit[i]=0; } return 0; } int main() { int i,j,k,l; while(scanf("%d",&n),n) { int sum=0,MAX=-inf; for(i=0; i<n; i++) { visit[i]=0; scanf("%d",&sticks[i]); sum+=sticks[i]; if(sticks[i]>MAX) MAX=sticks[i]; } sort(sticks,sticks+n,cmp); for(i=MAX; i<=sum; i++)//枚举原始木棒的长度,从单一木棒的最大值枚举到所有木棒的总长度 { if(sum%i!=0)//剪枝,如果枚举的木棒长度不被总长度整除,则肯定不是原始木棒的长度 continue; num=i; if(dfs(0,i,sum)) { printf("%d\n",i); break; } } } return 0; }这两题做下来,相对来说1011要更多的考虑剪枝的情况,不然很容易TLE。