코알라 스터디1 # 2559번 수열 ''' 10개의 원소가 있고 3 크기의 윈도우를 슬라이딩 시킨다고 하면 0 1 2 3 4 5 6 7 8 9 |___| | | | | | | | |___| | | | | | | |___| | | | | | |___| | | | | |___| | | | |___| | | |___| | |___| 반복 횟수는 10 - 3 + 1 = 8회다. 최초 윈도우를 설정하는1회를 제외한 7회동안 제일 앞과 제일 뒤의 값을 빼고 더하면 된다. n개의 원소와 k크기의 윈도우라면 n-k회가 된다. ''' import sys def sliding_window_sum(arr, n, k): window = sum(arr[0:k]) result = window for i in range(n-k): window = window - a.. 2024. 3. 9. 이전 1 다음