随机打乱数组很容易解决。我想做一个洗牌,但有一个额外的限制,即任何元素的变化都被限制在一个范围内。因此,如果 max allowed shift ,则由于洗牌,任何元素都不能在任一方向上= n
移动超过步数。n
所以给定这个数组,并且 n=3:
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
这将是一个有效的洗牌:
[2, 3, 4, 0, 1, 6, 5, 8, 9, 7]
虽然这些是无效的:
[2, 3, 4, 7, 1, 6, 5, 8, 9, 0]
[2, 3, 4, 6, 1, 7, 5, 8, 9, 0]
(注意范围不是旋转的)
我们正在寻找一种简单有效的方法来实现这一目标。最好就地进行,但如果它提供了一个好的解决方案,则可以使用第二个数组。
一个天真的入门解决方案是,使用第二个数组:
for element in array1:
get legal index range
filter out indexes already filled
select random index i from filtered range
array20[i] = element
编辑:
这是关于@ruakh 提出的概率失真问题,如果算法首先以相等的概率处理终端元素:
乍一看,我认为概率方差会随着数组大小的增加而减小,但情况似乎并非如此。下面的一些快速测试(我匆忙炮制了这个,所以可能有错误)。由于概率的失真很大,我认为它不能作为一般情况接受,但对于我自己的应用程序,我可以忍受它,就像我在评论中所说的那样。
import itertools
n = 2
def test(arlen):
ar = range(arlen)
lst = list(itertools.permutations(ar))
flst = [l for l in lst if not illegal(l)]
print 'array length', arlen
print 'total perms: ', len(lst)
print 'legal perms: ', len(flst)
frst = [0] * (n+1)
for l in flst:
frst[l[0]] +=1
print 'distribution of first element: ',frst
def illegal(l):
for i in range(len(l)):
if abs(l[i]-i)>n: return True
if __name__=="__main__":
arlen = range(4,10)
for ln in arlen:
test(ln)
------------ n=2
array length 4
total perms: 24
legal perms: 14
distribution of first element: [6, 4, 4]
array length 5
total perms: 120
legal perms: 31
distribution of first element: [14, 10, 7]
array length 6
total perms: 720
legal perms: 73
distribution of first element: [31, 24, 18]
array length 7
total perms: 5040
legal perms: 172
distribution of first element: [73, 55, 44]
array length 8
total perms: 40320
legal perms: 400
distribution of first element: [172, 128, 100]
array length 9
total perms: 362880
legal perms: 932
distribution of first element: [400, 300, 232]
------------ n=4
array length 4
total perms: 24
legal perms: 24
distribution of first element: [6, 6, 6, 6, 0]
array length 5
total perms: 120
legal perms: 120
distribution of first element: [24, 24, 24, 24, 24]
array length 6
total perms: 720
legal perms: 504
distribution of first element: [120, 96, 96, 96, 96]
array length 7
total perms: 5040
legal perms: 1902
distribution of first element: [504, 408, 330, 330, 330]
array length 8
total perms: 40320
legal perms: 6902
distribution of first element: [1902, 1572, 1296, 1066, 1066]
array length 9
total perms: 362880
legal perms: 25231
distribution of first element: [6902, 5836, 4916, 4126, 3451]