从带有加权元素的数组中获取随机元素

时间:2017-01-01 20:20:57

标签: javascript arrays

我有一系列对象代表我正在努力开发的游戏中的生物。这些对象具有(以及其他)唯一标识符和产生的权重(或概率)。

我正在尝试开发一种随机生成生物的算法,但我没有想出一种使用权重的方法(我真的不知道怎么做)。

有人可以帮忙吗?

生物数组的一个例子可能是:

var creatures = [
    {id: 1, weight: 25},
    {id: 2, weight: 15},
    {id: 3, weight: 5},
    {id: 4, weight: 45},
    {id: 5, weight: 10}
]

2 个答案:

答案 0 :(得分:8)

我发现在这个blog中用PHP实现了这个很好的算法,我认为migth适合你的需要。

我刚把它收录到JS。



var creatures = [{
    id: 1,
    weight: 25
  }, {
    id: 2,
    weight: 15
  }, {
    id: 3,
    weight: 5
  }, {
    id: 4,
    weight: 45
  }, {
    id: 5,
    weight: 10
  }],
  sumOfWeights = creatures.reduce(function(memo, creature) {
    return memo + creature.weight;
  }, 0),
  selectedWeigths = {};

function getRandom(sumOfWeights) {
  var random = Math.floor(Math.random() * (sumOfWeights + 1));

  return function(creature) {
    random -= creature.weight;
    return random <= 0;
  };
}

for (var i = 0; i < 1000; i++) {
  var creature = creatures.find(getRandom(sumOfWeights));
  selectedWeigths[creature.weight] = (selectedWeigths[creature.weight] || 0) + 1;
}

console.log(selectedWeigths);
&#13;
&#13;
&#13;

希望它有所帮助。

答案 1 :(得分:0)

创建一个新数组,并将每个生物的id添加到数组中加权的次数。然后得到一个介于0和数组大小之间的随机数,并返回该位置的id号。

var creatureIds = [];
for(var i=0;i<creatures.length;i++){
    for(var x=0;x<creatures[i].weight;x++){
        creatureIds.push(creatures[i].id);
    }
}

// get a random index between 0 and the ids length.
var min = 0;
var max = creatureIds.length;
var index = Math.floor(Math.random() * (max - min + 1)) + min;

var randomWeightedCreatureId = creatureIds[index];