递归3 X 3魔方

时间:2015-06-22 13:42:18

标签: c++ algorithm math recursion

我试图找到3X3魔方的所有可能解决方案。 应该有8个解决方案。

我的代码可以获得所有代码,但有很多重复。我很难跟踪递归步骤,看看为什么我会得到所有重复。

// This program finds all solutions to the magic square for a 3X3     
// square where each column, row and diagonal sum is equal

#include <iostream>
using namespace std;
#define SQUARE_SIZE 9

int anyLine = 0;
int currLine = 0;
int numSolutions = 0;


// swap two values in the square.
void swap(int arr[], int idxa, int idxb)
{
    int tmp = arr[idxa];
    arr[idxa] = arr[idxb];
    arr[idxb] = tmp;
}

void printArray(int arr[])
{
    for (int i = 0; i < SQUARE_SIZE; i++)
    {
        cout << arr[i] << " ";
        if ((i + 1) % 3 == 0)
            cout << endl;
    }
    cout << endl;
}

// this function tests to see if we have a "good" arrangement of numbers            
// i.e the sum of each row, column and diagonal is equal

bool checkArr(int arr[])
{
    anyLine = arr[0] + arr[1] + arr[2];
    currLine = 0;
    for (int i = 0; i < SQUARE_SIZE; i++)
    {
        currLine += arr[i];
        if ((i + 1) % 3 == 0)
        {
            if (currLine != anyLine)
                return false;
            currLine = 0;
        }
    }

    // check vertically
    for (int col = 0; col <3; col++)
    {
        for (int row = 0; row <3; row++)
        {
            currLine += arr[col + 3 * row];
        }

        if (currLine != anyLine)
            return false;

        currLine = 0;
    }

    // check the diagonals
    if ((arr[2] + arr[4] + arr[6]) != anyLine)
        return false;

    if ((arr[0] + arr[4] + arr[8]) != anyLine)
        return false;

    return true;
}

void solve(int arr[], int pos)
{
    if (pos == 8)
    {
        if (checkArr(arr))
        {
            printArray(arr);
            numSolutions++;
        }
    } else 
    {
        for (int i = 0; i < 9; i++)
        {
            if (i == pos) continue;

            if (checkArr(arr))
            {
                printArray(arr);
                numSolutions++;
            }
            swap(arr, pos, i);
            solve(arr, pos + 1);
        }
    }
}

int main()
{
    int arr[SQUARE_SIZE] = { 1, 2, 3, 4, 5, 6, 7, 8, 9 };

    solve(arr, 0);
    cout << "number of solutions is: " << numSolutions << endl;

    return 0;
}

3 个答案:

答案 0 :(得分:1)

基本上,您使用recursive permutation algorithm找到数组的所有排列。

您需要更改4件事:

首先,从pos开始循环,而不是0

其次,在递归后回交元素(回溯)

第三,只有在你完成每个完整的排列后才会测试(当pos = 8时),否则你将不止一次地测试相同的排列。

第四,交换元素本身(即不交换它)是一种有效的排列,因为元素可以保留在原始位置。

void solve(int arr[], int pos)
{
    if (pos == 8)
    {
        if (checkArr(arr))
        {
            printArray(arr);
            numSolutions++;
        }
    }
    else
    {
        for (int i = pos ; i < 9; i++)
        { 
            swap(arr,pos,i);
            solve(arr,pos +1);
            swap(arr,pos,i); 
        }
    }
}

Demo

答案 1 :(得分:1)

您的代码从两个地方调用printArray - 递归的基本情况(即pos == 8时)和调用swap之前的循环。第二次调用是不必要的:当您到达pos == 8状态时,您将获得相同的方格。

这会降低重复次数,但由于生成方块的方式不会消除它们。您需要跟踪已打印的内容。一种方法是制作一组您找到的解决方案,并在打印新找到的解决方案之前进行检查:

set<int> seen;

int key(int arr[]) {
    return arr[0]
    + 10 * arr[1]
    + 100 * arr[2]
    + 1000 * arr[3]
    + 10000 * arr[4]
    + 100000 * arr[5]
    + 1000000 * arr[6]
    + 10000000 * arr[7]
    + 100000000 * arr[8];
}

 void printArray(int arr[]) {
    if (!seen.insert(key(arr)).second) {
        // second is set to false when a duplicate is found
        return;
    }
    numSolutions++;
    for (int i = 0; i < SQUARE_SIZE; i++) {
        cout << arr[i] << " ";
        if((i+1) % 3 == 0)
            cout << endl;
    }
    cout << endl;
 }

Demo.

有关上述解决方案的一些注意事项:

  • key(int[])将方块转换为单个十进制数,因此此方法仅适用于由十进制数字组成的方块。对于任意数字,您需要一个不同的策略 - 例如,使用一组以逗号分隔的字符串。
  • 解决方案的计数移至printArray(int[])。您可以完全删除numSolutions,然后使用seen.size()代替;它提供了相同的答案。

答案 2 :(得分:0)

如果您不想出于练习目的而递归地解决此问题,我建议您使用void solve(int(&arr)[SQUARE_SIZE], int pos) { sort(std::begin(arr), std::end(arr)); do { if (checkArr(arr)) { numSolutions++; printArray(arr); } } while (next_permutation(begin(arr), end(arr))); }

public static class DynamicBundles
{
    public static IHtmlString RenderSkin(string skinDirectory)
    {            
        BundleTable.Bundles.Add(new StyleBundle("~/Content/css/" + skinDirectory).Include(
        "~/Content/" + skinDirectory + "/reset.css", 
        "~/Content/" + skinDirectory + "/site.css", 
        "~/Content/" + skinDirectory + "/grids.css"));

    return Styles.Render("~/Content/css/" + skinDirectory);
    }
}