字节跳动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>
将属于 b 的 a 存入一个set 中,目的是去重。order对应的是a在vector中的行数。

C++实现

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

int N;
string p1,p2;
    
int main()
{
    map model;
    vector> vec;
    int order = 0;
    
    cin >> N;
    while(N--)
    {
        cin >> p1 >> p2;
        if(model.find(p2)==model.end())
        {
            model[p2] = order;
            vec.push_back(set{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;
}

总结

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

你可能感兴趣的:(刷题吧,c++,字节跳动,leetcode)