poj 2363 poj 1011 dfs + 剪枝

#include <stdio.h>#include <stdlib.h>#include <string.h>int sticks[20];bool visit[20];int cmp(const void * a,const void *b){return *(int *)b - *(int *)a;}bool dfs(int s,int N,int cur,int left,int num,int lengthside) // num 是已经拼好的 长度{ //s 是搜索木棍的起始位置 N 是木棍的总数 left是要拼凑的长度 cur是当前的长度if (num==3) //找到三根 其他的就不用找了return true;int i;for (i=s;i<N;i++){if (!visit[i]){if (cur + sticks[i] == lengthside) // 刚好拼完一根{visit[i] = true;if (dfs(0,N,0,lengthside,num+1,lengthside))return true;visit[i] = false;}else if (cur + sticks[i] < lengthside){visit[i] = true;if (dfs(i+1,N,cur+sticks[i],left - sticks[i],num,lengthside))return true;visit[i] = false;}}}return false;}int main(){int test,N,i,sum,max,lengthside;scanf("%d",&test);while(test--){scanf("%d",&N);sum = 0;max = 0;for (i=0;i<N;i++){scanf("%d",&sticks[i]);sum += sticks[i];if (max < sticks[i])max = sticks[i];}if ( sum % 4 != 0 || sum / max <4) //基本剪枝 {printf("no\n");continue;}memset(visit,false,sizeof(visit));qsort(sticks,N,sizeof(int),cmp);lengthside = sum / 4;if (dfs(0,N,0,lengthside,0,lengthside))printf("yes\n");elseprintf("no\n");}return 0;}

两题基本一样

先对木棍排序 ，然后按照质指定的长度在数组里搜索。

刚开始，这两题都是按照子集和的方法做的，一直TLE，无论怎么剪枝都不行，后来一想，还是按照指定长度搜索正解效率更高。