qduoj 分辣条1 （搜索+剪枝）

最新推荐文章于 2017-08-28 18:06:26 发布

原创最新推荐文章于 2017-08-28 18:06:26 发布 · 874 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#dfs #剪枝 #算法

搜索专栏收录该内容

35 篇文章

订阅专栏

本文探讨了一个有趣的辣条分配问题：如何将不同重量的辣条均匀分成两份。通过递归深度优先搜索算法，并结合剪枝策略提高效率，解决了这一问题。

分辣条

发布时间: 2016年6月26日 20:36 最后更新: 2016年6月26日 20:37 时间限制: 1000ms 内存限制: 128M

描述

“你喝的酸奶是我买的，辣条也是我买的，你现在要跟我分手，你把我当什么？”

“因为你每次分辣条的时候都比我多一根！”

可见分好辣条是一件多么重要的事情。。

现在有n（1<=n<=200）根辣条，每根辣条的重量为a1,a2...ai..an(1<=ai<=100)。

那么能不能把这些辣条分为重量相等的两份呢？

输入

输入有多组数据。每组数据第一行为一个整数n（1<=n<=200），n代表辣条的数量。第二行有n个整数，a1,a2...ai..an，其中ai(1<=ai<=100)代表第i根辣条的重量

输出

若能分成重量相等的两份，则输出"YES",否则输出"NO"每个答案占一行

样例输入1 复制

样例输出1

YES
NO

/*
	这题和poj2362基本一模一样吧
*/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn = 205;
int n, a[maxn], avg, maxl, flag;
bool book[maxn];
void dfs(int num, int cur)
{
    if(flag) return ;
    if(num == avg) { flag = 1; return ; }
    for(int i = cur; i < n; i++)
    {
        if(num + a[i] > avg) return ;   //剪枝2，因为有序，到这根时已经过重，更不用说后面的了
        if(!book[i] && num+a[i] <= avg)
        {
            book[i] = 1;
            dfs(num+a[i], i+1);
            book[i] = 0;
            if(flag) return ;   //剪枝3， 一找到就回
        }
    }
}
int main(void)
{
    while(cin >> n)
    {
        memset(book, 0, sizeof(book));
        maxl = avg = flag = 0;
        for(int i = 0; i < n; i++)
        {
            scanf("%d", &a[i]);
            if(a[i] > maxl) maxl = a[i];
            avg += a[i];
        }
        if(avg % 2 || maxl > avg/2) printf("NO\n");     //剪枝1， 对不能平分和单个辣条过重的进行剪枝
        else
        {
            avg /= 2;
            sort(a, a+n);       //便于之后剪枝
            dfs(0, 0);
            if(flag) printf("YES\n");
            else printf("NO\n");
        }
    }
    return 0;
}