[经典DFS]poj1011Sticks

来源:互联网 发布:geohot编程能力 编辑:程序博客网 时间:2024/06/03 19:34

这题是我遇到的最卡剪枝的题了.很显然的dfs,代码也可以很短,很精简,基本没有甚么好说的.然而它的剪枝如此面面俱到,

题目数据是64,解答树甚为巨大,试想,2^64!!!要想在10MB,1s内筛选出正解!! 所以说,这题看似体面,代码都非常简单明了,但是

却时刻提醒你: 永远不要忘记一旦使用了作为对树的盲目遍历---DFS算法,一定不要忘记剪枝的重要性!

#include <cstdio>#include <cstring>#include <algorithm>using namespace std;const int MAX = 128;int a[MAX], ans, sum, n, tot;bool vis[MAX];bool cmp(int a, int b) {return a > b;}/* 从第p个位置(0~n-1)开始搜,当前长度len,当前根数cnt */bool dfs(int p, int len, int cnt) {    if (cnt == tot) return true;//拼完了    for (int i = p; i < n; ++i) {        if (vis[i] || a[i] + len > ans) continue;        if (a[i] + len == ans) {            vis[i] = true;            if (dfs(cnt+1, 0, cnt+1)) return true; //拼好这一根,下一根从cnt+1开始拼.为什么?因为a[]是降序的,首先明确前cnt根(也就是最长的cnt                                                   //根)必须已经拼完.因为如果它们中只要有一根没有拼完,由于之前已经从它开始,尝试过和后面所有木                                                   //棍的组合了,而且该尝试结果为false(否则就不会搜到这一步,而是直接往上返回),那它一定没法和                                                   //剩下的组合了.所以从cnt+1开始.这也算个小剪枝,但是不重要.            return vis[i] = false; //如果拼好这根,剩下的没法拼了,说明当前只要存在当前这个长度len是一定无法拼完的.剪枝(1)        } else if (a[i] + len < ans) {            vis[i] = true;            if (dfs(i+1, len+a[i], cnt)) return true;            vis[i] = false;            if (!len) return false;//如果当前是从长度为0开始拼,都没法拼出来,也说明无法拼完.剪枝(2)        }        while (i < n-1 && a[i] == a[i+1]) ++i; //上面没有返回(说明加a[i]上去一定拼不完),那么和a[i]相同长度的也一定品不完.剪枝(3)    }    return false;}int main() {while (~scanf(" %d", &n) && n) {sum = 0, ans = -1;        for (int i = 0; i < n; ++i) {            scanf(" %d", a+i);            sum += a[i];            if (a[i] > ans) ans = a[i];        }        sort(a, a+n, cmp);        for (; ans <= sum; ++ans) {if (sum % ans) continue;tot = sum / ans;memset(vis, false, sizeof(vis));            if (dfs(0, 0, 0)) break;        }        printf("%d\n", ans);}return 0;}


0 0
原创粉丝点击