정렬된 리스트에 원소 삽입

Gino·2022년 8월 2일
1

문제출처 : 프로그래머스

정렬된 리스트에 원소 삽입

리스트 L과 정수 x가 주어질 때, 리스트 내 올바른 위치에 x를 삽입하시오.

# input
L= [20,37,58,72,91]
x = 65 

# output = [20,37,58,65,72,91]

내가 푼 코드

def solution(L, x):
    
    if L[-1] < x: 
        L.append(x)
    else:
        for i in range(len(L)):
            if L[i] >= x:
                L.insert(i, x)
                break
            
    answer = L
    return answer
solution(L, x)
[20, 37, 58, 65, 72, 91]

다른 사람 코드

def solutions(L, x):
    
    for idx, num in enumerate(L):
        if num > x:
            L.insert(idx, x)
            break
        
        if L[-1] < x:
            L.append(x)
        else:
            pass 
    
    return L
profile
나무를 심는 사람

0개의 댓글