sc2 Random

# -*- coding: utf-8 -*-
import random
print(" **欢迎使用 谐星*随机抽取三兵种程序**")
print("开发（小白）兼测试兼文案 maybe#5383  本人可乐于为广大星际玩家提供此类形式的程序的技术支持" )
print(" **本规则在有裁判的情况下谐星效果更加**")
print("建议使用方法：")
print ()
print("1、玩家各自运行一次程序并抽取单位，开始游戏（不能保证不是试了好多遍碰出来）")
print ()
print("2、由裁判公开运行程序，为玩家抽取单位，开始游戏")
print ()
print("3、由裁判运行程序 单独通知玩家可使用单位后开始游戏 增加戏剧性（需要正直的裁♂判）")

zerg = ['跳虫','爆虫','蟑螂','破坏者','刺蛇','潜伏者','腐化者','巢虫领主','宿主','王虫（升级空投）','虫后','飞蛇','雷兽','感染者','异龙']
protoss = ['狂热者','使徒','机械哨兵','追猎者','高阶圣堂武士','执政官','黑暗圣堂武士','折跃棱镜','不朽者','巨像','干扰者','凤凰','虚空辉光舰','先知','航母','风暴战舰']
terran = ['陆战队员','劫掠者','收割者','幽灵','恶火（蝠）','攻城坦克','飓风','雷神','寡妇雷','医疗运输机','维京战机（机甲）','女妖','解放者','铁鸦','战列巡航舰']

dict = {'爆虫':'跳虫','破坏者':'蟑螂', '潜伏者':'刺蛇', '巢虫领主':'腐化者', };

def Count(I,R):
    if I == '执政官': return R.count('高阶圣堂武士')+R.count('黑暗圣堂武士')
    if dict.get(I)!=None:  return R.count(dict.get(I));

def Start(C='Z',N='3'):
    if(N=='65535'):
        print ()
        print ()
        print ("▄︻┻┳═一……请输入种族 z p t ?")
        C = input().lower()#choice
        while(['z','p','t'].count(C)==0):
            print ("输入有误,重新输入")
            C = input().lower()#choice
        print ("▄︻┻┳═一……请输入兵种数量")
        N = input()#number
        while(not N.isdigit()):
            print ("请输入不超过10的整数")
            N = input()
    N=int(N)
    CR = zerg;
    if C =="z":
        CR = zerg;
    elif C =="p":
        CR = protoss;
    elif C =="t":
        CR = terran;
    R= [];
    for i in range(N):
        now = random.choice(CR)
        while(R.count(now) != 0):
            now = random.choice(CR)
        R.append(now)
    for i in R:
        if Count(i,R) == 0:
            #print(R,' eroor');
            Start(C,N)
            return;
    print(R);

while True:
    Start('Z','65535');