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。