Question

我必须做一个程序，使用回溯来提供n个数{1,2,3..n}的所有排列。我设法在C中完成它，它运行得很好，这是代码：

int st[25], n=4;

int valid(int k)
{
    int i;
    for (i = 1; i <= k - 1; i++)
        if (st[k] == st[i])
            return 0;
    return 1;
}
void bktr(int k)
{
    int i;
    if (k == n + 1)
    {
        for (i = 1; i <= n; i++)
            printf("%d ", st[i]);
        printf("\n");
    }
    else
        for (i = 1; i <= n; i++)
        {
            st[k] = i;
            if (valid(k))
                bktr(k + 1);
        }
    }
int main()
{
    bktr(1);
    return 0;
}

现在我必须用Python编写它。这是我做的：

st=[]
n=4
def bktr(k):
    if k==n+1:
        for i in range(1,n):
            print (st[i])

    else:
        for i in range(1,n):
            st[k]=i
            if valid(k):
                bktr(k+1)
def valid(k):
    for i in range(1,k-1):
        if st[k]==st[i]:
            return 0
    return 1
bktr(1)

我收到此错误：

list assignment index out of range

st[k]==st[i]。

Answer 1

Python在itertools模块中有一个“排列”函数：

import itertools
itertools.permutations([1,2,3])

如果您需要自己编写代码（例如，如果这是作业），这就是问题：

Python列表没有预定的大小，所以你不能只设置例如第10个元素到3.您只能更改现有元素或添加到最后。

Python列表（和C数组）也从0开始。这意味着您必须使用st[0]访问第一个元素，而不是st[1]。

当您启动程序时，st的长度为0;这意味着您无法分配到st[1]，因为它不是结束。

如果这令人困惑，我建议你改用st.append(element)方法，它总是添加到最后。

如果代码已经完成且有效，我建议你转到code review stack exchange，因为还有很多东西可以改进。

从C到Python的回溯排列

1 个答案: