计蒜客第28题---等和的分割子集

晓萌希望将1到N的连续整数组成的集合划分成两个子集合,且保证每个集合的数字和是相等。例如,对于N=3,对应的集合{1,2,3}能被划分成{3} 和 {1,2}两个子集合.
这两个子集合中元素分别的和是相等的。
对于N=3,我们只有一种划分方法,而对于N=7时,我们将有4种划分的方案。
输入包括一行,仅一个整数,表示N的值(1≤N≤39)。
输出包括一行,仅一个整数,晓萌可以划分对应N的集合的方案的个数。当没发划分时,输出0。
样例1
输入:
7
输出:
4

答案正确,但是复杂度高,会超时,使用递归算法,可以搜索“子集和数问题”,方法类似。

#include 
using namespace std;

//初始s=0,k=1,r是集合中元素的总和,x是解的01表示向量,m是要求的和,w是1到N连续整数组成的集合
void SumOfSub(int s,int k,int r,int *x,int m,int *w,int &solution)//solution为划分的个数
{
    x[k]=1;
    if(s+w[k]==m)//得到一个可行解
        solution++;
    else if(s+w[k]+w[k+1]<=m)
        SumOfSub(s+w[k],k+1,r-w[k],x,m,w,solution);//搜索左子树
    if((s+r-w[k]>=m)&&(s+w[k+1]<=m))
    {
        x[k]=0;
        SumOfSub(s,k+1,r-w[k],x,m,w,solution);//搜索右子树
    }
}

int Partition(int n)
{
    int sum =0;
    for(int i=1;i<=n;i++)
        sum+=i;
    if(sum%2)//集合所有元素的和为奇数,因此没有划分,返回0
        return 0;
    else
    {
        int *w=new int[n];
        int *x=new int[n];
        for(int i=1;i<=n;i++)//对连续整数组成的集合进行赋值
            w[i-1]=i;
        int half_sum=sum/2;
        int solution=0;
        SumOfSub(0,1,sum,x,half_sum,w,solution);
        delete []x;
        delete []w;
        return solution;
    }
}

int main()
{
    int a,result;
    cin >>a;
    cout<return 0;
}

你可能感兴趣的:(计蒜客)