본문 바로가기
PS/BOJ

[자바] 백준 6799 - Computer Purchase (java)

by Nahwasa 2022. 9. 8.

 문제 : boj6799


 

필요 알고리즘 개념

  • 정렬
    • 원하는 방식으로 정렬을 할 수 있으면 쉽게 풀 수 있다.

※ 제 코드에서 왜 main 함수에 로직을 직접 작성하지 않았는지, 왜 Scanner를 쓰지 않고 BufferedReader를 사용했는지 등에 대해서는 '자바로 백준 풀 때의 팁 및 주의점' 글을 참고해주세요. 백준을 자바로 풀어보려고 시작하시는 분이나, 백준에서 자바로 풀 때의 팁을 원하시는 분들도 보시는걸 추천드립니다.

 


 

풀이

  N개의 데이터를 입력받은 후, SQL로 따지면 ORDER BY value DESC, name ASC로 정렬한 뒤 가장 앞에 있는 2개를 순서대로 출력해주면 되는 정렬문제이다. 여기서 value는 2 × R + 3 × S + D를 뜻하고, name은 computer name을 뜻한다. 즉, 2 × R + 3 × S + D이 큰 순서대로 출력하되, 해당 값이 동일한 경우엔 이름이 사전순으로 빠른걸 먼저 출력해주면 된다.

 

  자바의 경우 Comparable의 compareTo를 아래와 같이 구현해주면 된다.

@Override
public int compareTo(Computer o) {
    if (o.factor == this.factor)
        return this.name.compareTo(o.name);
    return o.factor - this.factor;
}

 

  혹은 Arrays.sort 시, new Comparator를 구현해서 넣어줘도 된다.

 


 

코드 : github

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

class Computer implements Comparable<Computer> {
    String name;
    int factor;
    public Computer(String name, int r, int s, int d) {
        this.name = name;
        this.factor = 2*r+3*s+d;
    }

    @Override
    public int compareTo(Computer o) {
        if (o.factor == this.factor)
            return this.name.compareTo(o.name);
        return o.factor - this.factor;
    }
}

public class Main {
    private void solution() throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());
        Computer[] arr = new Computer[n];
        for (int i = 0; i < n; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            arr[i] = new Computer(st.nextToken(), Integer.parseInt(st.nextToken()), Integer.parseInt(st.nextToken()), Integer.parseInt(st.nextToken()));
        }
        if (n == 0) {
            return;
        }
        if (n == 1) {
            System.out.println(arr[0].name);
            return;
        }
        Arrays.sort(arr);
        System.out.println(arr[0].name + "\n" + arr[1].name);
    }
    public static void main(String[] args) throws Exception {
        new Main().solution();
    }
}

댓글