結果

問題 No.615 集合に分けよう
ユーザー aaaaasatoriaaaaasatori
提出日時 2018-02-15 17:18:38
言語 Java21
(openjdk 21)
結果
WA  
実行時間 -
コード長 1,088 bytes
コンパイル時間 3,998 ms
コンパイル使用メモリ 77,544 KB
実行使用メモリ 55,936 KB
最終ジャッジ日時 2024-06-07 19:30:07
合計ジャッジ時間 11,206 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 119 ms
46,404 KB
testcase_01 AC 118 ms
41,456 KB
testcase_02 AC 124 ms
41,248 KB
testcase_03 AC 125 ms
40,992 KB
testcase_04 AC 123 ms
41,424 KB
testcase_05 AC 118 ms
41,060 KB
testcase_06 WA -
testcase_07 AC 128 ms
41,268 KB
testcase_08 AC 131 ms
41,068 KB
testcase_09 AC 132 ms
41,420 KB
testcase_10 AC 126 ms
41,324 KB
testcase_11 AC 126 ms
41,364 KB
testcase_12 AC 113 ms
40,744 KB
testcase_13 WA -
testcase_14 WA -
testcase_15 AC 273 ms
48,604 KB
testcase_16 AC 115 ms
41,008 KB
testcase_17 AC 497 ms
48,580 KB
testcase_18 WA -
testcase_19 WA -
testcase_20 TLE -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
権限があれば一括ダウンロードができます

ソースコード

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