[그리디] 백준 13305 주유소

13305번: 주유소

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

 

# 코드

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

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());
        StringTokenizer st;

        long[] distance = new long[n-1];
        st = new StringTokenizer(br.readLine(), " ");
        for (int i = 0; i < n-1; i++) {
            distance[i] = Long.parseLong(st.nextToken());
        }

        long[] oil_price = new long[n];
        st = new StringTokenizer(br.readLine(), " ");
        for (int i = 0; i < n; i++) {
            oil_price[i] = Long.parseLong(st.nextToken());
        }

        long cost = 0; // 총 비용
        int destination = n-1; // 목적지
        long min_price = Long.MAX_VALUE; // 최솟값
        int min_loc = Integer.MAX_VALUE; // 최솟값의 위치

        while (destination != 0) {
            // 목적지 전까지 기름이 가장 저렴한 곳 찾기
            for (int i = 0; i < destination; i++) {
                if (oil_price[i] < min_price) {
                    min_price = oil_price[i];
                    min_loc = i;
                }
            }

            // min_loc ~ 목적지 어치 기름을 전부 구매
            long rest_distance_sum = 0; // 목적지까지 남은 거리
            for (int k = min_loc; k < destination; k++) {
                rest_distance_sum += distance[k];
            }
            cost += rest_distance_sum * min_price;

            // 목적지 재설정 + 변수 초기화
            destination = min_loc;
            min_price = Long.MAX_VALUE;
            min_loc = Integer.MAX_VALUE;
        }

        System.out.println(cost);
        br.close();
    }
}

• 기름값이 가장 저렴한 곳에 도착했을 때 최종 목적지까지 남은 거리의 모든 기름을 구매한다면 [가장 싼 곳 ~ 최종 목적지] 까지는 최소 비용을 보장한다는 해결법에서 고안했다.
• 가장 저렴한 곳까지 도달하기 위해서도 같은 방식으로 해결하여 뒤에서부터 비용을 계산하는 로직이다.
• 내가 짠돌이라면 ...??!! 더 쉽게 해결할 수 있다. ( $ ω $ )✧

 

 

하지만 내 코드는 매 범위마다 최솟값을 새롭게 찾기 때문에 시간복잡도가 매우 높다는 문제점이 있다.

2500 ms ..... !

 

 

따라서 문제의 본질을 파악해 계산 과정을 줄이는 방법을 생각해내야 한다.

이미 짠 로직에 갇혀 코드짜기가 쉽지 않았는데, 구글링해보니 기름 가격이 내림차순일 경우에만 주유하면 된다고 한다.

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

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());

        long[] dist = new long[N - 1];	// 거리
        long[] cost = new long[N];	// 비용

        StringTokenizer st = new StringTokenizer(br.readLine(), " ");
        for (int i = 0; i < N - 1; i++) {
            dist[i] = Long.parseLong(st.nextToken());
        }

        st = new StringTokenizer(br.readLine(), " ");
        for (int i = 0; i < N; i++) {
            cost[i] = Long.parseLong(st.nextToken());
        }

        long sum = 0;
        long minCost = cost[0];	// 이전 까지의 과정 중 주유 최소 비용

        for (int i = 0; i < N - 1; i++) {
            /*
             *  현재 주유소가 이전 주유소의 기름값보다 쌀 경우
             *  minCost를 갱신해준다.
             */
            if(cost[i] < minCost) {
                minCost = cost[i];
            }

            sum += (minCost * dist[i]);
        }

        System.out.println(sum);
        br.close();
    }
}

 

 

채점해보면 차이가 많이 나는 것을 볼 수 있다.

거의 8배 차이난다

 

 

참고한 블로그는 👇🏻 매번 설명을 잘해주셔서 신세지고 있다.

[백준] 13305번 : 주유소 - JAVA [자바]