[SW Expert Academy] - (D3)4466. 최대 성적표 만들기

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AWOUfCJ6qVMDFAWg&categoryId=AWOUfCJ6qVMDFAWg&categoryType=CODE

SW Expert Academy

 

코드

import java.util.Arrays;
import java.util.Scanner;

public class Solution {

	public static void main(String[] args) {
		Scanner scan = new Scanner(System.in);
		
		int t = scan.nextInt();
		
		for(int tc=1; tc<=t; tc++) {
			int N = scan.nextInt();
			int K = scan.nextInt();
			int[] scores = new int[N];
			int sum = 0;
			
			for(int i=0; i<N; i++) {
				scores[i] = scan.nextInt();
			}
			
			Arrays.sort(scores);

			// K개의 과목을 선택하므로, 더할때마다 -1씩 하고 0이될경우 탈출
			for(int i=scores.length-1; i>=0; i--) {
				sum += scores[i];
				K --;
				if(K == 0) {
					break;
				}
			}
			
			System.out.println("#" + tc + " " + sum);
			
		}
		
		scan.close();
	}

}

풀이

 

성적이 주어지고 K개의 과목만 선택해서 얻을 수 있는 최대점수를 출력하면 된다.

배열을 정렬해서 뒤에서 부터(값이 큰값부터) K개수만큼 더한다.

 

 

+++

음 오름차순 정렬이므로 가장 큰 값이 뒤에서부터 정렬되고, 뒤에서부터 더하다보니 for문안에 불필요한 if문이 더 추가되었다.

만약 i값이 0부터라면 i=0; i<K 와 같이 간단하게 할 수 있다. 조금만 생각해보면 뒤에서 부터 출력해도 i값을 

배열의 길이에서 K값을 뺀만큼만 실행하면 동일한 결과이다. 따라서 다음 코드도 동일하게 작동한다.

import java.util.Arrays;
import java.util.Scanner;

public class Solution {

	public static void main(String[] args) {
		Scanner scan = new Scanner(System.in);
		
		int t = scan.nextInt();
		
		for(int tc=1; tc<=t; tc++) {
			int N = scan.nextInt();
			int K = scan.nextInt();
			int[] scores = new int[N];
			int sum = 0;
			
			for(int i=0; i<N; i++) {
				scores[i] = scan.nextInt();
			}
			
			Arrays.sort(scores);

			// K개의 과목을 선택하므로, 더할때마다 -1씩 하고 0이될경우 탈출
			for(int i=scores.length-1; i>=scores.length-K; i--) {
				sum += scores[i];
			}
			
			System.out.println("#" + tc + " " + sum);
			
		}
		
		scan.close();
	}

}

 

for문에서 if문을 제거하고, 조건에서 K과목의 개수만큼만 더하도록 변경했다.

			// K개의 과목을 선택하므로, 더할때마다 -1씩 하고 0이될경우 탈출
			for(int i=scores.length-1; i>=scores.length-K; i--) {
				sum += scores[i];
			}