一步步解析Python斗牛游戏的概率

 # encoding=utf-8 __author__ = '[email protected]' import os import cPickle from copy import copy from collections import Counter import itertools ''' 计算斗牛游戏的概率 ''' class Poker(): ''' 一张牌 ''' def __init__(self, num, type): self.num = num # 牌数 self.type = type # 花色 class GamePoker(): ''' 一手牌，即5张Poker ''' COMMON_NIU = 1 # 普通的牛，即牛一-牛七 NO_NIU = 0 # 没有牛 EIGHT_NINE_NIU = 2 # 牛九或牛八 TEN_NIU = 3 # 牛十 THREE_SAME = 4 # 三条 FOUR_SAME = 5 # 四条 def __init__(self, pokers): assert len(pokers) == 5 self.pokers = pokers self.num_pokers = [p.num for p in self.pokers] # self.weight = None # 牌的权重，权重大的牌胜 # self.money_weight = None # 如果该牌赢，赢钱的权重 self.result = self.sumary() def is_niu(self): ''' 是否有牛 :return: ''' # if self.is_three_same(): # return 0 for three in itertools.combinations(self.num_pokers, 3): if sum(three) % 10 == 0: left = copy(self.num_pokers) for item in three: left.remove(item) point = sum(left) % 10 return 10 if point == 0 else point return 0 def is_three_same(self): ''' 是否3条 :return: ''' # if self.is_four_same(): # return 0 count = Counter([p.num for p in self.pokers]) for num in count: if count[num] == 3: return num return 0 def is_four_same(self): ''' 是否4条 :return: ''' count = Counter([p.num for p in self.pokers]) for num in count: if count[num] == 4: return num return 0 def sumary(self): ''' 计算牌 ''' if self.is_four_same(): return GamePoker.FOUR_SAME if self.is_three_same(): return GamePoker.THREE_SAME niu_point = self.is_niu() if niu_point in (8, 9): return GamePoker.EIGHT_NINE_NIU elif niu_point == 10: return GamePoker.TEN_NIU elif niu_point > 0: return GamePoker.COMMON_NIU else: return GamePoker.NO_NIU def get_all_pokers(): ''' 生成所有的Poker，共四十个 :return: ''' pokers = [] for i in range(1, 11): for j in ('A', 'B', 'C', 'D'): pokers.append(Poker(i, j)) return pokers def get_all_game_poker(is_new=0): ''' 生成所有game_poker :param pokers: :return: ''' pokers = get_all_pokers() game_pokers = [] if not is_new and os.path.exists('game_pokers'): with open('game_pokers', 'r') as f: return cPickle.loads(f.read()) for pokers in itertools.combinations(pokers, 5): # 5代表五张牌 game_pokers.append(GamePoker(pokers)) with open('game_pokers', 'w') as f: f.write(cPickle.dumps(game_pokers)) return game_pokers def print_rate(game_pokers): total_num = float(len(game_pokers)) four_num = len([game_poker for game_poker in game_pokers if game_poker.re<mark style="color:transparent">来源gaodaimacom搞#^代%!码网</mark>sult == GamePoker.FOUR_SAME]) three_num = len([game_poker for game_poker in game_pokers if game_poker.result == GamePoker.THREE_SAME]) ten_num = len([game_poker for game_poker in game_pokers if game_poker.result == GamePoker.TEN_NIU]) eight_nine_num = len([game_poker for game_poker in game_pokers if game_poker.result == GamePoker.EIGHT_NINE_NIU]) common_num = len([game_poker for game_poker in game_pokers if game_poker.result == GamePoker.COMMON_NIU]) no_num = len([game_poker for game_poker in game_pokers if game_poker.result == GamePoker.NO_NIU]) print '所有牌的组合数：%d' % total_num print '出现四条的组合数:%d,概率 :%.2f%%' % (four_num, four_num * 100 / total_num) print '出现三条的组合数:%d,概率 :%.2f%%' % (three_num, three_num * 100 / total_num) print '出现牛十的组合数:%d,概率 :%.2f%%' % (ten_num, ten_num * 100 / total_num) print '出现牛九或牛八的组合数:%d,概率 :%.2f%%' % (eight_nine_num, eight_nine_num * 100 / total_num) print '出现牛一到牛七的组合数:%d,概率 :%.2f%%' % (common_num, common_num * 100 / total_num) print '出现没有牛的组合数:%d,概率 :%.2f%%' % (no_num, no_num * 100 / total_num) def main(): game_pokers = get_all_game_poker() # 658008种 print_rate(game_pokers) main()