Algorithm/CodingTest

[Baekjoon] 1026번 보물 (Java)

누구세연 2023. 10. 28. 23:33

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

 

1026번: 보물

첫째 줄에 N이 주어진다. 둘째 줄에는 A에 있는 N개의 수가 순서대로 주어지고, 셋째 줄에는 B에 있는 수가 순서대로 주어진다. N은 50보다 작거나 같은 자연수이고, A와 B의 각 원소는 100보다 작거

www.acmicpc.net

 

보물이라는 문제 자바로 풀이하겠습니다.

 

주어진 문제는 두 배열 A와 B를 이용하여 S를 정의하고, S의 값을 최소로 만들기 위해 A배열의 순서를 재배열하는 것입니다.

N의 크기가 비교적 작기 때문에 간단한 정렬 알고리즘을 사용하여 해결할 수 있습니다.

 

다음과 같이 풀이하였습니다.

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 NumberFormatException, IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int N = Integer.parseInt(br.readLine());
		
		StringTokenizer st = new StringTokenizer(br.readLine());
		int[] arrA = new int[N];
		for(int i=0; i<N; i++) {
			arrA[i] = Integer.parseInt(st.nextToken());
		}
		
		st = new StringTokenizer(br.readLine());
		int[] arrB = new int[N];
		for(int i=0; i<N; i++) {
			arrB[i] = Integer.parseInt(st.nextToken());
		}
		
		Arrays.sort(arrA);
		Arrays.sort(arrB);
		int sum = 0;
		for(int i=0; i<N; i++) {
			sum += arrA[i]*arrB[N-i-1];
		}
		
		System.out.println(sum);

	}

}

 

1. BufferedReader와 InputStreamReader를 사용하여 입력을 처리합니다.
    - System.in을 사용하여 표준 입력 스트림을 가져옵니다.
    - InputStreamReader 객체를 생성하고 이를 BufferedReader의 생성자에 전달하여 BufferedReader 객체를 만듭니다.
    - BufferedReader의 readLine() 메서드를 사용하여 한 줄씩 입력을 읽어 들입니다. 반환값은 String 타입입니다.

2. 입력된 N값을 기반으로 두 개의 배열 arrA와 arrB를 초기화합니다.

3. StringTokenizer를 사용하여 두 배열의 값을 입력받습니다.
   StringTokenizer는 문자열을 특정 구분자를 기준으로 쪼개어 처리하는 데 사용되는 클래스입니다.
   위는 입력된 값을 공백을 기준으로 배열로 받는 방법입니다.
   만약 다른 구분자로 문자열을 나눠 받으려면 생성자에 구분자를 전달해서 사용합니다

StringTokenizer st = new StringTokenizer(br.readLine(), ",");

4. 두 배열을 각각 정렬합니다.

5. 각 배열의 요소들을 이용하여 S값을 계산합니다.
   arrA의 값은 순서대로,  arrB의 값은 역순으로 곱한 후 sum에 누적합니다.

6. sum의 값을 출력합니다.

 

문제의 크기가 비교적 작기 때문에 위의 방법을 사용하여 문제를 해결할 수 있었습니다.

 

이견이 있으시다면 자유롭게 댓글 남겨주세요.