結果

問題 No.615 集合に分けよう
コンテスト
ユーザー aaaaasatori
提出日時 2018-02-15 17:18:38
言語 Java
(openjdk 23)
結果
WA  
実行時間 -
コード長 1,088 bytes
コンパイル時間 4,383 ms
コンパイル使用メモリ 77,508 KB
実行使用メモリ 98,032 KB
最終ジャッジ日時 2024-12-26 04:08:37
合計ジャッジ時間 24,253 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 17 WA * 6 TLE * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.Arrays;
import java.util.Scanner;

public class No615 {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int N = sc.nextInt();
		int K = sc.nextInt();
		
		if(N == K) {
			System.out.println(0);
		}else {
			long a[] = new long[N];
			for(int i = 0;i < N;i++) {
				a[i] = sc.nextLong();
			}
			Arrays.sort(a);
			if(K == 1) {
				System.out.println(a[N-1] - a[0]);
			}else {
			
				int b[] = new int[K-1]; //分割する位置を入れる配列(a[i] a[i+1]の間で分割するならiが入る)
				long max[] = new long[K-1];
				long sum = 0;
				for(int i = 0;i < K-1;i++) {
					max[i] = 0;
				}
				for(int i = 0;i < N-1;i++) {
					if(max[0] < a[i+1] - a[i]) {
						max[0] = a[i+1] - a[i];
						b[0] = i;
						Arrays.sort(max);
						Arrays.sort(b);
					}
				}
				
				for(int i = 0;i < K-1;i++) {
					if(i == 0) {
						sum += a[b[i]] - a[0];
					}else {
						sum += a[b[i]] - a[b[i-1]+1];
					}
					if(i + 1 == K - 1) {
						sum += a[N-1] - a[b[i]+1];
					}
				}
				
				System.out.println(sum);
			}
		}
	}
}
0