数的划分(dfs+剪枝)

题目描述
将整数 n 分成 k 份，且每份不能为空，任意两个方案不相同（不考虑顺序）。

例如：n=7，k=3，下面三种分法被认为是相同的。

1,1,51,1,5;
1,5,11,5,1;
5,1,15,1,1.

问有多少种不同的分法。

输入格式
n,k （6<n <=200,2≤k≤6）

输出格式
1 个整数，即不同的分法。

输入输出样例
输入
7 3
输出
4
说明/提示
四种分法为：
1,1,5
1,2,4
1,3,3
2,2,3

40分做法：主要思路：本题很容易想到的方法是枚举1~(n-sum)的k个数，然后去判断当前的和是否等于n，此题还需要考虑到去重，所以可以用一个set集合，把k个数存储进容器再存储进集合中，最后输出集合的个数即可，代码如下：

#include<bits/stdc++.h>
using namespace std;
long long res=0;
set<vector<int> > st;
void fun(int n,int k,int current,vector<int> v)
{
    if(k==0)
    {
        if(current==n)
        {
            sort(v.begin(),v.end());
            st.insert(v);
        }
    }
    else if(k>0)
    {
        for(int i=1;i<=n-current;i++)
        {
            v.push_back(i);
            fun(n,k-1,current+i,v);
            v.pop_back();

        }
    }

}
int main()
{
    int n,k;
    cin>>n>>k;
    vector<int> v;
    fun(n,k,0,v);
    int size = st.size();
    /*for(set<vector<int> >::iterator it=st.begin();it!=st.end();it++)
    {
        int length = (*it).size();
        for(int i=0;i<length;i++)
        {
            cout<<(*it)[i]<<" ";
        }
        cout<<endl;
    }*/
    cout<<st.size()<<"\n";
    return 0;
}

**100分做法：**上面那样做后面三个测试点会超时，仔细思考发现我们再加入集合的时候用到了sort()函数进行排序，使得如果一样的数组都会升序排列，比如5 1 1 、 1 5 1再加入的时候都会变成1 1 5，那么其实可以在for循环里面进行剪枝操作，加上一个限制条件，每次加入的数不得比上一个数小，修改后代码如下：

#include<bits/stdc++.h>
using namespace std;
int n,k,res;
void fun(int last,int sum,int cur)
{
    if(cur==k)
    {
        if(sum==n)
        {
           res++;
        }
        return;
    }
    else if(k>0)
    {
        //sum+i*(k-cur)<=n 有人判断条件是这个，但是我不是很能想到。
        for(int i=last;i<=n-sum;i++)
        {
            fun(i,sum+i,cur+1);
        }
    }

}
int main()
{
    cin>>n>>k;
    fun(1,0,0);
    cout<<res<<"\n";
    return 0;
}