본문 바로가기
Algorithm/코딩테스트 (Python)

[SWEA 4835][파이썬 S/W 문제해결 기본]. 1일차 - 구간합

by 싱브이 2023. 9. 22.
728x90
반응형

문제

N개의 정수가 들어있는 배열에서 이웃한 M개의 합을 계산하는 것은 디지털 필터링의 기초연산이다.
M개의 합이 가장 큰 경우와 가장 작은 경우의 차이를 출력하는 프로그램을 작성하시오.

 

내 코드

T = int(input())
for test_case in range(1, T + 1):
    N, M = map(int, input().split())
    numbers = list(map(int, input().split()))
    li = []
    for i in range(N-M+1):
        total = 0
        for j in range(i, M+i):
            total += numbers[j] 
        li.append(total)
    answer = max(li) - min(li)
    print(f'#{test_case} {answer}')
728x90
반응형

댓글