Wybierz x losowe elementy z listy ważonej w C # (bez zastąpienia)
Aktualizacja: mój problem został rozwiązany, zaktualizowałem źródło kodu w moim pytaniu, aby pasowało do odpowiedzi Jason'a. Zauważ, że odpowiedź rikitikitik rozwiązuje problem wybierania kart z próbki z wymianą.
Chcę wybrać x losowe elementy z listy ważonej. Pobieranie próbek jest bez wymiany. Znalazłem tę odpowiedź:https://stackoverflow.com/a/2149533/57369 z implementacją w Pythonie. Zaimplementowałem go w C # i przetestowałem. Ale wyniki (jak opisano poniżej) nie były zgodne z oczekiwaniami. Nie znam Pythona, więc jestem pewien, że popełniłem błąd podczas przenoszenia kodu do C #, ale nie widzę, gdzie kod w Pythong był naprawdę dobrze udokumentowany.
Wybrałem jedną kartę 10000 razy i takie wyniki uzyskałem (wynik jest spójny w różnych wykonaniach):
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)
Jak widzisz, karta 1 i karta 4 mają zarówno wagę 1, jak i kartę 1, są wybierane częściej niż karta 4 (nawet jeśli wybieram 2 lub 3 karty).
Dane testowe:
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 %
};
Oto moja implementacja w 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; }
}