题意:有N个硬币,编号1-N,其中有一个是假的,与真的硬币重量不同。给出K组称量数据(天平左边的硬币编号,天平右边的编号以及称量结果),问能否确定那个硬币是假的。
分析:天平平衡时的硬币全是真的,用数组标记一下。不平衡时,一定有一个是假的,有可能是一边某个轻了,也有可能是某个重了,但没出现在这组中的硬币必然是真的。而且假硬币的出现次数和天平不平衡的次数相等。
Code:
#include <algorithm>
#include <iostream>
#include <cstring>
#include <string>
#include <cstdio>
#include <vector>
#include <queue>
#include <cmath>
#include <map>
#include <set>
#define LL long long
#define pb push_back
#define Max(a,b) ((a)>(b)?(a):(b))
#define Min(a,b) ((a)<(b)?(a):(b))
using namespace std;
const int inf=0x3f3f3f3f;
const int maxn=1005;
int time[maxn],lf[maxn],rt[maxn];
bool ext[maxn],sure[maxn];
char s[5];
int n,m,t;
int main()
{
scanf("%d %d",&n,&m);
int tot=0;
while(m--){
scanf("%d",&t);
for(int i=0;i<t;i++) scanf("%d",&lf[i]);
for(int i=0;i<t;i++) scanf("%d",&rt[i]);
scanf("%s",s);
if(s[0]=='='){//这组中的硬币一定是真的
for(int i=0;i<t;i++) sure[lf[i]]=true,sure[rt[i]]=true;
}
else if(s[0]=='<'){
for(int i=1;i<=n;i++){//没在这组数据中出现的硬币一定是真的
if(sure[i]) continue;
bool flag=true;
for(int j=0;j<t;j++){
if(lf[j]==i||rt[j]==i) {flag=false;break;}
}
if(flag) sure[i]=true;
}
tot++;
for(int i=0;i<t;i++){
time[lf[i]]--;
time[rt[i]]++;
}
}
else {
for(int i=1;i<=n;i++){//没在这组数据中出现的硬币一定是真的
if(sure[i]) continue;
bool flag=true;
for(int j=0;j<t;j++){
if(lf[j]==i||rt[j]==i) {flag=false;break;}
}
if(flag) sure[i]=true;
}
tot++;
for(int i=0;i<t;i++){
time[lf[i]]++;
time[rt[i]]--;
}
}
}
int cnt=0,id=0;
for(int i=1;i<=n;i++){
if(sure[i]) continue;
if(time[i]==tot||time[i]==-tot) cnt++,id=i;
}
if(cnt==1) printf("%d\n",id);
else printf("0\n");
return 0;
}