풀이
처음에는 순열로 접근했다가 시간초과가 발생해서 문제를 다시 보았는데 굳이 순열을 쓸 필요없이 입력받은 과목의 수준을 정렬해서 숫자가 큰 순서대로 K개를 뽑아 K개에서 가장 작은 수 부터 실력을 갱신하여 해결했습니다.
코드
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;
public class Solution {
public static int N, K;
public static double answer;
public static Integer[] subject;
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringBuilder sb = new StringBuilder("");
StringTokenizer st;
int T = Integer.parseInt(br.readLine());
for (int tc = 1; tc <= T; ++tc) {
st = new StringTokenizer(br.readLine(), " ");
N = Integer.parseInt(st.nextToken());
K = Integer.parseInt(st.nextToken());
subject = new Integer[N];
answer = 0;
st = new StringTokenizer(br.readLine(), " ");
for (int i = 0; i < N; ++i) {
subject[i] = Integer.parseInt(st.nextToken());
}
Arrays.sort(subject);
for (int i = N - K; i < N; ++i) {
answer = (answer + (double)subject[i]) / 2.0;
}
sb.append("#").append(tc).append(" ").append(String.format("%.6f", answer)).append("\n");
}
System.out.println(sb);
}
}
'알고리즘 > SWEA' 카테고리의 다른 글
[Java] SWEA 3347번 올림픽 종목 투표 (0) | 2021.03.12 |
---|---|
[Java] SWEA 7854번 최약수 (0) | 2021.03.11 |
[Java] SWEA 1238번 Contact (0) | 2021.03.09 |
[Java] SWEA 1218번 괄호 짝짓기 (0) | 2021.03.08 |
[Java] SWEA 1868번 파핑파핑 지뢰찾기 (0) | 2021.03.05 |