이쿠의 슬기로운 개발생활

함께 성장하기 위한 보안 개발자 EverNote 내용 공유

코딩테스트

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

이쿠우우 2022. 3. 19. 17:13
반응형

 

https://programmers.co.kr/learn/courses/30/lessons/12915

 

코딩테스트 연습 - 문자열 내 마음대로 정렬하기

문자열로 구성된 리스트 strings와, 정수 n이 주어졌을 때, 각 문자열의 인덱스 n번째 글자를 기준으로 오름차순 정렬하려 합니다. 예를 들어 strings가 ["sun", "bed", "car"]이고 n이 1이면 각 단어의 인덱

programmers.co.kr

 

글쓴이의 답

개인적인 풀이 임으로

이것보다 더 좋은 알고리즘은 많음...

이렇게도 풀이하는구나.. 공유하기 위해 올림...

#include <string>
#include <vector>
#include <map>
#include <set>
#include <iostream>

using namespace std;

vector<string> solution(vector<string> strings, int n) {
    vector<string> answer;
    map<char, set<string>> result;   
    
    for(size_t i=0; i< strings.size(); i++){
        string temp = strings[i];        
        result[temp[n]].insert(temp);
    }
    
    for(auto it = result.begin(); it != result.end(); it++){
        set<string> setTemp = it->second;
        vector<string> temp(setTemp.begin(), setTemp.end());
        answer.insert(answer.end(), temp.begin(), temp.end());
    }
   
    return answer;
}

 

꾸준히 하다보면 실력이 늘겠지..

반응형