3

我想在 python 中生成所有 3 位回文数的列表。我可以用粗略的方式编写代码，但有没有使用列表理解itertools等直观的方式？

而且，如果给定的数字是k数字而不是3，该怎么办？

4

2 回答 2

11 
>>> L = [int("%d%d%d" % (x,y,x)) for x in range(1,10) for y in range(10)]
>>> L
[101, 111, 121, 131, 141, 151, 161, 171, 181, 191, 202, 212, 222, 232, 242, 252,
 262, 272, 282, 292, 303, 313, 323, 333, 343, 353, 363, 373, 383, 393, 404, 414,
 424, 434, 444, 454, 464, 474, 484, 494, 505, 515, 525, 535, 545, 555, 565, 575,
 585, 595, 606, 616, 626, 636, 646, 656, 666, 676, 686, 696, 707, 717, 727, 737,
 747, 757, 767, 777, 787, 797, 808, 818, 828, 838, 848, 858, 868, 878, 888, 898,
 909, 919, 929, 939, 949, 959, 969, 979, 989, 999]

更新：为了提高内存和速度效率，您可以将字符串格式和 int 转换替换为x+y*10+x*100. 谢谢@larsmans。

更新2：那是k数字！

[int(''.join(map(str, (([x]+list(ys)+[z]+list(ys)[::-1]+[x]) if k%2
                  else ([x]+list(ys)+list(ys)[::-1]+[x])))))
            for x in range(1,10)
            for ys in itertools.permutations(range(10), k/2-1)
            for z in (range(10) if k%2 else (None,))]

并且优化为不使用字符串！

[sum([n*(10**i) for i,n in enumerate(([x]+list(ys)+[z]+list(ys)[::-1]+[x]) if k%2
                                else ([x]+list(ys)+list(ys)[::-1]+[x]))])
            for x in range(1,10)
            for ys in itertools.permutations(range(10), k/2-1)
            for z in (range(10) if k%2 else (None,))]

我使用了排列，第一个数字使用了一个不同的循环，它不能是0，最后一个循环是在中间添加所有可能的数字 if k%2 == 1(k 是奇数)。

欢迎提出优化建议！

于 2012-05-20T12:37:35.190 回答
2

对于k数字的泛化，最明显的方法是执行以下操作：

palindromes = [x for x in itertools.permutations(string.digits, k) if x == x[::-1]]

但它不是很有效 - 它生成所有可能的 3 位数字，并丢弃那些不是回文的数字。但是，可以推广@jadkik94 之类的解决方案 - 您需要做的是生成一半长度的每个组合（向下舍入），并将其镜像粘贴到最后：

palindromes = [x + x[::-1] for x in permutations(digits, k//2)]

将适用于所有偶数k- 对于奇数k，您添加一个额外的循环将所有 0-9 放在镜像的两半之间。

于 2012-05-20T12:44:27.953 回答