题目:

以下文字摘自《灵机一动·好玩的数学》:“狼人杀”游戏分为狼人、好人两大阵营。在一局“狼人杀”游戏中,1 号玩家说:“2 号是狼人”,2 号玩家说:“3 号是好人”,3 号玩家说:“4 号是狼人”,4 号玩家说:“5 号是好人”,5 号玩家说:“4 号是好人”。已知这 5 名玩家中有 2 人扮演狼人角色,有 2 人说的不是实话,有狼人撒谎但并不是所有狼人都在撒谎。扮演狼人角色的是哪两号玩家?
本题是这个问题的升级版:已知 N 名玩家中有 2 人扮演狼人角色,有 2 人说的不是实话,有狼人撒谎但并不是所有狼人都在撒谎。要求你找出扮演狼人角色的是哪几号玩家?
输入格式:
输入在第一行中给出一个正整数 N(5≤N≤100)。随后 N 行,第 i 行给出第 i 号玩家说的话(1≤i≤N),即一个玩家编号,用正号表示好人,负号表示狼人。
输出格式:
如果有解,在一行中按递增顺序输出 2 个狼人的编号,其间以空格分隔,行首尾不得有多余空格。如果解不唯一,则输出最小序列解 —— 即对于两个序列 A=a[1],…,a[M] 和 B=b[1],…,b[M],若存在 0≤k<M 使得 a[i]=b[i] (i≤k),且 a[k+1]<b[k+1],则称序列 A 小于序列 B。若无解则输出 No Solution。
输入样例 1:
5
-2
+3
-4
+5
+4
输出样例 1:
1 4
输入样例 2:
6
+6
+3
+1
-5
-2
+4
输出样例 2(解不唯一):
1 5
输入样例 3:
5
-2
-3
-4
-5
-1
输出样例 3:
No Solution

思路:题目确实有点难理解,思路也比较难想出来,算是20分题目中比较难的一道了,花了很多时间理解了。用三个循环进行遍历,第一个和第二个的 i 和 j 假设为俩个狼,然后第三个循环来判断说话的人是否说谎,说谎的条件就俩种,仔细看代码,最后得到符合要求的 i 和 j ,因为 i 是从小遍历的,所以直接输出就是答案。

def main():
    N = int(input())
    re = [0]
    for _ in range(N):
        tmp = input()
        re.append(tmp)
    re = list(map(int, re))
    # re数据形式为[0,-2,3,-4,5,4]
    for i in range(1, N):  # 假设i为狼
        for j in range(i + 1, N + 1):  # 假设j为狼
            lr = hr = 0  # 狼人数量和好人数量
            for k in range(1, N + 1):
                # 判断k个人是否说谎
                if (re[k] < 0 and abs(re[k]) != i and abs(re[k]) != j) \
                        or (re[k] > 0 and (re[k] == i or re[k] == j)):
                    if k == i or k == j:  # 狼人说谎
                        lr += 1
                        if lr > 1:
                            break
                    else:  # 好人说谎
                        hr += 1
                        if hr > 1:
                            break
            else:
                # 如果满足条件,输出结束程序
                if lr == hr and lr == 1:
                    print(i, j)
                    return
    print("No Solution")


main()