字节跳动2021实习笔试题第一题

题目

一行数据有两个字符串a和b，共有N行，一个b可能对应多个a，输出所有的b和属于b的a。
例如：
输入：
3
a txt1
b txt1
c txt2
输出：
txt1 a b
txt2 c

思路

1. multimap

key：b，value：a
这样就把所有属于b的a都存入了multimap[b]中，再输出就行。但是该题有个坑（题目没告知）：输入的数据中有重复的，即存在多行的数据是一样的，那么就要考虑去重的问题。

2. map+vector

map: key: b, value: order ;vector<set<string>>
将属于 b 的 a 存入一个set 中，目的是去重。order对应的是a在vector中的行数。

C++实现

#include<iostream>
#include<map>
#include<string>
#include<vector>
#include<set>
using namespace std;

int N;
string p1,p2;
    
int main()
{
    map<string,int> model;
    vector<set<string>> vec;
    int order = 0;
    
    cin >> N;
    while(N--)
    {
        cin >> p1 >> p2;
        if(model.find(p2)==model.end())
        {
            model[p2] = order;
            vec.push_back(set<string>{p1}); 
            order++;
        }
        else
            vec[model[p2]].insert(p1);
    }
    for(auto i:model)
    {
        cout << i.first << ' ';
        for(auto s:vec[i.second])
            cout << s << ' ';
        cout << endl;
    }

    return 0;
}

总结

太菜了，先不写了，赶紧刷题去。。。