我可以粗略地对此进行编码,但使用列表理解或itertools等有一些直观的方法吗?
而且,如果给出的数字是k
个数字而不是3个,怎么办?
答案 0 :(得分: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]
UPDATE:为了提高内存和速度,您可以用x+y*10+x*100
替换字符串格式和int转换。谢谢@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
,最后一个循环是在k%2 == 1
(k为奇数)时在中间添加所有可能的数字。
欢迎提出优化建议!
答案 1 :(得分:2)
对于k
数字的推广,最明显的方法是执行以下操作:
palindromes = [x for x in itertools.permutations(string.digits, k) if x == x[::-1]]
但效率不高 - 它会生成每个可能的3位数字,并丢弃那些不是回文的数字。但是,有可能推广像@ jadkik94这样的解决方案 - 您需要做的是生成 half 长度的每个组合(向下舍入),并在最后粘贴它的镜像:
palindromes = [x + x[::-1] for x in permutations(digits, k//2)]
对于所有偶数k
都适用 - 对于奇数k
,你添加一个额外的循环来将所有0-9放在镜像的一半之间。