【模拟】Poj1029 False Coin

题目地址：http://poj.org/problem?id=1029

题目大意：有一堆硬币，其中一枚是假的，但是不知道它是比真的重还是轻，输入文件给出一些天平称量结果，要求判断出哪一个硬币是假币。

这个题蛋疼了一上午，老是wa，然后看了网上的思路，就a掉了。

首先，我们可以确定，假币一定是在天平不平衡的那些数据中，没有涉及到的硬币都是真的；然后，我们来考虑涉及到的可能有问题的硬币，对于这些硬币，它有两种可能：和假币在一组，不和假币在一组。因为不知道假币是轻了还是重了，上述条件就表现为：一颗硬币，如果既出现在轻的一边过，又出现在重的一边过，那么这个硬币就是没有问题的。

具体点说就是，令equal、light、heavy三个数组分别表示硬币的状态（0/1），如果equal[i]=1，那么i号硬币没有问题，else if light[i]和heavy[i]同时为1，也可确定i号硬币没有问题，如果都不满足，i号硬币就是有问题的，对每一组数据，都对这三个数组进行维护，结束后进行一次扫描，如果只有一个有问题的，则输出有问题的，有多个或没有则输出0.

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>

using namespace std;

int n,k,equ[1050],lig[1050],hea[1050],flag[1050],tmp[1050];

int main(){
    scanf("%d%d",&n,&k);

    memset(equ,0,sizeof(equ));
    memset(lig,0,sizeof(lig));
    memset(hea,0,sizeof(hea));
    memset(tmp,0,sizeof(tmp));


    for (int i=1;i<=k;i++){
        int x;
		char ch;
		scanf("%d",&x);
		for(int j=0;j<2*x;j++) scanf("%d",tmp+j);
		ch=getchar();
		while(ch!='=' && ch!='<' && ch!='>'){
                    ch=getchar();
		}

                if(ch=='<')
		{
			for(int j=0;j<x;j++)
				lig[tmp[j]]=1;
			for(int j=x;j<2*x;j++)
				hea[tmp[j]]=1;
			for(int j=1;j<=n;j++)
			{
				bool flag=true;
				for(int k=0;k<2*x;k++)
					if(j==tmp[k])
					{
						flag=false;
						break;
					}
				if(flag)
					equ[j]=1;
			}
		}
		else if(ch=='=')
		{
			for(int j=0;j<2*x;j++)
				equ[tmp[j]]=1;
		}
		else if(ch=='>')
		{
			for(int j=0;j<x;j++)
				hea[tmp[j]]=1;
			for(int j=x;j<2*x;j++)
				lig[tmp[j]]=1;
			for(int j=1;j<=n;j++)
			{
				bool flag=true;
				for(int k=0;k<2*x;k++)
					if(j==tmp[k])
					{
						flag=false;
						break;
					}
				if(flag)
					equ[j]=1;
			}
		}

    }


        int ans,count=0;
	for(int i=1;i<=n;i++)
	{
		if(equ[i])
			continue;
		if(lig[i]&&hea[i])
			continue;
		ans=i;
		count++;
	}
	if(count>1)
		printf("0\n");
	else
		printf("%d\n", ans);

    //cout<<endl;
    //for (int i=1;i<=n;i++){cout<<i<<": "<<equ[i]<<","<<lig[i]<<","<<hea[i]<<endl;}

    return 0;
}