[프로그래머스] 체육복 - python

코드

처음 코드

def solution(n, lost, reserve):
    lst = sorted(lost)
    reserve.sort()
    for i in lst1:
        if i-1 in reserve:
            reserve.remove(i-1)
            lost.remove(i)
        elif i in reserve:
            reserve.remove(i)
            lost.remove(i)
        elif i+1 in reserve:
            reserve.remove(i+1)
            lost.remove(i)
    return n-len(lost)

아 . .!

감사합니다. .. .

 

다시 도전

def solution(n, lost, reserve):
    lst = sorted(lost)
    reserve.sort()
    for i in lst:
        if i in reserve:
            reserve.remove(i)
            lost.remove(i)
    lst2 = lost[:]
    for i in lst2:
        if i-1 in reserve:
            reserve.remove(i-1)
            lost.remove(i)
        elif i+1 in reserve:
            reserve.remove(i+1)
            lost.remove(i)
    return n-len(lost)

이랬더니 이제 test 13,14가 안됨

;;;; 뭐가 문제야 도대체

아악

 

아 lst = sorted(lost) 이렇게 해줘도 lst2는 lost를 사용하기 때문에

lst2 = lost[:] 이렇게 하면 정렬이 되지 않은 lost를 사용하게 됨

 

🗯️ 반례 입력값 〉 5, [4, 2], [3, 5] 기댓값 〉 5

 

해결코드

def solution(n, lost, reserve):
    lst = sorted(lost)
    reserve.sort()
    for i in lst:
        if i in reserve:
            reserve.remove(i)
            lost.remove(i)
    lst2 = sorted(lost)
    for i in lst2:
        if i-1 in reserve:
            reserve.remove(i-1)
            lost.remove(i)
        elif i+1 in reserve:
            reserve.remove(i+1)
            lost.remove(i)
    return n-len(lost)

 

다른 사람 풀이 

def solution(n, lost, reserve):
    answer = 0
    students, rent  = [1]*(n+1), [0]*(n+2)
    students[0],rent[0] = 3,3
    
    for lo in lost:
        students[lo] = 0
    for re in reserve:
        rent[re] = 1
        
    for i in range(1,n+1):
        if students[i] == 0:
            if rent[i] == 1:
                rent[i] = 0 
                students[i] =1
            elif rent[i-1] == 1 and students[i-1] != 0: 
                rent[i-1] = 0
                students[i] =1
            elif rent[i+1] == 1 and students[i+1] != 0:
                rent[i+1] = 0
                students[i] =1
                
    for student in students:
        if student == 1:
            answer += 1
    return answer

 

참고자료 

https://star-ccomputer-go.tistory.com/75

(Python) 체육복(프로그래머스Lv.1)