Selecione x elementos aleatórios de uma lista ponderada em C # (sem substituição)
Atualizar: meu problema foi resolvido, atualizei a fonte do código na minha pergunta para combinar com a resposta de Jason. Observe que a resposta do rikitikitik é resolver a questão de escolher cartões de uma amostra com substituição.
Eu quero selecionar x elementos aleatórios de uma lista ponderada. A amostragem é sem substituição. Eu encontrei esta resposta:https://stackoverflow.com/a/2149533/57369 com uma implementação em Python. Eu implementei em C # e testei. Mas os resultados (conforme descrito abaixo) não estavam combinando com o que eu esperava. Eu não tenho conhecimento do Python, então tenho certeza que cometi um erro ao portar o código para o C #, mas não consigo ver onde o código em Pythong foi realmente bem documentado.
Eu escolhi um cartão 10000 vezes e este é o resultado que obtive (o resultado é consistente em todas as execuções):
Card 1: 18.25 % (10.00 % expected)
Card 2: 26.85 % (30.00 % expected)
Card 3: 46.22 % (50.00 % expected)
Card 4: 8.68 % (10.00 % expected)
Como você pode ver, o Cartão 1 e o Cartão 4 têm um peso de 1, mas o Cartão 1 é de awlays escolhido com mais frequência do que o cartão 4 (mesmo que eu escolha 2 ou 3 cartões).
Dados de teste:
var cards = new List<Card>
{
new Card { Id = 1, AttributionRate = 1 }, // 10 %
new Card { Id = 2, AttributionRate = 3 }, // 30 %
new Card { Id = 3, AttributionRate = 5 }, // 50 %
new Card { Id = 4, AttributionRate = 1 }, // 10 %
};
Aqui está a minha implementação em c #
public class CardAttributor : ICardsAttributor
{
private static Random random = new Random();
private List<Node> GenerateHeap(List<Card> cards)
{
List<Node> nodes = new List<Node>();
nodes.Add(null);
foreach (Card card in cards)
{
nodes.Add(new Node(card.AttributionRate, card, card.AttributionRate));
}
for (int i = nodes.Count - 1; i > 1; i--)
{
nodes[i>>1].TotalWeight += nodes[i].TotalWeight;
}
return nodes;
}
private Card PopFromHeap(List<Node> heap)
{
Card card = null;
int gas = random.Next(heap[1].TotalWeight);
int i = 1;
while (gas >= heap[i].Weight)
{
gas -= heap[i].Weight;
i <<= 1;
if (gas >= heap[i].TotalWeight)
{
gas -= heap[i].TotalWeight;
i += 1;
}
}
int weight = heap[i].Weight;
card = heap[i].Value;
heap[i].Weight = 0;
while (i > 0)
{
heap[i].TotalWeight -= weight;
i >>= 1;
}
return card;
}
public List<Card> PickMultipleCards(List<Card> cards, int cardsToPickCount)
{
List<Card> pickedCards = new List<Card>();
List<Node> heap = GenerateHeap(cards);
for (int i = 0; i < cardsToPickCount; i++)
{
pickedCards.Add(PopFromHeap(heap));
}
return pickedCards;
}
}
class Node
{
public int Weight { get; set; }
public Card Value { get; set; }
public int TotalWeight { get; set; }
public Node(int weight, Card value, int totalWeight)
{
Weight = weight;
Value = value;
TotalWeight = totalWeight;
}
}
public class Card
{
public int Id { get; set; }
public int AttributionRate { get; set; }
}