最大平均值dp-python

J. easy problem

【问题描述】

给定一个长度为 n 的非负整数序列 ，求一个平均数最大的，长度不小于 m的子段。

【输入形式】

第 1 行用空格分隔的两个整数 n 和 m；

第 2 行到第 n + 1 行每行一个数字 。

【输出形式】

输出一个整数，表示这个平均数的 1000 倍。不用四舍五入，直接输出。

【样例输入】

10 6

6

4

2

10

3

8

5

9

4

1

【样例输出】

6500

【样例说明】

选 10 3 8 5 9 4

【数据范围】

n,m=map(int,input().split())
l=[]
for i in range(n):
    a=int(input())
    l.append(a)

def s(a,b):
    ans=0
    for i in range(a):
        ans+=b[i]
    return ans

dp=[[0 for i in range(n+1)]for j in range(n+1)]

for i in range(1,n+1):
    
    dp[i][1]=max(l[:i])


for i in range(1,n+1):
    for j in range(1,m+1):
        if i>j:
            dp[i][j]=max(dp[i-1][j],(s(i,l)-s(i-j,l))/j)
a=max(dp[n][m:n+1])
print(int(1000*a))