[JAVA]백준_13164_행복유치원

문제링크

https://www.acmicpc.net/problem/13164

13164번: 행복 유치원

 

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main {

	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());
		
		int N = Integer.parseInt(st.nextToken());
		int K = Integer.parseInt(st.nextToken());
		
		int arr[] = new int[N];
		int diff[] = new int[N-1];
		
		st = new StringTokenizer(br.readLine());
		
		for(int i=0;i<N;i++) {
			arr[i] = Integer.parseInt(st.nextToken());
		}
		
		for(int i=0;i<N-1;i++) {
			diff[i] = arr[i+1]-arr[i];
		}
		
//		System.out.println(Arrays.toString(arr));
//		System.out.println(Arrays.toString(diff));
		
		Arrays.sort(diff);
		
		int res = 0;
		
		for(int i=0;i<N-K;i++) {
			res+= diff[i];
		}
		
		System.out.println(res);
	}
}

 

arr을 통해 학생들을 입력받은 후 두 학생 사이의 차이를 저장한다 (diff)

diff를 Arrays.sort를 통해 정렬하면 키 차이가 가장 작은 학생부터 정렬된다.

그 수를 N-K만큼 더해주면 해결된다.