POJ - 1029 False Coin模拟

题目链接
给你n个硬币，硬币的编号从1到n，n个硬币中有一个硬币是假的，他的重量与真硬币的质量不一样，或重或轻，真硬币的重量都相同。给你k组测量结果，判断是否能找出假硬币。每组结果分为2行，第一行第一个数字代表秤盘左右硬币的个数m，前m个表示左秤盘的硬币编号，后m个表示右秤盘的硬币数量，第二行是测量结果（= < >）三者之一。判断k组结果能否找出假硬币，能输出硬币的编号，否则输出0。

拿到该题目第一反应不知道如何下手，需要综合每一组测量的结果才能进行判断，所以我们需要把每一步的结果进行分析并保持。

1. 结果为等号两边的硬币一定是真硬币，我们做上true的标记
2. 结果为不等号的硬币一定含有假硬币，我们将轻硬币在left数组标记，将重硬币在right数组标记
3. 因为只有一个假硬币，硬币不会一会儿轻，一会重。由于不等号出现的结果必定是应为假硬币的出现，所以硬币在left/right数组出现的次数与不等号结果的次数相同则为假硬币。
4. 如果满足以上的结果只有一组就给予输出，否则输出0

#include <cstdio>
#include <cstring>
#include <iostream>
#include <vector>
using namespace std;
const int MAXN = 1e3 + 5;
int a[MAXN], b[MAXN], l[MAXN], r[MAXN];
char s[MAXN];
bool T[MAXN];
vector<int> v;
int main()
{
    int n, k;
    scanf("%d%d", &n, &k);
    int cnt = 0;
    for (int j = 0; j < k; ++j)
    {
        int m;
        scanf("%d", &m);// 秤盘测量左右硬币的个数
        for (int i = 0; i < m; ++i)
        {
            scanf("%d", &a[i]);
        }
        for (int i = 0; i < m; ++i)
        {
            scanf("%d", &b[i]);
        }
        getchar();// 吸收缓冲区scanf残余的回车字符
        gets(s);
        if (strcmp(s, "<") == 0)
        {
            cnt++;// 不等号出现次数加1
            // 标记硬币在l/r出现的次数
            for (int i = 0; i < m; ++i)
            {
                l[a[i]]++;
            }
            for (int i = 0; i < m; ++i)
            {
                r[b[i]]++;
            }
        }
        else if (strcmp(s, "=") == 0)
        {
            for (int i = 0; i < m; ++i)
            {
                T[a[i]] = true;
            }
            for (int i = 0; i < m; ++i)
            {
                T[b[i]] = true;
            }
        }
        else if (strcmp(s, ">") == 0)
        {
            cnt++;
            for (int i = 0; i < m; ++i)
            {
                r[a[i]]++;
            }
            for (int i = 0; i < m; ++i)
            {
                l[b[i]]++;
            }
        }
    }

    for (int i = 1; i <= n; ++i)
    {
        // 非真硬币，且出现次数与不等号次数一致一定为假硬币
        if (!T[i] && (l[i] == cnt || r[i] == cnt))
        {
           v.push_back(i);
        }
    }

    // 判断假硬币的可能结果
    if (v.size() == 1)
    {
        printf("%d", v[0]);
    }
    else
    {
        printf("0");
    }
    return 0;
}