Python经典游戏:简易版斗地主2.0(增加记牌器+重玩)

# -*- coding: utf-8 -*-# @Author : 小红牛# 微信公众号：wdPythonimport randomfrom typing import List, Tuple, Optional, Any# ---------- 牌的定义 ----------RANK_MAP = {    '3': 3, '4': 4, '5': 5, '6': 6, '7': 7, '8': 8, '9': 9, '10': 10,    'J': 11, 'Q': 12, 'K': 13, 'A': 14, '2': 15, '小王': 16, '大王': 17}VALUE_TO_RANK = {v: k for k, v in RANK_MAP.items()}SUITS = ['♠', '♥', '♣', '♦']class Card:    def __init__(self, rank: str, suit: str = ''):        self.rank = rank        self.suit = suit        self.value = RANK_MAP[rank]    def __repr__(self):        return f"{self.suit}{self.rank}" if self.suit else self.rank    def __str__(self):        return self.__repr__()    def __lt__(self, other):        return self.value < other.value# ---------- 牌型判断 ----------def get_card_type(cards: List[Card]):    if not cards:        return 'invalid', None    sorted_cards = sorted(cards, key=lambda c: c.value)    values = [c.value for c in sorted_cards]    count_dict = {}    for v in values:        count_dict[v] = count_dict.get(v, 0) + 1    n = len(cards)    # 火箭    if n == 2 and 16 in count_dict and 17 in count_dict:        return 'rocket', 100    # 炸弹    bombs = [v for v, cnt in count_dict.items() if cnt == 4]    if len(bombs) == 1 and n == 4:        return 'bomb', bombs[0]    # 四带二    if len(bombs) == 1:        bomb_val = bombs[0]        remaining = [v for v in values if v != bomb_val]        if len(remaining) == 2:            if remaining[0] == remaining[1]:                return 'four_with_pair', bomb_val            else:                return 'four_with_single', bomb_val        elif len(remaining) == 4:            rem_count = {}            for v in remaining:                rem_count[v] = rem_count.get(v, 0) + 1            if all(cnt == 2 for cnt in rem_count.values()) and len(rem_count) == 2:                return 'four_with_two_pairs', bomb_val    # 三张相关    trips = [v for v, cnt in count_dict.items() if cnt == 3]    if trips:        if len(trips) == 1 and n == 3:            return 'triple', trips[0]        if len(trips) == 1 and n == 4:            return 'three_with_one', trips[0]        if len(trips) == 1 and n == 5:            remaining = [v for v in values if v != trips[0]]            if len(remaining) == 2 and remaining[0] == remaining[1]:                return 'three_with_pair', trips[0]        if len(trips) >= 2:            trips_sorted = sorted(trips)            if all(v < 15 for v in trips_sorted):                if all(trips_sorted[i+1] - trips_sorted[i] == 1 for i in range(len(trips_sorted)-1)):                    triple_cnt = len(trips)                    remaining_vals = [v for v in values if v not in trips]                    rem_n = len(remaining_vals)                    if rem_n == 0:                        return 'airplane', (trips_sorted[0], triple_cnt)                    if rem_n == triple_cnt:                        rem_count = {}                        for v in remaining_vals:                            rem_count[v] = rem_count.get(v, 0) + 1                        if all(cnt == 1 for cnt in rem_count.values()):                            return 'airplane_with_single', (trips_sorted[0], triple_cnt)                    if rem_n == 2 * triple_cnt:                        rem_count = {}                        for v in remaining_vals:                            rem_count[v] = rem_count.get(v, 0) + 1                        if all(cnt == 2 for cnt in rem_count.values()) and len(rem_count) == triple_cnt:                            return 'airplane_with_pair', (trips_sorted[0], triple_cnt)    # 顺子    if n >= 5 and all(cnt == 1 for cnt in count_dict.values()):        sorted_vals = sorted(count_dict.keys())        if max(sorted_vals) <= 14 and min(sorted_vals) >= 3:            if all(sorted_vals[i+1] - sorted_vals[i] == 1 for i in range(len(sorted_vals)-1)):                return 'straight', (sorted_vals[0], n)    # 连对    if n >= 6 and n % 2 == 0 and all(cnt == 2 for cnt in count_dict.values()):        sorted_vals = sorted(count_dict.keys())        if max(sorted_vals) <= 14 and min(sorted_vals) >= 3:            if all(sorted_vals[i+1] - sorted_vals[i] == 1 for i in range(len(sorted_vals)-1)):                return 'straight_pair', (sorted_vals[0], n // 2)    # 单张    if n == 1:        return 'single', values[0]    # 对子    if n == 2 and values[0] == values[1]:        return 'pair', values[0]    return 'invalid', Nonedef can_beat(last_type: str, last_key: Any, cur_type: str, cur_key: Any) -> bool:    if cur_type == 'rocket':        return True    if last_type == 'rocket':        return False    if cur_type == 'bomb' and last_type != 'bomb':        return True    if last_type == 'bomb' and cur_type != 'bomb':        return False    if last_type == 'bomb' and cur_type == 'bomb':        return cur_key > last_key    if cur_type != last_type:        return False    if isinstance(last_key, tuple) and isinstance(cur_key, tuple):        if last_key[1] != cur_key[1]:            return False        return cur_key[0] > last_key[0]    else:        return cur_key > last_key# ---------- 玩家类 ----------class Player:    def __init__(self, name: str):        self.name = name        self.hand: List[Card] = []        self.is_landlord = False    def add_cards(self, cards: List[Card]):        self.hand.extend(cards)    def remove_cards(self, cards: List[Card]):        for card in cards:            self.hand.remove(card)    def show_hand(self) -> str:        sorted_hand = sorted(self.hand)        return ' '.join(str(c) for c in sorted_hand)# ---------- 游戏类 ----------class Game:    def __init__(self, player_names: List[str]):        self.players = [Player(name) for name in player_names]        self.deck = self._create_deck()        self.bottom_cards: List[Card] = []        self.landlord_index: Optional[int] = None        self.current_player_index: int = 0        self.last_play: Optional[Tuple[int, List[Card], str, Any]] = None        self.pass_count: int = 0        # 记牌器：记录每个点数还剩余的张数（初始为全部牌）        self.remaining_cards = {rank: 4 for rank in ['3','4','5','6','7','8','9','10','J','Q','K','A','2']}        self.remaining_cards['小王'] = 1        self.remaining_cards['大王'] = 1    def _create_deck(self) -> List[Card]:        deck = []        for rank, val in RANK_MAP.items():            if rank in ('小王', '大王'):                deck.append(Card(rank))            else:                for suit in SUITS:                    deck.append(Card(rank, suit))        random.shuffle(deck)        return deck    def deal_cards(self):        """发牌，每人17张，留3张底牌（不扣记牌器，因为牌还在玩家手中）"""        for i in range(3):            for j in range(17):                card = self.deck.pop()                self.players[j % 3].add_cards([card])                # 发牌时不扣记牌器，牌还在手上未打出        self.bottom_cards = self.deck        self.deck = []    def choose_landlord(self):        """随机选择地主，获得底牌（不扣记牌器）"""        self.landlord_index = random.randint(0, 2)        self.players[self.landlord_index].is_landlord = True        self.players[self.landlord_index].add_cards(self.bottom_cards)        # 底牌加入地主，仍然未打出，不扣记牌器        self.current_player_index = self.landlord_index        print(f"\n🎲 地主是 {self.players[self.landlord_index].name}，底牌: {', '.join(str(c) for c in self.bottom_cards)}")        # 游戏开始前显示一次记牌器（此时所有牌都在玩家手中）        self.show_card_counter()    def show_card_counter(self):        """显示记牌器：按点数顺序列出剩余张数（未打出的牌）"""        order = ['3','4','5','6','7','8','9','10','J','Q','K','A','2','小王','大王']        parts = []        for rank in order:            count = self.remaining_cards[rank]            # 只显示还剩余>0的牌，如果全部为0则显示“无牌”            if count > 0:                parts.append(f"{rank}:{count}")        print("\n" + "="*50)        if parts:            print("📋 记牌器 (未打出): " + "  ".join(parts))        else:            print("📋 记牌器: 所有牌均已打出")        print("="*50)    def next_player(self):        self.current_player_index = (self.current_player_index + 1) % 3    def play_round(self):        """进行一轮出牌，直到有人获胜"""        while True:            # 每回合开始显示记牌器            self.show_card_counter()            player = self.players[self.current_player_index]            print(f"\n--- 轮到 {player.name} ---")            print(f"手牌: {player.show_hand()}")            if self.last_play is None:                print("上一手牌: 无")            else:                last_player = self.players[self.last_play[0]]                cards_str = ' '.join(str(c) for c in self.last_play[1])                print(f"上一手牌: {last_player.name} 出了 {cards_str}")            action = self._player_action(player)            if action is None:  # 过                print(f"{player.name} 不要")                self.pass_count += 1                if self.pass_count >= 2:                    self.last_play = None                    self.pass_count = 0            else:                cards, ctype, key = action                player.remove_cards(cards)                # 从记牌器中扣除出的牌（这些牌已经打出）                for card in cards:                    self.remaining_cards[card.rank] -= 1                print(f"{player.name} 出了: {' '.join(str(c) for c in cards)}")                self.last_play = (self.current_player_index, cards, ctype, key)                self.pass_count = 0                if not player.hand:                    print(f"\n🏆 {player.name} 获胜！游戏结束。")                    return            self.next_player()    def _player_action(self, player: Player):        if player.name == "玩家":            return self._human_action(player)        else:            return self._ai_action(player)    def _human_action(self, player: Player):        while True:            inp = input("请输入要出的牌（点数，空格分隔）或输入 'pass' 过：").strip()            if inp.lower() == 'pass':                if self.last_play is None:                    print("现在是你先出牌，不能过")                    continue                return None            ranks = inp.split()            selected = []            hand_copy = player.hand[:]            for r in ranks:                found = False                for card in hand_copy:                    if card.rank == r:                        selected.append(card)                        hand_copy.remove(card)                        found = True                        break                if not found:                    print(f"手牌中没有 {r}，请重新输入")                    break            else:                ctype, key = get_card_type(selected)                if ctype == 'invalid':                    print("牌型不合法，请重新出牌")                    continue                if self.last_play is None:                    return selected, ctype, key                else:                    if can_beat(self.last_play[2], self.last_play[3], ctype, key):                        return selected, ctype, key                    else:                        print("不能压过上家，请重新出牌")                        continue    def _ai_action(self, player: Player):        hand = player.hand        if not hand:            return None        def all_plays(hand):            plays = []            n = len(hand)            for mask in range(1, 1 << n):                cards = [hand[i] for i in range(n) if (mask >> i) & 1]                ctype, key = get_card_type(cards)                if ctype != 'invalid':                    plays.append((cards, ctype, key))            return plays        if self.last_play is None:            plays = all_plays(hand)            for play in plays:                if play[1] == 'single':                    return play            return plays[0] if plays else None        else:            last_type, last_key = self.last_play[2], self.last_play[3]            plays = all_plays(hand)            beats = [p for p in plays if can_beat(last_type, last_key, p[1], p[2])]            if beats:                return beats[0]            else:                return None# ---------- 主程序，支持重复游戏 ----------if __name__ == "__main__":    print("🎴 欢迎来斗地主！")    while True:        game = Game(["玩家", "AI1", "AI2"])        game.deal_cards()        game.choose_landlord()        game.play_round()        again = input("\n是否再来一局？(y/n): ").strip().lower()        if again != 'y':            print("感谢游玩，再见！")            break