結果
| 問題 | No.2548 Problem Selection |
| ユーザー |
 Asahi
|
| 提出日時 | 2023-11-27 01:12:57 |
| 言語 | Java21 (openjdk 21) |
| 結果 |
AC
|
| 実行時間 | 268 ms / 2,000 ms |
| コード長 | 12,852 bytes |
| コンパイル時間 | 3,394 ms |
| コンパイル使用メモリ | 104,668 KB |
| 実行使用メモリ | 60,956 KB |
| 最終ジャッジ日時 | 2023-11-27 01:13:07 |
| 合計ジャッジ時間 | 9,291 ms |
|
ジャッジサーバーID （参考情報） |
judge14 / judge12 |
（要ログイン）
テストケース
テストケース表示| 入力 | 結果 | 実行時間 実行使用メモリ |
|---|---|---|
| testcase_00 | AC | 241 ms
60,316 KB |
| testcase_01 | AC | 189 ms
59,316 KB |
| testcase_02 | AC | 265 ms
60,440 KB |
| testcase_03 | AC | 172 ms
58,768 KB |
| testcase_04 | AC | 224 ms
60,396 KB |
| testcase_05 | AC | 263 ms
60,664 KB |
| testcase_06 | AC | 170 ms
58,400 KB |
| testcase_07 | AC | 246 ms
60,472 KB |
| testcase_08 | AC | 245 ms
60,396 KB |
| testcase_09 | AC | 252 ms
60,324 KB |
| testcase_10 | AC | 268 ms
60,956 KB |
| testcase_11 | AC | 262 ms
60,832 KB |
| testcase_12 | AC | 267 ms
60,440 KB |
| testcase_13 | AC | 117 ms
57,080 KB |
| testcase_14 | AC | 120 ms
57,468 KB |
| testcase_15 | AC | 120 ms
57,448 KB |
| testcase_16 | AC | 120 ms
57,456 KB |
| testcase_17 | AC | 120 ms
57,712 KB |
| testcase_18 | AC | 119 ms
57,116 KB |
| testcase_19 | AC | 124 ms
58,068 KB |
| testcase_20 | AC | 120 ms
57,724 KB |
| testcase_21 | AC | 119 ms
57,460 KB |
| testcase_22 | AC | 119 ms
57,464 KB |
| testcase_23 | AC | 120 ms
57,448 KB |
| testcase_24 | AC | 117 ms
57,104 KB |
| testcase_25 | AC | 126 ms
57,068 KB |
| testcase_26 | AC | 120 ms
57,452 KB |
ソースコード
import java.util.*;
import java.io.*;
import java.math.*;
import java.util.stream.*;
import java.util.function.*;
class Main implements Runnable {
public void solve(){
int n = in.nextInt() , m = in.nextInt();
long [] a = in.nextLong(n);
long ans = lnf ;
long total = 0 ;
Arrays.sort(a);
for(int i = 0 , j = 0 ; i < n ; i ++) {
while(j + 1 < n && (j - i + 1) < m) {
total += (a[j + 1] - a[j]) * (a[j + 1] - a[j]);
j ++;
}
if((j - i + 1) == m ) ans = Math.min(ans , total);
if(i != n - 1) total -= (a[i + 1] - a[i]) * (a[i + 1] - a[i]);
}
print(ans);
}
public record Pair<T,S> (T fi , S se) {
public String toString() {
return "("+fi+","+se+")";
}
}
public record Trio<T,S,U>(T fi , S se , U th) {
public String toString() {
return "("+fi+","+se+","+th+")";
}
}
public PrintWriter out = new PrintWriter(System.out);
public In in = new In();
public static final int mod7 = 1000000007;
public static final int mod9 = 998244353;
public static final int inf = (1 << 30);
public static final long lnf = (1L << 60);
public static final String yes = "Yes";
public static final String no = "No" ;
public static final int [] dy4 = {-1,0,1,0};
public static final int [] dx4 = {0,1,0,-1};
public static final int [] dy8 = {-1,-1,-1,0,1,1,1,0};
public static final int [] dx8 = {-1,0,1,1,1,0,-1,-1};
public boolean isOver(int y , int x , int h , int w) {
return y < 0 || x < 0 || y >= h || x >= w ;
}
public void swap(int [] array , int l , int r) {
int tmp = array[l] ;
array[l] = array[r] ;
array[r] = tmp ;
}
public void swap(long [] array , int l , int r) {
long tmp = array[l] ;
array[l] = array[r] ;
array[r] = tmp ;
}
public String swap(String string , int l , int r) {
StringBuilder m = new StringBuilder(string) ;
m.setCharAt(l, string.charAt(r)); m.setCharAt(r, string.charAt(l));
return m.toString();
}
public String binarytoString(int a , int len) {
String b = Integer.toBinaryString(a);
while(b.length() < len) b = "0" + b ;
return b ;
}
public String binarytoString(long a , int len) {
String b = Long.toBinaryString(a);
while(b.length() < len) b = "0" + b ;
return b ;
}
public <T extends Comparable<T>> int lower_count(List<T> A , T key) {
return upperbound(A, key);
}
public <T extends Comparable<T>> int lowercount(List<T> A , T key) {
return lowerbound(A, key);
}
public <T extends Comparable<T>> int higher_count(List<T> A , T key) {
return A.size() - lowerbound(A, key);
}
public <T extends Comparable<T>> int highercount(List<T> A , T key) {
return A.size() - upperbound(A, key);
}
private <T extends Comparable<T>> int lowerbound(List<T> A, T key) {
int left = 0 , right = A.size();
while (left < right) {
int mid = (left + right) / 2;
if (A.get(mid).compareTo(key) < 0) left = mid + 1;
else right = mid;
}
return right;
}
private <T extends Comparable<T>> int upperbound(List<T> A, T key) {
int left = 0 , right = A.size();
while (left < right) {
int mid = (left + right) / 2;
if (A.get(mid).compareTo(key) <= 0) left = mid + 1;
else right = mid;
}
return right;
}
public Integer [] toInteger(int [] a) {
return Arrays.stream(a).boxed().toArray(Integer[]::new);
}
public Long [] toLong(long [] a) {
return Arrays.stream(a).boxed().toArray(Long[]::new);
}
public Double [] toDouble(double [] a) {
return Arrays.stream(a).boxed().toArray(Double[]::new);
}
public <T> List<T> ArrayList(T [] a) {
return Arrays.stream(a).collect(Collectors.toCollection(ArrayList::new));
}
public <T> HashSet<T> HashSet(T [] a) {
return Arrays.stream(a).collect(Collectors.toCollection(HashSet::new));
}
public <T> TreeSet<T> TreeSet(T [] a) {
return Arrays.stream(a).collect(Collectors.toCollection(TreeSet::new));
}
public <T> Deque<T> Deque(T [] a) {
return Arrays.stream(a).collect(Collectors.toCollection(ArrayDeque::new));
}
public <T> PriorityQueue<T> PriorityQueue(T [] a) {
return Arrays.stream(a).collect(Collectors.toCollection(PriorityQueue::new));
}
public <T> List <T> [] ArrayList(int n) {
@SuppressWarnings("unchecked")
List<T> [] G = new ArrayList[n];
for(int i = 0 ; i < n ; i ++ ) G[i] = new ArrayList<>();
return G ;
}
public <T> HashSet <T> [] HashSet(int n) {
@SuppressWarnings("unchecked")
HashSet<T> [] G = new HashSet[n];
for(int i = 0 ; i < n ; i ++ ) G[i] = new HashSet<>() ;
return G ;
}
public <T> TreeSet <T> [] TreeSet(int n) {
@SuppressWarnings("unchecked")
TreeSet<T> [] G = new TreeSet[n];
for(int i = 0 ; i < n ; i ++ ) G[i] = new TreeSet<>() ;
return G ;
}
public <T> TreeSet <T> [] TreeSet(int n , Comparator<? super T> comparator) {
@SuppressWarnings("unchecked")
TreeSet<T> [] G = new TreeSet[n];
for(int i = 0 ; i < n ; i ++ ) G[i] = new TreeSet<>(comparator) ;
return G ;
}
public <T> Deque <T> [] Deque(int n) {
@SuppressWarnings("unchecked")
Deque<T> [] G = new ArrayDeque[n];
for(int i = 0 ; i < n ; i ++ ) G[i] = new ArrayDeque<>();
return G ;
}
public <T> PriorityQueue <T> [] PriorityQueue(int n) {
@SuppressWarnings("unchecked")
PriorityQueue<T> [] G = new PriorityQueue[n];
for(int i = 0 ; i < n ; i ++ ) G[i] = new PriorityQueue<>();
return G ;
}
public <T> PriorityQueue <T> [] PriorityQueue(int n , Comparator<? super T> comparator) {
@SuppressWarnings("unchecked")
PriorityQueue<T> [] G = new PriorityQueue[n];
for(int i = 0 ; i < n ; i ++ ) G[i] = new PriorityQueue<>(comparator);
return G ;
}
public <K,V> HashMap <K,V> [] HashMap(int n) {
@SuppressWarnings("unchecked")
HashMap<K,V> [] G = new HashMap[n];
for(int i = 0 ; i < n ; i ++ ) G[i] = new HashMap<>() ;
return G ;
}
public <K,V> TreeMap <K,V> [] TreeMap(int n) {
@SuppressWarnings("unchecked")
TreeMap<K,V> [] G = new TreeMap[n];
for(int i = 0 ; i < n ; i ++ ) G[i] = new TreeMap<>() ;
return G ;
}
public <K,V> TreeMap <K,V> [] TreeMap(int n , Comparator<? super K> comparator) {
@SuppressWarnings("unchecked")
TreeMap<K,V> [] G = new TreeMap[n];
for(int i = 0 ; i < n ; i ++ ) G[i] = new TreeMap<>(comparator) ;
return G ;
}
//---------------------------------------------------------------------------------------//
@SuppressWarnings("unchecked")
public <T> void print(T ... data) {
StringBuilder pr = new StringBuilder();
for(var temp : data) pr.append(temp+" ");
out.println(pr.toString());
out.flush();
}
public void print(String [][] array) {
StringBuilder pr = new StringBuilder();
for(var temp : array) {
for(var data : temp) {
pr.append(data+" ");
}
pr.append("\n");
}
if(0 < pr.length()) pr.deleteCharAt(pr.length() - 1);
out.println(pr.toString());
out.flush();
}
public void print(char [][] array) {
StringBuilder pr = new StringBuilder();
for(var temp : array) {
for(var data : temp) {
pr.append(data);
}
pr.append("\n");
}
if(0 < pr.length()) pr.deleteCharAt(pr.length() - 1);
out.println(pr.toString());
out.flush();
}
public <T> void print(Collection<T> collection) {
StringBuilder pr = new StringBuilder();
for(T data : collection) pr.append(data +" ");
out.println(pr.toString());
out.flush();
}
public static void main(String ... args) {
new Thread(null, new Main(), "", Runtime.getRuntime().maxMemory()).start();
}
public void run() {
solve();
out.flush();
}
}
class In {
private final InputStream in = System.in;
private final Scanner sc = new Scanner(System.in);
private final byte[] buffer = new byte[1024];
private int ptr = 0;
private int buflen = 0;
private boolean hasNextByte() {
if (ptr < buflen) {
return true;
}else{
ptr = 0;
try {
buflen = in.read(buffer);
} catch (IOException e) {
e.printStackTrace();
}
if (buflen <= 0) {
return false;
}
}
return true;
}
private int readByte() {
if (hasNextByte()) return buffer[ptr++]; else return -1;
}
private static boolean isPrintableChar(int c) {
return 33 <= c && c <= 126;
}
private boolean hasNext() {
while(hasNextByte() && !isPrintableChar(buffer[ptr])) {
ptr++;
}
return hasNextByte();
}
public String next() {
if (!hasNext()) throw new NoSuchElementException();
StringBuilder sb = new StringBuilder();
int b = readByte();
while(isPrintableChar(b)) {
sb.appendCodePoint(b);
b = readByte();
}
return sb.toString();
}
public long nextLong() {
if (!hasNext()) throw new NoSuchElementException();
long n = 0;
boolean minus = false;
int b = readByte();
if (b == '-') {
minus = true;
b = readByte();
}
if (b < '0' || '9' < b) {
throw new NumberFormatException();
}
while(true){
if ('0' <= b && b <= '9') {
n *= 10;
n += b - '0';
}else if(b == -1 || !isPrintableChar(b)){
return minus ? -n : n;
}else{
throw new NumberFormatException();
}
b = readByte();
}
}
public int nextInt() {
long nl = nextLong();
if (nl < Integer.MIN_VALUE || nl > Integer.MAX_VALUE) throw new NumberFormatException();
return (int) nl;
}
public double nextDouble() {
return Double.parseDouble(next());
}
public char nextChar() {
return next().charAt(0);
}
public BigInteger nexBigInteger() {
return sc.nextBigInteger();
}
public int [] nextInt(int n) {
int [] array = new int[n];
for(int i = 0 ; i < n ; i ++) {
array[i] = nextInt();
}
return array ;
}
public int [][] nextInt(int n , int m) {
int [][] array = new int[n][m];
for(int i = 0 ; i < n ; i ++) {
array[i] = nextInt(m);
}
return array ;
}
public long [] nextLong(int n) {
long [] array = new long[n];
for(int i = 0 ; i < n ; i ++) {
array[i] = nextLong();
}
return array ;
}
public long [][] nextLong(int n , int m) {
long [][] array = new long[n][m];
for(int i = 0 ; i < n ; i ++) {
array[i] = nextLong(m);
}
return array ;
}
public double [] nextDouble(int n) {
double [] array = new double[n];
for(int i = 0 ; i < n ; i ++) {
array[i] = nextDouble();
}
return array ;
}
public String [] next(int n) {
String [] array = new String[n];
for(int i = 0 ; i < n ; i ++) {
array[i] = next();
}
return array ;
}
public String [][] next(int n , int m) {
String [][] array = new String[n][m];
for(int i = 0 ; i < n ; i ++) {
array[i] = next(m);
}
return array ;
}
public char [] nextChar(int n) {
char [] array = new char[n];
String string = next() ;
for(int i = 0 ; i < n ; i ++) {
array[i] = string.charAt(i);
}
return array ;
}
public char [][] nextChar(int n , int m) {
char [][] array = new char[n][m];
for(int i = 0 ; i < n ; i ++) {
array[i] = nextChar(m);
}
return array ;
}
}