[Python][PAT]1089 狼人杀-简单版 （20 分）

1089 狼人杀-简单版 （20 分）

以下文字摘自《灵机一动·好玩的数学》：“狼人杀”游戏分为狼人、好人两大阵营。在一局“狼人杀”游戏中，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

思路：

先指定两个狼人，  然后循环遍历所有人说的话，，，如果当前遍历的这个人说的狼人 不在我指定的两只狼人中，那么这个人必然说了谎，同理，如果这个人说的好人 是我指定的两只狼人的其中之一，那么此人也必然说了谎，

代码：

def main():
    n = input()
    n = int(n)
    s = "0"
    for i in range(n):
        temp = input()
        s = s + " " + temp
    s = s.split()
    s = list(map(int, s))
    for k in range(1, n):
        for j in range(k+1, n+1):
            lr = hr = 0
            for i in range(1, n+1):
                if (s[i]<0 and abs(s[i])!=k and abs(s[i])!=j) or (s[i]>0 and (s[i]==k or s[i]==j)):
                    if i == k or i == j:
                        lr += 1
                        if lr > 1:
                            break
                    else:
                        hr += 1
                        if hr > 1:
                            break
            else:
                if lr == hr and lr == 1:
                    print(k, j)
                    return

    print("No Solution")


main()