题解 | #数的划分#

数的划分

http://www.nowcoder.com/practice/24c2045f2cce40a5bf410a369a001da8

数的划分

问题描述:将整数n分成k份,且每份不能为空,任意两个方案不能相同(不考虑顺序)。
例如:n=7,k=3,下面三种分法被认为是相同的。
1,1,5;
1,5,1;
5,1,1;
问有多少种不同的分法。
输入:n,k ( 6 < n ≤ 200,2 ≤ k ≤ 6 )
输出:一个整数,即不同的分法。
示例
输入:7,3
返回值:4
说明:整数7分为3份有4种办法:[1,1,5][1,2,4][1,3,3][2,2,3]    

方法一

思路分析

      使用动态规划的办法进行分析,本题可以类比将$n$个小球放入$k$个盒子中,小球、盒子无差别,且盒子不能为空。设置将$n$个小球放入$k$个盒子中方法为$f[n][k]$,因此为了使每个盒子中不为空,先拿出$k$个小球分别分别放到每个盒子中,再将$n-k$个小球放入$k$个盒子中,其方法数为$f[n-k][k]$;或者保证每个盒子至少有一个小球,先将1个小球放入1个盒子中,剩余的$n-1$个小球放入$k-1$个盒子中其方法数为$f[n-1][k-1]$,因此总的方法数为$f[n][k]= f[n-k][k]+ f[n-1][k-1]$。具体步骤如下:
  • 设置$f[n][k]$为整数$n$分成$k$份,且每份不能为空的方法数
  • 其转移方程为$f[n][k]=f[n-k][k]+f[n-1][k-1]$
  • 初始化$f[0][0]=1$,只有当$n$大于等于$k$才可进行划分。
状态转移方程:


图解

输入:5,3
返回值:2

C++核心代码

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param n int 被划分的数
     * @param k int 化成k份
     * @return int
     */
        int mod=1000000007;
    int divideNumber(int n, int k) {
        // write code here
        if(n<k) return 0;
        vector<vector<int>> f(n+1,vector<int>(k+1));
        f[0][0]=1;//初始化
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=k;j++)
            {
                if(i>=j)//类比小球数量必须大于盒子数量k
                   f[i][j]=(f[i-1][j-1]+f[i-j][j])%mod;
            }
        }
        return f[n][k];
    }
};

复杂度分析

  • 时间复杂度:设置内外两层嵌套循环,外层循环次数为$n$,内层循环次数$k$,因此总的时间复杂度为$O(n*k)$
  • 空间复杂度:借助于一个二维数组$f[n][k]$用于存放整数$n$分成$k$份,且每份不能为空的方法数,因此空间复杂度为$O(n*k)$

方法二

思路分析

     直接采用暴力搜索的办法,对于一个整数,首次分配我们可以分配1,2,...,接着如果选定第一次分配1,那么第二次可以分配1,2,...,一直往下,直到所有的分配数之和为整数n,分配次数为k,此时这是一种分配的方案数。

图解



C++核心代码

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param n int 被划分的数
     * @param k int 化成k份
     * @return int
     */
    int depth;
    int sum;//存储划分方案数
    int divideNumber(int n, int k) {
        // write code hereif
        depth=k;
        sum=0;
        if(n<k) return 0;
        search(0,n,1);//第一个参数表示搜索的深度层数,第二个参数表示整数n的剩余数,第三个参数表示分配数开始的数,由于不能为空,因此开始为1
        return sum;
        
        
    }
    void search(int dep,int res,int start)
    {
        if(dep==depth+1)
        {
            if(res==0)//整数n已经分配完成,是一种分配方案
                sum++;
            return;
        }
        for(int i=start;i<=res;i++)//继续向下搜索,并且层数+1,n剩余的数需要减去开始的数
        {
            search(dep+1,res-i,i);
        }
    }
    
};

复杂度分析

  • 时间复杂度:将整数$n$分为$k$个,且每个不能为空,可类比为将$n$个小球使用挡板隔开,挡板数为$k-1$,需要的操作为,因此时间复杂度为
  • 空间复杂度:每次递归的空间为$O(1)$,共递归$k$次,因此空间复杂度为$O(k)$




全部评论

相关推荐

无情咸鱼王的秋招日记之薛定谔的Offer:好拒信,偷了,希望有机会用到
点赞 评论 收藏
分享
1 1 评论
分享
牛客网
牛客企业服务