PAT 1089 狼人殺-簡單版

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;
}