maig-rps-weighted.py

#!/usr/local/bin/pypy

'''
RESULTS
========

6 / 64 :  684043  points [0, 7, 3]
5 / 64 :  680089  points [0, 6, 4]
7 / 64 :  679017  points [0, 8, 2]
8 / 64 :  670540  points [0, 9, 1]
4 / 64 :  665532  points [0, 5, 5]
3 / 64 :  631043  points [0, 4, 6]
2 / 64 :  573314  points [0, 3, 7]
15 / 64 :  518180  points [1, 6, 3]
14 / 64 :  516149  points [1, 5, 4]
16 / 64 :  512342  points [1, 7, 2]
'''

from random import randrange

def sumOfDigits(number):
    if number < 0:
		return None

	out = 0
	while number > 0:
		out += number % 10
		number = number // 10

	return out

def numberOfDigits(number):
	if number < 0:
		number = number * (-1)

	out = 0
	while number > 0:
		out += 1
		number = number // 10

	return out

def digitsAsList(number, minSize):
	out = []
	numberCopy = number

	while number > 0:
		out.append(number % 10)
		number = number // 10

	while minSize > numberOfDigits(numberCopy):
		out.append(0)
		minSize = minSize - 1

	return out[::-1] # reverses a list

def rps_amount(pick1, pick2):
	if pick1 == 'p' and pick2 == 'r':
		return 3

	if pick1 == 'r' and pick2 == 's':
		return 2

	if pick1 == 's' and pick2 == 'p':
		return 1
	
	return 0

def pickRPS(prob1, prob2):
	if rand1 <= prob1:
		return 'r'
	elif rand1 <= prob1 + prob2:
		return 'p'
	
	return 's'

numberOfRounds = 10000

if __name__ == '__main__':
	players = []

	for crt in range(911):
		if sumOfDigits(crt) == 10:
			players.append(digitsAsList(crt, 3))

	players.append((10, 0, 0))

	n = len(players)
	points = [0] * n

	for i in range(n):
		for j in range(i+1, n):
			# play a number of rounds between i and j
			for round in range(numberOfRounds):
				rand1 = randrange(99)
				rand2 = randrange(99)

				pick1 = pickRPS(players[i][0] * 10, players[i][1] * 10)
				pick2 = pickRPS(players[j][0] * 10, players[j][1] * 10)

				points[i] += rps_amount(pick1, pick2)
				points[j] += rps_amount(pick2, pick1)
	
	best = [0] * 10

	for i in range(len(players)):
		if points[i] > points[best[0]]:
			best[0] = i

			for j in range(len(best) - 1):
				if points[best[j]] > points[best[j+1]]:
					best[j], best[j+1] = best[j+1], best[j]
				else:
					break
	
	for crt in best[::-1]:
		print crt, '/', n, ': ', points[crt], ' points', players[crt]