按顺序生成数字分区的不同方法

时间:2015-06-17 21:49:28

标签: c++ algorithm number-theory

我在算法教科书中练习,要求我按顺序生成partitions of a number。我已经解决了这个问题,但在以传统的方式解决这个问题时,我提出了解决这个问题的新想法。

有传统方法:

#include <iostream>

#define MAX_PARTITIONS 100

int count = 1;

void printSolution(int* solution, int size)
{
    std::cout << count << "| ";
    count++;
    for(int i = 0; i < size; i++)
    {
        std::cout << solution[i] << " ";
    }
    std::cout << std::endl;
}

void generatePartitions(int* solution, int number, int sum, int partitions)
{
    if(sum == number)
    {
        printSolution(solution, partitions);
        return;
    }

    if(partitions > 1)
    {
        solution[partitions] = solution[partitions - 1];
    }
    else
    {
        solution[partitions] = 0;
    }

    while(solution[partitions] + sum < number)
    {
        solution[partitions]++;
        sum += solution[partitions];
        generatePartitions(solution, number, sum, partitions + 1);
        sum -= solution[partitions];
    }
}

int main(int argc, char const *argv[])
{
    int solution[MAX_PARTITIONS];

    generatePartitions(solution, 4, 0, 0);

    return 0;
}

基本上,当分区数大于1(存在前一个分区)时,我将前一个分区的值分配给当前分区,否则我将0分配给它。 / p>

有新方法:

#include <iostream>

#define MAX_PARTITIONS 100

int count = 1;

void printSolution(int* solution, int size)
{
    std::cout << count << "| ";
    count++;
    for(int i = 0; i < size; i++)
    {
        std::cout << solution[i] << " ";
    }
    std::cout << std::endl;
}

void generatePartitions(int* solution, int number, int sum, int partitions, 
    int start)
{
    if(sum == number)
    {
        printSolution(solution, partitions);
        return;
    }

    solution[partitions] = start;

    while(solution[partitions] + sum < number)
    {
        solution[partitions]++;
        sum += solution[partitions];
        generatePartitions(solution, number, sum, partitions + 1, 
            solution[partitions]);
        sum -= solution[partitions];
    }
}

int main(int argc, char const *argv[])
{
    int solution[MAX_PARTITIONS];

    generatePartitions(solution, 4, 0, 0, 0);

    return 0;
}

如您所见,函数generatePartitions现在采用了一个新参数start。最初,start的值为0,每次递归调用start都会获取当前solution[partitions]的值。所以它应该以与传统方法相同的方式工作。

但事实并非如此,这些程序的输出完全不同。我无法弄清楚出了什么问题。我究竟做错了什么?

1 个答案:

答案 0 :(得分:1)

当你递归时,你忘了移动solution数组的索引。

generatePartitions(solution, number, sum, partitions + 1, 
            solution[partitions]);

变为:

generatePartitions(solution, number, sum, partitions + 1, 
            solution[partitions-1]);

Coliru链接:http://coliru.stacked-crooked.com/a/83b0418e1b1c9bc1

另外,我建议使用某种形式的std::pair<int,int>和某种二叉树来为你做这件事 - 对我来说,从上到下更有意义,因为我们试图在技术上将数字划分或划分为较小的数字;算法可能有点复杂,但是当你完成它时也可能会显示一棵树。

相关问题