2017-06-17 80 views
0

我正在尝试在JavaScript中编写一个函数,通过在itertools.permutations的Python文档中移植code来生成给定数组的排列数组。 (我知道实际函数是用C编写的)这就是我所拥有的,它输出一个正确长度的数组--n!/(nr)!, n是数组的长度 - 但每个元素都是正确的原始数组,不重新排列。我会很感激一双崭新的眼睛对我的代码,因为我很为难:JavaScript排列功能 - 为什么不能正常工作?

function permutations(array, r) { 
    if (r === undefined) r = array.length; 
    if (r > array.length) return; 
    var indices = range(array.length); 
    var cycles = range(array.length, array.length - r, -1); 
    var result = [[]]; 
    for (var i = 0; i < r; i++) { 
     result[0].push(array[i]); 
    } 
    while (1) { 
     var exhausted = true; 
     for (var i = r - 1; i >= 0; i--) { 
      cycles[i] -= 1; 
      if (cycles[i] == 0) { 
       indices = indices.slice(0, i).concat(
        indices.slice(i + 1) 
       ).concat([indices[i]]); 
       cycles[i] = array.length - i; 
      } 
      else { 
       var j = cycles[i]; 
       swap(indices, i, indices.length - j); 
       var p = []; 
       for (var i = 0; i < r; i++) { 
        p.push(array[i]); 
       } 
       result.push(p); 
       exhausted = false; 
       break; 
      } 
     } 
     if (exhausted) break; 
    } 
    return result; 
} 
+0

最接近的等价物可能是一个生成器'函数*'并使用类似于python的'yield'关键字。 –

回答

0

蟒蛇代码(接近底部):

yield tuple(pool[i] for i in indices[:r]) 

你翻译,为:

  for (var i = 0; i < r; i++) { 
       p.push(array[i]); 
      } 
      result.push(p); 

但它应该是:

  for (var i = 0; i < r; i++) { 
       p.push(array[indices[i]]); 
      } 
      result.push(p); 

没有这个,indices从不使用,这应该是一个线索。