华为OD机试-寻找同班小朋友

// Online C++ compiler to run C++ program online
#include<iostream>
#include<vector>
#include<stdlib.h>
#include<algorithm>
#include<iomanip>
#include<string.h>
#include<exception> 
#include<map>
#include<cmath>
#include<unordered_map>
#include<set>
#include<climits>
#include<ctype.h>
#include<queue>
#include<stack>
#include<string>
using namespace std;

vector<string> split(string str, char ch) {
    vector<string> list;
    str += ch;
    while (str.find(ch) != string::npos)
    {
        int idx = str.find(ch);
        string token = str.substr(0, idx);
        list.push_back(token);
        str = str.substr(idx + 1);
    }

    return list;
}

int main() {
    string str;
    getline(cin, str);

    set<int> G[2];

    int id = 0;
    int idx = 0;
    vector<string> list = split(str, ' ');
    for (int i = 0; i < list.size(); ++i)
    {
        vector<string> m = split(list[i], '/');

        id = atoi(m[0].c_str());
        if (i == 0)
        {
            G[idx].insert(id);
            continue;
        }

        if (m[1] == "N")
        {
            idx = (idx + 1) % 2;
        }

        G[idx].insert(id);
    }

    for (auto x : G[0])
    {
        cout << x << " ";
    }

    cout << endl;

    for (auto x : G[1])
    {
        cout << x << " ";
    }
 
    return 0;
}