알고리즘 문제풀이/프로그래머스
[프로그래머스] 문자열 내 마음대로 정렬하기 [연습문제] [python]
coding of lks
2020. 9. 19. 18:28
[프로그래머스] 문자열 내 마음대로 정렬하기 [연습문제] [python] Level1
문제 설명
문자열로 구성된 리스트 strings와, 정수 n이 주어졌을 때, 각 문자열의 인덱스 n번째 글자를 기준으로 오름차순 정렬하려 합니다. 예를 들어 strings가 [sun, bed, car]이고 n이 1이면 각 단어의 인덱스 1의 문자 u, e, a로 strings를 정렬합니다.
입출력 예 설명
입출력 예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] 입니다.
설계 및 구현
my_solution1 실패
def solution(strings, n):
return sorted(strings, key=lambda x: x[n])
sorted()를 이용해 n번째 문자를 기준으로 정렬을 하였다.
하지만 n번째 문자가 같을 경우 사전순으로 정렬해야하는 부분이 구현이 안된 solution이다
my_solution2 성공
def solution(strings, n):
return sorted(sorted(strings), key=lambda x: x[n])
간단하게 sorted()의 매개 변수에 사전순으로 정렬된 strings를 넣어주면 해결이 되었다.
간단한 해결법이었지만 결국 질문하기에 있는 팁을 보고난후 구현 할 수 있었다.!!
반응형