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

 

 

 

• 문자열 내 마음대로 정렬하기

문제 설명

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

제한 조건

• strings는 길이 1 이상, 50이하인 배열입니다.
• strings의 원소는 소문자 알파벳으로 이루어져 있습니다.
• strings의 원소는 길이 1 이상, 100이하인 문자열입니다.
• 모든 strings의 원소의 길이는 n보다 큽니다.
• 인덱스 1의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.

입출력 예

stringsnreturn

["sun", "bed", "car"]	1	["car", "bed", "sun"]
["abce", "abcd", "cdx"]	2	["abcd", "abce", "cdx"]

입출력 예 설명

입출력 예 1
"sun", "bed", "car"의 1번째 인덱스 값은 각각 "u", "e", "a" 입니다. 이를 기준으로 strings를 정렬하면 ["car", "bed", "sun"] 입니다.

입출력 예 2
"abce"와 "abcd", "cdx"의 2번째 인덱스 값은 "c", "c", "x"입니다. 따라서 정렬 후에는 "cdx"가 가장 뒤에 위치합니다. "abce"와 "abcd"는 사전순으로 정렬하면 "abcd"가 우선하므로, 답은 ["abcd", "abce", "cdx"] 입니다.

 

 

 

나의 코드 

처음에는 lambda x : x[n] 으로 갔다가 한 번 더 sort가 필요해서 고민하다가 블로그 선배님들의 도움으로 힌트를 얻어서 풀었다 ! 

def solution(strings, n):
    new = []
    for i in strings:
        new.append(i[n]+i)
    new.sort()
    answer =[]
    for i in new:
        answer.append(i[1:])
    return answer

 

풀고 다른 사람들의 코드를 보니 lambda 조건도 두 개를 걸 수 있었다 ! 오우 신기해 

 

 

다른 분의 코드

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

x: 요 뒤에 () 튜플 형식으로 조건 2가지를 달 수 있고 오름차순이 기본 ! 내림차순이라면 - 마이너스를 붙이면 된다. 

 

 

 

lambda 표현으로 sort 두번 하기 ! 

2021.07.05 추가 

import sys
input = sys.stdin.readline
# from collections import deque
n = int(input())

a = []
for i in range(n):
    x,y = map(int,input().split())
    a.append([x,y])
b= a.copy()
c = a.copy()
print(a)
a = sorted(a, key = lambda x :x[0])
print(a)
a = sorted(a, key = lambda x: x[1])
print(a)


print(b)
b = sorted(b, key = lambda x :x[1])
print(b)
b = sorted(b, key = lambda x: x[0])
print(b)


c.sort(key = lambda x: (x[1], x[0]))
print(c)

* 주의점 c.sort(key = lambda x : (x[1], x[0]))  ---> x[0] 으로 먼저 sort 하고 두 번째로 x[1]로 sort 한다는 뜻 ! 

x[1]이 앞에 있어서 먼저 실행될 줄 알았는데 아래 분석(?) 을 해보니 아니었다. ** 주의하기 !