PAT 1090 危险品装箱 (STL 难度2) - 详细题解

学习了一下各种STL的使用

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <string>
#include <stdlib.h>
#include <vector>
#include <queue>
#include <cmath>
#include <map>
using namespace std;

#define ms(x, n) memset(x,n,sizeof(x));
typedef  long long LL;
const LL maxn = 1e5;

int n, m, k;
map<int, vector<int>> warn; //切记, 一个货物可能有多个不相容对
vector<int> goods;
bool used[maxn];
int main()
{
    cin >> n >> m;
    int id1, id2;
    for(int i = 1; i <= n; i++){
        cin >> id1 >> id2;
        warn[id1].push_back(id2); //插入不相容对
        warn[id2].push_back(id1);
    }

    while (m--){
        bool flag = true;
        int input;
        ms(used, 0);
        goods.clear();

        cin >> k;
        for(int i = 0; i < k; i++){
            cin >> input; goods.push_back(input);
            used[input] = true;
        }


        for(int i = 0; i < goods.size(); i++){
            for(int j = 0; j < warn[goods[i]].size(); j++)
                if(used[warn[goods[i]][j]]){ //分别查找该号对应的不相容对是否使用了
                    flag = false;
                    i = 1<<30; break;
                }
        }
        if(flag)
            cout << "Yes" << endl;
        else
            cout << "No" << endl;
    }

    return 0;
}