人马大战python代码教程 python人马大战攻略

import random

class Unit:
    def __init__(self, name, hp, attack, defense):
        self.name = name
        self.hp = hp
        self.attack = attack
        self.defense = defense

    def is_alive(self):
        return self.hp > 0

    def attack_target(self, target):
        damage = max(0, self.attack - target.defense) # 防御力抵消攻击力
        target.hp -= damage
        print(f"{self.name} attacks {target.name}, dealing {damage} damage!")

class Human(Unit):
    def __init__(self):
        super().__init__("Human", 10, 2, 1)

class Horse(Unit):
    def __init__(self):
        super().__init__("Horse", 15, 3, 0)


def battle(human_num, horse_num):
    humans = [Human() for _ in range(human_num)]
    horses = [Horse() for _ in range(horse_num)]
    
    turn = 0
    while humans and horses: #只要双方还有单位存活
        turn +=1
        print(f"\n---Turn {turn}---")
        attacker = random.choice(humans if turn % 2 else horses) #轮流攻击
        defender = random.choice(horses if turn % 2 else humans)

        if defender.is_alive():
            attacker.attack_target(defender)
            if not defender.is_alive():
                if isinstance(defender, Human):
                    humans.remove(defender)
                else:
                    horses.remove(defender)

    print("\nBattle over!")
    if humans:
        print("Humans win!")
    else:
        print("Horses win!")


# 开始游戏
human_num = 5
horse_num = 3
battle(human_num, horse_num)