[백준 15954] 인형들 python 파이썬

15954번 - 인형들

문제 이해

가능한 모든 길이가 K 이상인 구간의 표준편차의 최솟값을 구하면 되는 문제

접근

브루트포스로 풀었다. python3가 아닌 pypy3로 제출해야 맞았습니다를 받을 수 있다. python 언어 자체의 한계이다.

정답 코드

import sys
import math
input = sys.stdin.readline
def getInts(): return map(int, input().split())


n, K = getInts()
p = [*getInts()]


def getDis(subList):
    avg = sum(subList) / len(subList)
    res = 0
    for i in subList:
        res += (i-avg)**2
    return res/len(subList)


result = 1e32

for i in range(n-K+1):
    for j in range(n-K-i+2):
        result = min(result, getDis(p[i:i + K + j]))
print(math.sqrt(result))