Mafiascum.net

Code: Select all

def chanceToRoll (number, rolls):
  if number < 0 or number > rolls*2:
    return 0
  if rolls == 1:
    return 1/3
  return (chanceToRoll(number, rolls-1) + chanceToRoll(number - 1, rolls-1) + chanceToRoll(number - 2, rolls-1)) / 3

mode = int(input("If you wish to activate lab, enter 0. If you wish to attack, enter 1: "))
if mode == 0:
  myStat = int(input("Enter your knowlege: "))
  activationCost = int(input("Enter how many knowledge to activate the lab: "))
  result = 0
  for i in range (activationCost, myStat * 2 + 1):
    result += chanceToRoll (i, myStat)
  print ("You will activate the lab with probability of: " + str(round(result*100, 1)) + "%")
if mode == 1:
  myStat = int(input("Enter your roll stat: "))
  enemyStat = int(input("Enter enemy roll stat: "))
  bonus = int(input("Enter your bonus to roll (can be negative): "))
  knockOut = int(input("Enter how many damage is needed for KO: "))
  result = []
  for i in range (0, myStat*2 + bonus + 1):
    result.append(0)
  for i in range (0, enemyStat*2 + 1):
    for j in range (0, myStat*2 + bonus + 1):
      if j < i:
        result[0] += chanceToRoll(i, enemyStat) * chanceToRoll(j - bonus, myStat)
      else:
        result[j-i] += chanceToRoll(i, enemyStat) * chanceToRoll(j - bonus, myStat)
  for i in range (0, knockOut):
    print("Chance to deal " + str(i) + " damage is " + str(round(result[i]*100, 1)) + "%")
  a = 0
  for i in range (knockOut, myStat*2 + bonus + 1):
      a += result[i]
  print("Chance to kill the enemy is " + str(round(a * 100, 1)) + "%")