<Algorithm> 125. 성수의 프로그래밍 강좌 신청(SWExpert)

1. 성수의 프로그래밍 강좌 신청(SWExpert)

• 간단한 정렬 활용 문제

import java.util.LinkedList;
import java.util.List;
import java.util.Scanner;
import static java.util.stream.Collectors.*;
 
public class Solution {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int t = sc.nextInt();
        int testcase = 1;
        /******************************
         * 정렬을 이용해서 A+M/2이 최대가 되는
         * 규칙인 n-k 번째로 큰거부터 n까지
         * 연산을 통해 답을 출력한다.
         *******************************/
        
        while(t-- >0) {
            int n = sc.nextInt();
            int k = sc.nextInt();
            double a = 0;
            List<Integer> list = new LinkedList<>();
            for(int i = 0; i < n; i++) {
                list.add(sc.nextInt());
            }
            
            list = list.stream().sorted().collect(toList());
            
            if(k == 1) {
                System.out.println("#"+(testcase++)+" "+((double)list.get(n-1)/2));
                continue;
            }
            
            for(int i = n-k; i < n ; i++) {
                int num = list.get(i);
                a = (double)((a+num)/2);
            }
            System.out.println("#"+(testcase++)+" "+a);
        }
        
    }
}
 

참고 & 출처