유자차의 재테크 공부방

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

파이썬/알고리즘 문제 풀이

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

유자차H 2022. 3. 18. 20:47
반응형

알면 좋은 방법

# 첫번째 기준으로 정렬 후 두번째 기준으로 정렬
sorted([문자열 리스트], key=lambda x : ([첫번째 기준], [두번째 기준]))

 

 

풀이방법

def solution(strings, n):
	# [n번째 문자, 문자열]을 정렬
	res = sorted([i[n], i] for i in strings)
    ans=[i[1] for i in res] # 문자열만 뽑기
    return ans
    
    
def solution(strings, n):
	# n번째 문자열 기준으로 정렬 -> 문자열로 정렬
    return sorted(strings, key = lambda x : (x[n],x))
반응형
Comments