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

코딩테스트연습

• programmers.co.kr
• 문자열 내마음대로 정렬하기

 

---

 

def solution(strings, n):
    for idx in range(len(strings)):
        strings[idx] = strings[idx][n]+strings[idx]

    strings.sort()

    for idx in range(len(strings)):
        strings[idx] = strings[idx][1:]

    return strings

 

---

 

sorted를 사용해서 key를 지정하면 한줄도 가능하더라.

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