PAT 1089 狼人杀-简单版

时间:2023-02-10 15:15:09

PAT 1089 狼人杀-简单版

思路:利用二重循环for i=1:n-1, for j=i+1:n假设第i,j是狼人,记号temp中的值为-1,其余人为1表示好人。首先判断这两个狼人中有且只有一个撒谎,
如何判断一个人撒谎:vi[k]*temp[abs(vi[k])]<0。然后再对其他人进行判断是否有其他一个人说谎,只能是一个。题目中说了一个狼人说谎,一个好
人说谎。

#include<iostream>
#include<vector>
#include<math.h>
using namespace std;
int main(){
    int n, l1=0, l2=0;
    cin>>n;
    vector<int> vi(n+1);
    for(int i=1; i<=n; i++)
        cin>>vi[i];
    for(int i=1; i<=n-1; i++)
        for(int j=i+1; j<=n; j++){
            vector<int> temp(n+1, 1);
            int flag=0;
            temp[i]=temp[j]=-1;
            if(temp[abs(vi[i])]*vi[i]*temp[abs(vi[j])]*vi[j]<0){ \\只有一个狼人说谎
                for(int k=1; k<=n; k++)
                    if(k==i||k==j)
                        continue;
                    else if(vi[k]*temp[abs(vi[k])]<0){
                        flag++;
                        if(flag>1)
                            break;
                    } 
            }   
            if(flag==1){
                l1=i; l2=j;
                cout<<l1<<" "<<l2<<endl;
                return 0;
            }
        }
        cout<<"No Solution"<<endl;
        return 0;
}