[프로그래머스]문자열 내 마음대로 정렬하기 / multimap

 

문제

풀이

#include <string>
#include <vector>
#include <algorithm>
#include <map>

using namespace std;

vector<string> solution(vector<string> strings, int n) {
    vector<string> answer;
    multimap<char ,string> hash;
    sort(strings.begin(), strings.end());
    for(int i=0; i<strings.size(); i++) hash.insert(make_pair(strings[i][n],strings[i]));
    for(auto i : hash) answer.push_back(i.second);
    return answer;
}

key가 숫자가 아닌 char이어야 했기 때문에 map을 사용했고,

그중에도 중복을 허용하고 key를 오름차순으로 자동 정렬해주는 multimap을 사용했다.

 

 

map

-중복을 허용하지 않고, key를 기준으로 오름차순 자동 정렬된다.

 

multimap

-중복을 허용하고, key를 기준으로 오름차순 자동 정렬된다.

 

unordered_map

-중복을 허용하지 않고 정렬 없이 저장한다.

 

 

 

---

다른 사람 코드

#include <string>
#include <vector>
#include <algorithm>
using namespace std;
int i;

bool compare (string a, string b) {
    return a[i] == b[i] ? a < b : a[i] < b[i];
}

vector<string> solution(vector<string> strings, int n) {
    i = n;
    sort (strings.begin(), strings.end(), compare);
    return strings;
}

sort를 이용하여 단 몇 줄만으로 문제를 해결했다!

 

 

 

sort() 3번째 인자

정렬을 어떤 식으로 할 것인지 알려주는 함수

함수는 반환 타입이 bool타입이어야 하며,  매개변수는 두 개여야 하고 두 매개변수의 타입은 정렬할 데이터의 타입과 일치해야 한다.

 

 

---

https://rshday.tistory.com/27

[c++ STL] Multimap 기본 사용법 및 예제

https://eocoding.tistory.com/6

hash_map | unordered_map | map 차이 비교, 특징 정리 - C++

 

다른 사람 코드

https://leeeegun.tistory.com/5

[C++ STL] Sort() 사용법 및 compare 함수

https://breakcoding.tistory.com/117

[C++] vector (벡터) 정렬, 배열 정렬하기