[메모] python 정렬 + 관련 문제

참고 - https://wayhome25.github.io/python/2017/03/07/key-function/

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

key값을 기준으로 정렬할 수 있음을 이용!
람다식을 이용해서 n번째 알파벳을 기준으로 정렬하도록 한다.

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

만약 정렬하지 않은 strings를 사용하면

해당 테스트케이스에서 ["abce","abcd","cdx"]순으로 정렬되니 주의..

+추가

def solution(strings, n):
    answer = []
    tmp = []
    for s in strings:
        tmp.append(s[n])

    tmp.sort()
    strings.sort()
    for t in tmp:
        for s in strings:
            if s[n]==t:
                answer.append(s)
                strings.remove(s)
                break
    return answer

sort(key=)를 알기 전에는.. 이렇게 했다

좋은 웹페이지 즐겨찾기