PAT 顶级（Python） #1022 Werewolf (35 分)

最新推荐文章于 2023-05-24 19:25:01 发布

敲代码的小风

最新推荐文章于 2023-05-24 19:25:01 发布

阅读量426

点赞数 1

分类专栏： Python基础实验 PAT乙级考试（Python） PAT顶级考试（Python）

本文链接：https://blog.csdn.net/m0_46653437/article/details/118254411

版权

Python基础实验同时被 3 个专栏收录

232 篇文章 12 订阅

订阅专栏

PAT乙级考试（Python）

61 篇文章 12 订阅

订阅专栏

PAT顶级考试（Python）

1 篇文章 0 订阅

订阅专栏

#1022 Werewolf (35 分)

在这里插入图片描述

代码展示:

N, M, L = list(map(eval,input().split()))
statements = [None]
for i in range(N):
    statements.append(input())
no_solution = True

# print('N =', N,'\tM =', M, '\tL =',L)
# print('statements = ', statements)

def check(wolves_candidate):
    is_wolvs = ['+'] * (N+1)
    for i in wolves_candidate:
        is_wolvs[i] = '-'
    num_wolves_liars = count_total_liars(wolves_candidate,is_wolvs)
    num_total_liars = count_total_liars(list(range(1,N+1)),is_wolvs)
    if 1 <= num_wolves_liars < M and num_total_liars == L:
        global no_solution
        no_solution = False
        return True
    else:
        return False
    
def count_single_liar(person,is_wolvs):
    person_state = statements[person]
    if is_wolvs[int(person_state[1:])] != person_state[0]:
        return 1
    else:
        return 0
        
def count_total_liars(persons,is_wolvs):
    total = 0
    for person in persons:
        total += count_single_liar(person,is_wolvs)
    return total

def DFS(prior,ResN,ChosM):
    if not no_solution:
        return  
    if ResN < ChosM:
        return
    if ChosM < 1:
        result = prior
        # print(result)
        if check(result):
            result = list(map(str,result))
            print(' '.join(result))
        return
    for i in range(ResN,0,-1):
        DFS(prior+[i],i-1,ChosM-1)
 
def main():
    DFS([],N,M)
    global no_solution
    if no_solution:
        print('No Solution')
        
if __name__ == '__main__':
    main()