提出 #31632219

---

ソースコード 拡げる

import sys
import numpy as np
def solve(s):
    #s = input()
    #sl = list(input().split())
    n, k = s[:2]
    h = s[2:]
    #ここにコード
    dp = np.full(n, 10**10, np.int64)
    dp[0] = 0
     
    for i in range(n):
        dp[i:i+k+1] = np.minimum(dp[i:i+k+1], dp[i]+np.abs(h[i:i+k+1]-h[i]))


    #N = 1のときなどcorner caseを確認！
    return dp[-1]#answer


def main():
    stdin = np.fromstring(open(0).read(), dtype=np.int64, sep=' ')
    # stdin = np.fromstring(IN, dtype=np.int64, sep=' ')
    print(solve(stdin))

def cc_export():
    from numba.pycc import CC
    cc = CC('my_module')
    cc.export('solve', '(i8[:],)')(solve)
    cc.compile()
 
if __name__ == "__main__":
    if sys.argv[-1] == 'ONLINE_JUDGE':
        cc_export()
        sys.exit(0)
    
    from my_module import solve # type: ignore
    main()

提出情報

ジャッジ結果