본문 바로가기
Problem Solving/Baekjoon

[백준] 29767 점수를 최대로 - Greedy / Java

by graycode 2023. 11. 21.

 문제 링크

 

29767번: 점수를 최대로

단대소프트고에는 교실 $N$개가 있다. 교실은 $1$번부터 $N$번까지 $1, 2, \ldots, N$ 순서로 연달아 있다. 학교 밖에는 $K$명의 학생들이 있다. $K$명의 학생은 학교에 들어가기 전 학생마다 목적지 교실

www.acmicpc.net

 

 풀이 코드

import java.io.BufferedWriter;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.util.Arrays;

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        int n = (int) read(), k = (int) read();
        long[] arr = new long[n];

        arr[0] = read();
        for (int i = 1; i < n; i++) arr[i] = arr[i - 1] + read();

        Arrays.sort(arr);

        long sum = 0;
        while (k-- > 0) sum += arr[--n];

        bw.write(String.valueOf(sum));
        bw.flush();
    }

    private static long read() throws IOException {
        long c, n = System.in.read() & 15;
        boolean flag = n == 13;

        if (flag) n = System.in.read() & 15;
        while ((c = System.in.read()) > 32) n = (n << 3) + (n << 1) + (c & 15);

        return flag ? ~n + 1 : n;
    }

}

댓글