본문 바로가기

알고리즘

99클럽 코테 스터디 3일차 TIL

문제 설명



풀이 과정 & 코드

  • 해당 문제는 파이썬의 sort() or sorted()를 사용하면 쉽게 해결할 수 있는 문제임
  • 처음에는 sort()하나를 생략해서 틀리게 나왔는데 문제를 제대로 다시 읽어보니 주어진 n으로 정리했는데 순서가 같다면 알파벳 순이다 라는 말이 있었음
  • 그래서 sort()를 하나 더 추가해서 알파벳순으로 정렬을 하고 이후에 lambda를 사용해서 n의 순서로 정렬을 하게 되면 문제를 쉽게 풀 수 있음
def solution(strings, n):
    strings.sort()
    strings.sort(key=lambda x: x[n])
    return strings