[Programmers/Python3] 체육복

Code

from collections import Counter

def solution(n, lost, reserve):
    arr = [1] * (n+2)
    for x in reserve:
        arr[x] += 1
    for x in lost:
        arr[x] -= 1
    for i in range(len(arr)):
        if arr[i] == 0:
            if arr[i-1] == 2:
                arr[i] += 1
                arr[i-1] -= 1
            elif arr[i+1] ==2:
                arr[i] += 1
                arr[i+1] -= 1
    arr.pop(0)
    arr.pop()
    return len(arr) - Counter(arr)[0]

---

Result

정확성 테스트

테스트 1 〉	통과 (0.02ms, 10.3MB)
테스트 2 〉	통과 (0.04ms, 10.2MB)
테스트 3 〉	통과 (0.04ms, 10.3MB)
테스트 4 〉	통과 (0.03ms, 10.3MB)
테스트 5 〉	통과 (0.04ms, 10.2MB)
테스트 6 〉	통과 (0.03ms, 10.3MB)
테스트 7 〉	통과 (0.03ms, 10.3MB)
테스트 8 〉	통과 (0.04ms, 10.2MB)
테스트 9 〉	통과 (0.04ms, 10.3MB)
테스트 10 〉	통과 (0.07ms, 10.3MB)
테스트 11 〉	통과 (0.03ms, 10.3MB)
테스트 12 〉	통과 (0.03ms, 10.2MB)
테스트 13 〉	통과 (0.02ms, 10.3MB)
테스트 14 〉	통과 (0.04ms, 10.2MB)
테스트 15 〉	통과 (0.04ms, 10.2MB)
테스트 16 〉	통과 (0.03ms, 10.2MB)
테스트 17 〉	통과 (0.02ms, 10.2MB)
테스트 18 〉	통과 (0.03ms, 10.2MB)
테스트 19 〉	통과 (0.02ms, 10.3MB)
테스트 20 〉	통과 (0.03ms, 10.3MB)