n的k划分算法

最新推荐文章于 2023-12-13 09:24:01 发布

原创最新推荐文章于 2023-12-13 09:24:01 发布 · 2.3k 阅读

2 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#算法 #iterator #c

C/C++ 专栏收录该内容

4 篇文章

订阅专栏

本文探讨了如何解决自然数n的k划分问题，即找出所有因子为自然数且和为n的组合。通过使用分治法和格子模型进行分析，提出了一个(n行 * k列)的格子来表示这个问题。算法实现采用C++，并且讨论了几个拓展问题，如限制最大数不超过x的划分、只用奇数的划分以及确保所有整数不同的划分等。

问题：对给定的自然数n，求所有的运算因子为自然数且和为n的方案数？如NSolution(4)={1+1+1+1, 2+1+1, 1+3, 2+2} = 4

思路：分治法，因子的个数可以为2到n个，则子问题为：给定的k和n，(2<=k<=n)，求k个自然数相加，和为n的方案数。简称nk问题。

nk问题可以用格子模型来完美解决：

  *     模型：假设有一个(n行 * k列)的格子，每个格子里面至少放一块积木，并且a1<=a2<=a3<=...ak
*      那么n的k划分问题就是求 n=a1+a2+...ak的方案数。
*      第一列可以放1至n-k+1个积木，假设第一列放了i个积木，则所有列都至少放i个积木，即必须(n-k*i >= 0);

* *可以证明(n-k*i>=0)和条件(i<=n-k+1)前者涵盖后者，证明如下：
* * n>=k*i; n>=i+k-1
* * 若k*i>=i+k-1，则前者条件涵盖后者条件。
* * k*i>=i+k-1 => (k-1)i>=k-1 => i>=1
* * 由此证明n>=k*i>=i+k-1
* 则第二列的放法可以退化递归为(n-i-(k-1)*(i-1))块积木放到(k-1)列格子里去的格子模型问题。

nk问题的c++算法实现如下：

#include <iostream>
#include <list>
using namespace std;

std::list<int> l; // 存放中间结果
void PrintSolution(); //打印结果

int NKSolution(int n, int k)
{
    if (k > n) return 0;

    if (k == 1)
        {
        // found solution, print it
        l.push_back(n);
        PrintSolution();
        l.pop_back();
        return 1;
        }

    int count = 0;
    for(int i = 1; n-k*i >= 0; i++)
    {
        l.push_back(i);
        count += NKSolution(n-i-(k-1)*(i-1), k-1);
        l.pop_back();
    }

    return count;
}

void PrintSolution()
{
    std::list<int>::iterator it = l.begin();
    int previous = 1;
    while(it != l.end())
    {
        previous = *it + previous - 1;
        cout << previous << "  ";
        it++;
    }
    cout << endl;
    return;
}

拓展问题：

将n划分成最大数不超过x的划分数。
将n划分成若干奇正整数之和的划分数。
将n划分成若干不同整数之和的划分数。

。。。