알고리즘/Java

백준 알고리즘 13305번: 주유소

두넌 2024. 4. 16.

문제 정보


 

13305번: 주유소

표준 입력으로 다음 정보가 주어진다. 첫 번째 줄에는 도시의 개수를 나타내는 정수 N(2 ≤ N ≤ 100,000)이 주어진다. 다음 줄에는 인접한 두 도시를 연결하는 도로의 길이가 제일 왼쪽 도로부터 N-1

www.acmicpc.net

 

핵심


너무 복잡하게 생각하면 안된다

이 문제의 핵심은 당연하게도 가장 싼 주유소에서 주유를 하는 것이다

나는 처음에 이 문제를 어렵게 접근해서 풀었다 (결국 풀었지만)

 

첫번째 풀이

        int[][] station = new int[N][2];
        int remain = 0;
        for(int i=N-1; i>=0; i--) {
            remain += distance[i];
            station[i][0] = remain;
            station[i][1] = price[i];
        }

        Arrays.sort(station, (o1, o2) -> o1[1]-o2[1]);

        long result = 0;
        int current = 0;
        for(int[] st: station) {
            if(st[0] > current) {
                int buy = st[0] - current;
                result = result + (long) buy * st[1];
                current += buy;
            }
        }

해당 주유소를 기점으로, 그 뒤에 있는 지역까지의 거리 만큼을 미리 주유하려고 remain 이라는 해당 위치에서 남은 거리를 기록해 두었다

그리고, 가격을 기준으로 정렬한 뒤에 가격이 적은 순으로 만약 내가 지나온 거리보다 더 남았다면 (st[0] > current)

남은 거리(buy) 만큼을 해당 주유소에서 결제한 기름으로 이동한다

 

결국 풀었지만, 일단 효율적인 풀이는 아니었다는 생각이 들어서 다른 분들의 해설을 참고하여 다시 풀어냈던 것 같다

 

두번째 풀이

        long result = 0;
        int minPrice = Integer.MAX_VALUE;
        for(int i=0; i<N-1; i++) {
            minPrice = Math.min(minPrice, price[i]);
            result += minPrice * (long) distance[i];
        }
        System.out.println(result);

단순히 전체에서의 최소를 구할 필요도 없다

그냥 현재까지의 최소 가격으로 한칸 한칸 이동하면 된다

 

 

풀이


첫번째 풀이

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

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine()) - 1;
        int[] distance = Arrays.stream(br.readLine().split(" ")).mapToInt(Integer::parseInt).toArray();
        int[] price = Arrays.stream(br.readLine().split(" ")).mapToInt(Integer::parseInt).toArray();

        int[][] station = new int[N][2];
        int remain = 0;
        for(int i=N-1; i>=0; i--) {
            remain += distance[i];
            station[i][0] = remain;
            station[i][1] = price[i];
        }

        Arrays.sort(station, (o1, o2) -> o1[1]-o2[1]);

        long result = 0;
        int current = 0;
        for(int[] st: station) {
            if(st[0] > current) {
                int buy = st[0] - current;
                result = result + (long) buy * st[1];
                current += buy;
            }
        }
        System.out.println(result);
    }
}

 

두번째 풀이

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

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());
        int[] distance = Arrays.stream(br.readLine().split(" ")).mapToInt(Integer::parseInt).toArray();
        int[] price = Arrays.stream(br.readLine().split(" ")).mapToInt(Integer::parseInt).toArray();

        long result = 0;
        int minPrice = Integer.MAX_VALUE;
        for(int i=0; i<N-1; i++) {
            minPrice = Math.min(minPrice, price[i]);
            result += minPrice * (long) distance[i];
        }
        System.out.println(result);
    }
}

 

 

Source Code on GitHub


GitHub에서 소스코드 보기

 

 

댓글