import random

def combatDrill(numDice, bonusDamage, rolls):
    damage = 0
    heat = 0
    for i in range(rolls):
        dice = [random.randint(1,6) for r in range(numDice)]
        bonusDice = 0
        while 1 in dice:
            dice.sort(reverse=True)
            dice.pop()
            dice = dice + [random.randint(1,6) for r in range(2)]
            heat += 1
            bonusDice += 1
        dice.sort(reverse=True)
        damage += sum( dice[ :numDice + bonusDice] )
    print(f'Damage:{damage/rolls} Heat:{heat/rolls}')

def combatDrillCrit(numDice, bonusDamage, rolls):
    damage = 0
    heat = 0
    for i in range(rolls):
        dice = [random.randint(1,6) for r in range(numDice)]
        bonusDice = 0
        while 1 in dice:
            dice.sort(reverse=True)
            dice.pop()
            dice = dice + [random.randint(1,6) for r in range(2)]
            heat += 1
            bonusDice += 1
        dice.sort(reverse=True)
        damage += sum( dice[ :int(numDice/2) + bonusDice] )
    print(f'Damage:{damage/rolls} Heat:{heat/rolls}')

def combatDrillWeakCrit(numDice, bonusDamage, rolls):
    damage = 0
    heat = 0
    for i in range(rolls):
        dice = [random.randint(1,6) for r in range(numDice)]
        dice.sort(reverse=True)
        dice = dice[:int(numDice/2)]
        while 1 in dice:
            dice.sort(reverse=True)
            dice.pop()
            dice = dice + [random.randint(1,6) for r in range(2)]
            heat += 1
        damage += sum(dice)
    print(f'Damage:{damage/rolls} Heat:{heat/rolls}')

combatDrillCrit(8, 0, 1000000)
combatDrill(4,0,1000000)
combatDrillWeakCrit(8, 0, 1000000)