POJ 1029 False coin 找假币 模拟

题目：有N个硬币，里面有一个硬币是坏的，现在称了M次（称两边放数量一样的硬币）并把结果作为输入，根据称的结果判断第几号硬币是坏的。

例如：输入如下
5 3                  （5表示有5个硬币，3表示称了3次，下面依次是每次称的结果）
2 1 2 3 4         （2表示称的两边分别放两个硬币，分别是1、2和3、4）
<                     （1+2<3+4）
1 1 4               （1表示称的两边分别放一个硬币，分别是1和4）
=                     （1=4）
1 2 5               （1表示称的两边分别放一个硬币，分别是2和5）
=                     （2=5）

思路：

当两边一样重的时候，说明称量的硬币都是真的，对于这些硬币，我们直接做记号说明是真硬币。

当不一样重的时候，说明天平上肯定存在假硬币，但是我们不知道假硬币是在轻的一端还是重的一端。接下来这点很重要，天平不平衡了几次，假硬币肯定也出现了几次。所以对于不平衡时的称量，我们将称量的硬币出现的次数+1，注意，轻的一端和重的一端需要分开计数。最后我们比较天平不平衡的次数和每个硬币出现的次数，如果只有一个硬币称量的次数等于天平不平衡次数，说明该硬币就是假硬币。有多个时就无法判断出来。

#include<iostream>
#include<algorithm>
#include<string>
#include<cstring>
#include<cstdio>
const int maxn=100005;
using namespace std;
int flag[maxn]; //记录硬币真假,1为真
int t[maxn]; //记录硬币原始数据
int w[maxn]; //记录硬币出现的次数，轻的和重的要分开记录，为正表示重的出现次数，负表示轻的出现次数
int main()
{
// freopen("E:\\ACM\\test.txt","r",stdin);
int N,K,p;
int total=0; //记录不等式出现出现次数
char c;
cin>>N>>K;
while(K--)
{
cin>>p;
for(int i=0;i<2*p;i++) cin>>t[i];
cin>>c;

if(c=='=') //为真
{
for(int i=0;i<2*p;i++) flag[t[i]]=1;
}
else if(c=='<') //左轻右重
{
++total;
for(int i=0;i<p;i++) w[t[i]]--;
for(int i=p;i<2*p;i++) w[t[i]]++;
}
else if(c=='>') //左重右轻
{
++total;
for(int i=0;i<p;i++) w[t[i]]++;
for(int i=p;i<2*p;i++) w[t[i]]--;
}

}
int count=0,pos=0;
for(int i=1;i<=N;i++)
{
if(flag[i]==0&&(w[i]==total||w[i]==-total)) //找每次都出现的假币
{
++count;
pos=i;
}
}
if(count!=1) puts("0");
else cout<<pos<<endl;

return 0;
}