Created
May 17, 2022 10:55
-
-
Save dasunsucharith/ecf135549bebfe4fecef4d020128a150 to your computer and use it in GitHub Desktop.
CS50 2022 Lab 6 Tournament Solution
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# Simulate a sports tournament | |
import csv | |
import sys | |
import random | |
# Number of simluations to run | |
N = 1000 | |
def main(): | |
# Ensure correct usage | |
if len(sys.argv) != 2: | |
sys.exit("Usage: python tournament.py FILENAME") | |
teams = [] | |
filename = sys.argv[1] | |
# TODO: Read teams into memory from file | |
with open(filename) as file: | |
reader = csv.DictReader(file) | |
for row in reader: | |
row["rating"] = int(row["rating"]) | |
teams.append(row) | |
counts = {} | |
# TODO: Simulate N tournaments and keep track of win counts | |
for i in range(N): | |
winner = simulate_tournament(teams) | |
if winner in counts: | |
counts[winner] += 1 | |
else: | |
counts[winner] = 1 | |
# Print each team's chances of winning, according to simulation | |
for team in sorted(counts, key=lambda team: counts[team], reverse=True): | |
print(f"{team}: {counts[team] * 100 / N:.1f}% chance of winning") | |
def simulate_game(team1, team2): | |
"""Simulate a game. Return True if team1 wins, False otherwise.""" | |
rating1 = team1["rating"] | |
rating2 = team2["rating"] | |
probability = 1 / (1 + 10 ** ((rating2 - rating1) / 600)) | |
return random.random() < probability | |
def simulate_round(teams): | |
"""Simulate a round. Return a list of winning teams.""" | |
winners = [] | |
# Simulate games for all pairs of teams | |
for i in range(0, len(teams), 2): | |
if simulate_game(teams[i], teams[i + 1]): | |
winners.append(teams[i]) | |
else: | |
winners.append(teams[i + 1]) | |
return winners | |
def simulate_tournament(teams): | |
"""Simulate a tournament. Return name of winning team.""" | |
# TODO | |
while len(teams) > 1: | |
teams = simulate_round(teams) | |
return teams[0]["team"] | |
if __name__ == "__main__": | |
main() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment