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

(주)그렙

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

(C++, Python)

(글쓴날 : 2020.06.12)

 

---

* programmers, 프로그래머스 문제 C++, Python 언어 풀이입니다.

* 소스 코드의 저작권은 글쓴이에게 있습니다.

---

 

 

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

---

1) 문제

문제 링크 : https://programmers.co.kr/learn/courses/30/lessons/12915

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

---

2) 풀이 과정

* 시간 복잡도 : O(n log n)

 

문자열로 구성된 리스트와 정수 n이 주어질 때, n번째 인덱스를 기준으로 해당 문자열들을 오름차순으로 정렬하는 문제입니다.

단, n번째 인덱스의 값이 같을 경우 사전 순으로 정렬합니다.

 

저의 경우, C++와 Python을 사용했습니다.

C++의 경우 사용자 정의 compare 함수를 만들어 만약 n번째 인덱스가 같을 시 문자열 전체를 기준으로 오름차순, 다를 시 n번째 인덱스를 기준으로 오름차순 정렬하였고, Python의 경우 sorted 함수 key의 첫 번째 조건으로 n번째 인덱스, 두 번째 조건으로 문자열 전체를 기준으로 하는 lambda 함수를 적용하여 문제를 해결했습니다.

---

3) 코드

 

* C++ 코드

#include <string>
#include <vector>
#include <algorithm>
 
using namespace std;
 
int i;
 
bool compare(const string &a, const string &b)
{
    if (a[i] == b[i])
        return a < b;
    else
        return a[i] < b[i];
}
 
vector<string> solution(vector<string> strings, int n)
{
    i = n;
 
    sort(strings.begin(), strings.end(), compare);
 
    return strings;
}

---

* Python 코드

def solution(strings, n):
    return sorted(strings, key=lambda x: (x[n], x))

---