算法竞赛入门经典 第二版 习题5-4 交换学生 Foreign Exchange uva10763

题目:https://vjudge.net/problem/UVA-10763

思路:用学生的意向校映射来源校,用vector来保存每个意向校的学生的来源校,然后每次读取数据时查找是否有对应学校的学生想要交换,若有,则删去查找到的结果;若没有,将这条输入数据保存到映射中。最后判断映射是否为空就得出交换项目能否进行。
注:当映射的某个关键字对应的vector为空时,别忘把这个关键字删除。

代码:

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;

int main()
{
    map<int, vector<int> > student;
    int n;
    while(cin>>n)
    {
        if(n==0)
        {
            break;
        }
        student.clear();
        while(n--)
        {
            int from, to;
            cin >> from >> to;
            vector<int>::iterator it = find(student[to].begin(), student[to].end(), from);
            if(it==student[to].end())
            {
                student[from].push_back(to);
            }
            else
            {
                student[to].erase(it);
            }
            if(student[to].empty())
            {
                student.erase(to);
            }
        }
        if(student.empty())
        {
            printf("YES\n");
        }
        else
        {
            printf("NO\n");
        }
    }
    return 0;
}

你可能感兴趣的:(解题笔记,算法,uva,c++)