본문 바로가기
알고리즘이야기/프로그래머스_알고리즘_lv1

문자열 내 마음대로 정렬하기

by 효우너 2020. 5. 11.
728x90
반응형
문제 출처 : https://programmers.co.kr/learn/courses/30/lessons/12915

 

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

문자열로 구성된 리스트 strings와, 정수 n이 주어졌을 때, 각 문자열의 인덱스 n번째 글자를 기준으로 오름차순 정렬하려 합니다. 예를 들어 strings가 [sun, bed, car]이고 n이 1이면 각 단어의 인덱스 1�

programmers.co.kr


처음에 for문을 너무 복잡하게 구현해서 헷갈렸는데 생각을 바꿔보니

정렬해야하는 해당 문자열을 strings[0]자리 앞에 붙인 후 정렬을 하고 다시 없애주면 되는 문제였다.

def solution(strings, n):
    answer=[]
    result=[]
    strings.sort()
    for i in range(len(strings)):
        s=strings[i][n]+strings[i] //정렬해야하는 문자를 맨 앞에 붙여주기
        answer.append(s)
    answer.sort()
    for i in range(len(answer)):
        a=answer[i][1:]
        result.append(a)
    return result

 

728x90
반응형

댓글