UVA - 10763 Foreign Exchange (题解)

题目
Your non-profit organization (iCORE - international Confederation of Revolver Enthusiasts) coordinates a very successful foreign student exchange program. Over the last few years, demand has sky-rocketed and now you need assistance with your task. The program your organization runs works as follows: All candidates are asked for their original location and the location they would like to go to. The program works out only if every student has a suitable exchange partner. In other words, if a student wants to go from A to B, there must be another student who wants to go from B to A. This was an easy task when there were only about 50 candidates, however now there are up to 500000 candidates

Input
The input file contains multiple cases. Each test case will consist of a line containing n – the number of candidates (1 ≤ n ≤ 500000), followed by n lines representing the exchange information for each candidate. Each of these lines will contain 2 integers, separated by a single space, representing the candidate’s original location and the candidate’s target location respectively. Locations will be represented by nonnegative integer numbers. You may assume that no candidate will have his or her original location being the same as his or her target location as this would fall into the domestic exchange program. The input is terminated by a case where n = 0; this case should not be processed.

Output
For each test case, print ‘YES’ on a single line if there is a way for the exchange program to work out, otherwise print ‘NO’.

代码

#include 
#include 
#include 
#include 
using namespace std;
typedef vector<int> vi;
map<int, vi> dic;
void add_ele(int a, int b)
{
    if(dic.count(a) == 0)
        dic[a] = vi();
    dic[a].push_back(b);
}
int main()
{
    // freopen("i.txt", "r", stdin);
    // freopen("o.txt", "w", stdout);
    int n;
    while(scanf("%d", &n), n)
    {
        dic.clear();
        int a, b, ok = 1;
        while(n--)
        {
            scanf("%d%d", &a, &b);
            if(dic.count(b) == 0) add_ele(a, b);
            else
            {
                int sign = 1;
                vi::iterator it;
                for(it = dic[b].begin(); it != dic[b].end() && sign; it++)
                {
                    if(*it == a)
                    { dic[b].erase(it); sign = 0; }
                }
                if(sign) add_ele(a, b); 
            }
        }
        for(auto i : dic)
        {
            if(i.second.size() > 0)
            { ok = 0; break; }
        }
        printf("%s\n", ok ? "YES" : "NO");
    }
    return 0;
}