4-p04_guess_name

import math

NAMES='赵钱孙李周吴郑王冯陈褚卫蒋武韩'

def get_rows():
    num=len(NAMES)
    rows=int(math.log2(num)+1)
    result=[''for _ in range(rows)]
    id =1
    for name in NAMES:
        insert_name(id, name, result, rows)
        id+=1

    return result


def insert_name(id, name, result, rows):
    for row_id in range(len(result)):
        if id % 2 != 0:
            result[row_id] += name + ' '
        id //= 2


def get_name(answers):
    result = 0
    for answer in answers:
        result *= 2
        if answer:
            result += 1
    return NAMES[result-1]

if __name__=='__main__':
    rows=get_rows()
    answers=[]
    for row in rows:
        print(row)
        answer=input('你猜的姓在这一行中吗？(y,n,yes,no)')
        if answer is None or len(answer)==0:
            break
        if answer.lower() in ('y','yes'):
            answers.append(True)
        else:
            answer.append(False)
    print('你猜的姓是：',get_name(answers))

D:\Anaconda\python.exe D:/AI20/06_codes/deeplearning_20/p04_guess_name.py
赵 孙 周 郑 冯 褚 蒋 韩 
你猜的姓在这一行中吗？(y,n,yes,no)y
钱 孙 吴 郑 陈 褚 武 韩 
你猜的姓在这一行中吗？(y,n,yes,no)y
李 周 吴 郑 卫 蒋 武 韩 
你猜的姓在这一行中吗？(y,n,yes,no)y
王 冯 陈 褚 卫 蒋 武 韩 
你猜的姓在这一行中吗？(y,n,yes,no)y
你猜的姓是： 韩

Process finished with exit code 0