[코딩테스트] Java_ 백준 1026번 보물 (그리디 알고리즘) - Rowen Jobs

백준 1026번 보물

 

우선.. 주어진 조건을 끝까지 지키고 반례를 찾기 위해 정말 오래 시간이 걸렸다..

결국은 반례를 못 찾고 다른 사람들이 진행했던대로는 금방 풀었다..  

B 배열은 정렬을 하지말고 풀어야 해서 B 관련되게 담아준 List 도 정렬 없이 진행을 하였는데 예시 또는 다른 예시들까지 적용했지만 반례를 찾지 못하였다.. 

 

내가 생각한 프로세스는 아래와 같다.

1. A배열 오름차순 진행

2. B배열(정렬하지 않는 조건으로 우선 담아둠)과 List 변수에 값 삽입

3. List 배열만큼 반복문 진행

4. List 내부 최댓값과 A배열 인덱스 최솟값 곱하여 result 에 증감

5. List 내부 최댓값 제거하여 반복문 횟수 줄임

 

문제

옛날 옛적에 수학이 항상 큰 골칫거리였던 나라가 있었다. 이 나라의 국왕 김지민은 다음과 같은 문제를 내고 큰 상금을 걸었다.

길이가 N인 정수 배열 A와 B가 있다. 다음과 같이 함수 S를 정의하자.

S = A[0] × B[0] + ... + A[N-1] × B[N-1]

S의 값을 가장 작게 만들기 위해 A의 수를 재배열하자. 단, B에 있는 수는 재배열하면 안 된다.

S의 최솟값을 출력하는 프로그램을 작성하시오.

입력

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

출력

첫째 줄에 S의 최솟값을 출력한다.

예제 입력 1 복사

5
1 1 1 6 0
2 7 8 3 1

예제 출력 1 복사

18

예제 입력 2 복사

3
1 1 3
10 30 20

예제 출력 2 복사

80

예제 입력 3 복사

9
5 15 100 31 39 0 0 3 26
11 12 13 2 3 4 5 9 1

예제 출력 3 복사

528

힌트

예제 1의 경우 A를 {1, 1, 0, 1, 6}과 같이 재배열하면 된다.

출처

• 문제를 번역한 사람: baekjoon

알고리즘 분류

• 수학
• 그리디 알고리즘
• 정렬

 

 

A배열을 오름차순으로 하고.. 결국 B 배열을 내림차순하여 풀었던 코드이다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());

        int num = Integer.parseInt(st.nextToken());
        int[] first_array = new int[num];
        int[] second_array = new int[num];

        int result = 0;
        for(int i = 0; i < 2; i++){
            st = new StringTokenizer(br.readLine());
            for(int j = 0; j < num; j++){
                if(i == 0){
                    first_array[j] = Integer.parseInt(st.nextToken());
                }else{
                    second_array[j] = Integer.parseInt(st.nextToken());
                }
            }
        }

        // 오름차순
        Arrays.sort(first_array);

        // 내림차순
        Integer[] second_array2 = Arrays.stream(second_array).boxed().toArray(Integer[]::new);
        Arrays.sort(second_array2, Collections.reverseOrder());

        for(int i = 0; i < num; i++){
            result += (first_array[i] * second_array2[i]);
        }

        System.out.println(result);
    }
}

 

 

 

여기는 반례를 찾지 못하고 풀었던 코드이다. 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());

        int num = Integer.parseInt(st.nextToken());
        int[] first_array = new int[num];
        int[] second_array = new int[num];
        List<Integer> list = new ArrayList<>();

        int result = 0;
        for(int i = 0; i < 2; i++){
            st = new StringTokenizer(br.readLine());
            for(int j = 0; j < num; j++){
                if(i == 0){
                    first_array[j] = Integer.parseInt(st.nextToken());
                }else{
                    second_array[j] = Integer.parseInt(st.nextToken());
                    list.add(j, second_array[j]);
                }
            }
        }

        Arrays.sort(first_array);

        int cnt = 0;
        while (list.size() != 0){
            int max = 0;
            int temp = 0;
            for (int i = 0; i < list.size(); i++){
                if(temp < list.get(i)){
                    max = list.get(i);
                    temp = list.get(i);
                }else{
                    temp = list.get(i);
                }
            }
            result += max * first_array[cnt];
            list.remove(list.indexOf(max));
            cnt++;
        }
        System.out.println(result);
    }
}