TOJ 3583: A and B Problem

描述

相信你已经AC了A + B Problem，是不是很简单呢。
下面继续A and B Problem这个简单的问题，将若干个长度不同的木棍，分成A堆和B堆，并且保证A堆所有木棍的长度之和等于B堆所有木棍的长度之和。

输入

输入数据有多组。
每组测试数据第一行为一个正整数n(n<=100)，代表总的木棍个数；
第二行为n个以空格隔开的正整数ci(ci<=100)。

输出

如果n根木棍可以分成两堆木棍（A堆和B堆），并且A堆所有木棍的长度之和等于B堆所有木棍的长度之和，则输出"Yes" ,否则输出"No"（A堆和B堆中的木棍个数不必相等）。

样例输入

4
1 2 3 4
2
1 2

样例输出

Yes
No

dfs暴力法：

先算出所有木条的长度和，若是奇数则必定不能二等分，反之则进行dfs，每条木头都选择加或者不加，若递归中遇到累加的和等于总长的一般则返回。
代码：

#include<bits/stdc++.h>
using namespace std;
int n,ls[101],sum,lim;
bool f;
void dfs(int val,int pos)
{
    if(val<lim||pos==n||f)return;
    if(val==lim)
    {
        f=true;
        return;
    }
    dfs(val-ls[pos],pos+1);
    dfs(val,pos+1);
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        sum=0;
        f=false;
        for(int i=0;i<n;i++)
        {
            scanf("%d",&ls[i]);
            sum+=ls[i];
        }
        lim=sum>>1;
        if(!(sum&1))dfs(sum,0);
        printf("%s\n",f?"Yes":"No");
    }
    return 0;
}

dfs法的最坏情况应该是2^n，题目木条数最大为100，这谁顶的住啊—。—；
估计训练时0ms过应该是侥幸了。
出题解后，参考了下学长的代码（https://www.cnblogs.com/taozi1115402474/p/10884255.html），初步理解了一点dp；

dp稳健法：
dfs法的效率之所以低，是因为在递归的过程中存在着许多重复的环节，而dp法做的就是记录下每个环节的值，避免了重复计算，大大提高了效率；
与上述dfs法类似从第一个木棒一个个遍历过去，记录下添加这个木棒后可以得到哪些值；
代码：

#include<stdio.h>
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        int val[101],sum=0,k;
        bool dp[5001]={false};
        for(int i=0;i<n;i++)
        {
            scanf("%d",&val[i]);
            sum+=val[i];
        }k=sum>>1;
        if(!(sum&1))
        {
            dp[0]=true;
            for(int i=0;i<n;i++)
                for(int j=k;j>=val[i];j--)
                    {
                        dp[j]=dp[j]|dp[j-val[i]];
                        if(dp[k])goto xx;
                    }
        }
       xx: printf("%s\n",dp[k]?"Yes":"No");
    }
    return 0;
}

emmmm感觉半个脚趾头进入dp大门了，争取早日半只脚^ _ ^;