結果

問題 No.2548 Problem Selection
ユーザー tipstar0125tipstar0125
提出日時 2023-11-25 13:14:00
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 3,055 bytes
コンパイル時間 415 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 110,720 KB
最終ジャッジ日時 2024-09-26 10:23:47
合計ジャッジ時間 6,986 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 206 ms
104,448 KB
testcase_01 AC 184 ms
91,008 KB
testcase_02 AC 221 ms
110,592 KB
testcase_03 AC 182 ms
89,344 KB
testcase_04 AC 200 ms
100,480 KB
testcase_05 AC 220 ms
110,336 KB
testcase_06 AC 180 ms
89,216 KB
testcase_07 AC 205 ms
103,552 KB
testcase_08 AC 214 ms
105,088 KB
testcase_09 AC 209 ms
107,776 KB
testcase_10 AC 218 ms
110,592 KB
testcase_11 AC 219 ms
110,336 KB
testcase_12 AC 218 ms
110,720 KB
testcase_13 WA -
testcase_14 WA -
testcase_15 AC 171 ms
89,088 KB
testcase_16 AC 173 ms
89,088 KB
testcase_17 WA -
testcase_18 AC 174 ms
89,216 KB
testcase_19 AC 174 ms
89,088 KB
testcase_20 WA -
testcase_21 AC 172 ms
89,088 KB
testcase_22 AC 172 ms
89,344 KB
testcase_23 AC 171 ms
89,088 KB
testcase_24 AC 173 ms
89,088 KB
testcase_25 AC 170 ms
89,216 KB
testcase_26 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

from __future__ import annotations

import array
import bisect
import copy
import fractions
import heapq
import itertools
import math
import random
import re
import string
import sys
import time
from collections import defaultdict, deque
from functools import lru_cache

sys.setrecursionlimit(10**6)


def read_int_list():
    return list(map(int, input().split()))


def read_int():
    return int(input())


def read_str_list():
    return list(input().split())


def read_str():
    return input()


def is_prime(n: int) -> bool:
    if n < 2:
        return False
    i = 2
    ok = True
    while i * i <= n:
        if n % i == 0:
            ok = False
        i += 1
    return ok


def eratosthenes(n: int) -> list[bool]:

    is_prime_list = ([False, True] * (n // 2 + 1))[0 : n + 1]
    is_prime_list[1] = False
    is_prime_list[2] = True
    for i in range(3, n + 1, 2):
        if not (is_prime_list[i]):
            continue
        if i * i > n:
            break
        for k in range(i * i, n + 1, i):
            is_prime_list[k] = False
    return is_prime_list


def legendre(n: int, p: int) -> int:
    cnt = 0
    pp = p
    while pp <= n:
        cnt += n // pp
        pp *= p

    return cnt


def prime_factorize(n: int) -> defaultdict[int, int]:
    nn = n
    i = 2
    d: defaultdict[int, int] = defaultdict(int)
    while i * i <= n:
        while nn % i == 0:
            d[i] += 1
            nn //= i
        i += 1
    if nn != 1:
        d[nn] += 1
    return d


def make_divisors(n: int) -> list[int]:
    i = 1
    ret = []
    while i * i <= n:
        if n % i == 0:
            ret.append(i)
            if i != n // i:
                ret.append(n // i)
        i += 1
    ret.sort()
    return ret


def gcd(a: int, b: int) -> int:

    if a == 0:
        return b
    else:
        return gcd(b % a, a)


def lcm(a: int, b: int) -> int:
    return a * b // gcd(a, b)


def ext_gcd(a: int, b: int) -> tuple[int, int, int]:
    if a == 0:
        return (0, 1, b)
    x, y, g = ext_gcd(b % a, a)
    return (y - b // a * x, x, g)

class UnionFind:
    def __init__(self, n):
        self.parent = [-1] * n

    def find(self, x):
        if self.parent[x] < 0:
            return x
        else:
            self.parent[x] = self.find(self.parent[x])
            return self.parent[x]

    def merge(self, x, y):
        x = self.find(x)
        y = self.find(y)

        if x != y:
            if self.parent[x] > self.parent[y]:
                x, y = y, x
            self.parent[x] += self.parent[y]
            self.parent[y] = x

    def same(self, x, y):
        return self.find(x) == self.find(y)

    def size(self, x):
        return -self.parent[self.find(x)]



def solve():

    N,M=read_int_list()
    A=read_int_list()
    ans=1<<60
    A.sort()
    S=[0]
    for i in range(1,N):
        s=(A[i]-A[i-1])**2+S[-1]
        S.append(s)
    for i in range(1,N-M+1):
        ans=min(ans, S[M+i-1]-S[i])
    print(ans)


def main():
    # t = read_int()
    # for _ in range(t):
    solve()


main()
0