C++分治算法之集合划分问题

问题描述

找出给定数目m集合中子集中元素数为n的子集个数，以及m个元素的非空子集数目。

题目分析

关于找子集的问题我们其实有另一种方法，就是通过深度优先搜索与回溯法找出这个集合中的所有子集，减去空子集之后求子集的数目。
但这个方法有一些麻烦，通常如果需要我们列出所有子集的时候我们会使用DFS+回溯。但是这个问题我们只是求数目，所以可以用分治法解决。
首先我们介绍一个重要的概念:Strling数。
Strling数分为两类，分别是第1类Strling数和第2类Strling数，这个问题我们只需要用到第2类，所以第1类感兴趣的可以自己查阅，这里只介绍第2类。
第2类Strling数实际上是集合的一个拆分，表示将m个不同的元素拆分成n个集合的方案数,记为S(m,n)。

下面介绍一下第2类Strling数的递推式:
假设把m+1个元素分成m个集合:
(1).m个元素构成了n-1个集合，第m+1个元素单独成了一个集合，则方案数为S(m,n-1)
(2).如果m个元素已经构成了n个集合，将第m+1个元素插入到任意一个集合中即可，此时方案数为n * S(m,n)
所以总体的递推式为:
S(m,n)=n*S(m-1,n) + S(m-1,n-1)
S(n,k)表示将基数为n的集合划分为正好k个非空集合的方法的数目
接下来是Bell数：
贝尔数是用来求所有非空子集个数的，我们只需要把所有Strling数加起来即可，因为总的子集数包括各种长度的子集。

所以我们就可以利用用代码来实现这两个公式来解决问题。
我们通过将问题分为m-1个元素分n-1个子集和m-1个元素分n个子集两种，最后合并即可。
先求出S，一个二维矩阵表示S(i,j),(0<=i<=m,0<=j<=n)
其中有几个特例:

S[m][m+1]=0
S[m][0]=0
S[0][0]=1

然后通过求出的S，利用贝尔数公式进行求和，得到给定m的所有非空子集数目。

代码

#include <iostream>
using namespace std;
int S[100][100];
void Strilingnumber2(int m,int n)
{
    S[0][0] = 1;
    for(int i = 1;i <= m;i++)
        S[i][0] = 0;
    for(int i = 0;i < m;i++)
        S[i][i+1] = 0;
    int min;
    for(int i = 1;i <= m;i++)
    {
        if(i < n)
            min = i;
        else
            min = n;
        for(int j = 1;j <= min;j++)
        {
            S[i][j] = j*S[i-1][j] + S[i-1][j-1];
        }
    }
}
int computeB(int m)
{
    int *B = new int[m];
    Strilingnumber2(m,m);
    for(int i = 0;i < m;i++)
    {
        B[i] = 0;
    }
    for(int i = 1;i<=m;i++)
    {
        for(int j = 0;j <= i;j++)
        {
            B[i-1] += S[i][j];
        }
    }
    return B[m-1];
}
int main()
{
    int m,n;
    cin>>m>>n;
    cout<<computeB(m)<<endl;
    cout<<S[m][n];
    return 0;
}

总结

时间复杂度O(n^2)