점심시간에 도둑이 들어, 일부 학생이 체육복을 도난당했습니다. 다행히 여벌 체육복이 있는 학생이 이들에게 체육복을 빌려주려 합니다. 학생들의 번호는 체격 순으로 매겨져 있어, 바로 앞번호의 학생이나 바로 뒷번호의 학생에게만 체육복을 빌려줄 수 있습니다. 예를 들어, 4번 학생은 3번 학생이나 5번 학생에게만 체육복을 빌려줄 수 있습니다. 체육복이 없으면 수업을 들을 수 없기 때문에 체육복을 적절히 빌려 최대한 많은 학생이 체육수업을 들어야 합니다.
전체 학생의 수 n, 체육복을 도난당한 학생들의 번호가 담긴 배열 lost, 여벌의 체육복을 가져온 학생들의 번호가 담긴 배열 reserve가 매개변수로 주어질 때, 체육수업을 들을 수 있는 학생의 최댓값을 return 하도록 solution 함수를 작성해주세요.
n | lost | reserve | return |
---|---|---|---|
5 | [2, 4] | [1, 3, 5] | 5 |
5 | [2, 4] | [3] | 4 |
3 | [3] | [1] | 2 |
1번 학생이 2번 학생에게 체육복을 빌려주고, 3번 학생이나 5번 학생이 4번 학생에게 체육복을 빌려주면 학생 5명이 체육수업을 들을 수 있습니다.
3번 학생이 2번 학생이나 4번 학생에게 체육복을 빌려주면 학생 4명이 체육수업을 들을 수 있습니다.
def solution(n, lost, reserve):
student = dict()
for i in range(1, n + 1):
student[i] = 1
for l_cloth in lost:
student[l_cloth] -= 1
for r_cloth in reserve:
student[r_cloth] += 1
for k, v in student.items():
if v == 0 and student[k - 1] >= 1:
student[k - 1] -= 1
student[k] += 1
elif v == 0 and student[k + 1] >= 1:
student[k + 1] -= 1
student[k] += 1
answer = 0
for k, v in student.items():
if v >= 1:
answer += 1
return answer
딕셔너리에 체육벌을 잃어버린 학생, 한 벌만 있는 학생, 여벌을 갖고 있는 학생을 각각 0, 1, 2 값으로 저장하고 왼쪽에서 빌리고 그게 안되면 오른쪽에서 빌리는 형식으로 짰다.
테스트 1 〉 실패 (런타임 에러)
테스트 2 〉 실패 (런타임 에러)
테스트 3 〉 실패 (런타임 에러)
테스트 4 〉 실패 (런타임 에러)
테스트 5 〉 실패 (0.02ms, 10.1MB)
테스트 6 〉 통과 (0.01ms, 10.3MB)
테스트 7 〉 실패 (0.02ms, 10.3MB)
테스트 8 〉 통과 (0.01ms, 10.3MB)
테스트 9 〉 실패 (런타임 에러)
테스트 10 〉 실패 (런타임 에러)
테스트 11 〉 실패 (런타임 에러)
테스트 12 〉 실패 (런타임 에러)
채점 결과
정확성: 16.7
합계: 16.7 / 100.0
일단 첫 번째 애가 체육복을 잃어버리면 KeyError가 나는 것을 확인했다.
실패는 이 부분을 수정한 후 확인해봐야겠다.
def solution(n, lost, reserve):
for r in reserve:
if r in lost:
lost.remove(r)
elif r - 1 in lost:
lost.remove(r - 1)
elif r + 1 in lost:
lost.remove(r + 1)
return n - len(lost)
딕셔너리를 만들지않고 생각할 수 있을까 하고 보다보니 lost에서 뺄거빼고 계산하는 것이 나을 것이라고 생각했다.
테스트 1 〉 통과 (0.00ms, 10.2MB)
테스트 2 〉 통과 (0.01ms, 10.3MB)
테스트 3 〉 통과 (0.01ms, 10.2MB)
테스트 4 〉 통과 (0.01ms, 10.2MB)
테스트 5 〉 실패 (0.01ms, 10.2MB)
테스트 6 〉 통과 (0.00ms, 10.3MB)
테스트 7 〉 실패 (0.02ms, 10.2MB)
테스트 8 〉 통과 (0.01ms, 10.2MB)
테스트 9 〉 통과 (0.00ms, 10.3MB)
테스트 10 〉 통과 (0.01ms, 10.2MB)
테스트 11 〉 통과 (0.00ms, 10MB)
테스트 12 〉 통과 (0.00ms, 10.2MB)
채점 결과
정확성: 83.3
합계: 83.3 / 100.0
n = 5, lost = [1, 2, 4, 5], reserve = [2, 3, 4] 일 때 reserve 값이 3인 애가 4에게 옷을 빌려준다. 그래서 제대로 되면 리턴값이 3이여야하는데 위 코드대로면 4가 나온다.
def solution(n, lost, reserve):
new_lost = list(set(lost) - set(reserve))
new_reserve = list(set(reserve) - set(lost))
for r in new_reserve:
if r - 1 in new_lost:
new_lost.remove(r - 1)
elif r + 1 in new_lost:
new_lost.remove(r + 1)
return n - len(new_lost)
차집합의 개념을 이용해서 lost와 reserve를 정리해준 후, 위의 코드와 동일하게 실행하니 해결되었다. 그리고 이제 위에서 사용한
if r in lost:
lost.remove(r)
이 코드는 필요하지 않아서 삭제했다.
def solution(n, lost, reserve):
_reserve = [r for r in reserve if r not in lost]
_lost = [l for l in lost if l not in reserve]
for r in _reserve:
f = r - 1
b = r + 1
if f in _lost:
_lost.remove(f)
elif b in _lost:
_lost.remove(b)
return n - len(_lost)
reserve와 lost를 조건문을 포함한 리스트로 만드는 것을 배워둬야 할 것 같다.