zoj 1909 || poj 2362 Square

 

啊。。搜索。。。

 

这题比poj1011算是简化版的吧。

 

直接按长度从大到小排序。如果长度之和%4不等于0直接输出no。

 

然后DFS，如果满足题意，尽快退出DFS。

 

如果不满足题意，也尽快退出，比如，如果从第一个数找不到满足题意的，就可以退出了，因为每个数都必须用到的，这一点，我开始没有优化，实现的办法木有想清楚，270+ms。。。就刚才。。想通了点，加了一句话。。。10ms了。。。激动下~就是，如果找的点遇见第一个点被赋值为0了，那说明第一个点回溯了，这是不应该出现的，如果第一个点就回溯，那么肯定从第一个点就找不到符合的点，肯定不符合题意了。

 

呃。。这个题解写得很纠结。。。因为写着写着想起来剪枝方法了。。。就去改代码。。。

 

又改了一个版本。。。zoj 0ms了^ ^。。。poj 16ms。。

 

这次优化的是回溯的时候flag还没等于1，直接return了，因为深搜到最后一步，如果还没得到要的答案，那肯定是找不到正确答案了，直接退出。

 

啊。又加了句话，poj也0ms啦~就是，比如5+4已经超界了，那如果有多个4并列，跳过这些4即可。

 

这些好多都是1011的优化方法，可是我1011还是过不了。。。呜呜呜。。。。

 

#include <cstdio> #include <cstdlib> #include <iostream> #include <string.h> #include <queue> #include <limits.h> #include <algorithm> using namespace std; int cmp(int a,int b) { return a > b; } int len[30],n,vis[30],l,num,flag; int check() { int i; for(i=0; i<n; i++) if( !vis[i] ) return 0; return 1; } void DFS(int x,int sum,int cur) { int i; if( flag != 0 ) return ; if( sum == l ) { num++; if( num == 4 && check() ) { flag = 1; return ; } while( vis[cur] ) // 纳闷，如果去掉这个就0ms。。加上就16ms。。可是，这个是优化啊！ cur++; DFS(cur,0,cur); if( flag != 1 ) { flag = 0; return ; } return ; } for(i=x; i<n; i++) { if( !vis[i] && sum + len[i] <= l ) { vis[i] = 1; DFS(i+1,sum+len[i],cur); if( flag ) return ; if( sum + len[i] == l ) num--; vis[i] = 0; while( len[i] == len[i+1] ) // 如果相等就不用再搜了 i++; } if( !vis[cur] ) // cur里存的是，这几个数开始搜的位置。 { flag = -1; return ; } } } int main() { int ncases; int i,sum; scanf("%d",&ncases); while( ncases-- ) { scanf("%d",&n); sum = 0; for(i=0; i<n; i++) { scanf("%d",&len[i]); sum += len[i]; } if( sum % 4 != 0 ) { printf("no/n"); continue; } sort(len,len+n,cmp); l = sum/4; memset(vis,0,sizeof(vis)); flag = num = 0; DFS(0,0,0); if( flag == 1 ) printf("yes/n"); else printf("no/n"); } return 0; }