글 읽기 - [자바] 코드 로직을 한번 봐주실 수 있을까요?

mommoo 3년 전 0

주가의 최대 값을 찾아서, 최대값 이전에는 주가를 팔고

남은날에서도 위와 같이 최대 값을 찾아 최대 값 이전에 주가를 파는 식으로 코드를 작성했습니다.

왜 답이 틀리다고 나오는걸까요?

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
import java.util.Objects;
import java.util.PriorityQueue;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws Exception {
        BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(System.out));
        int N = Integer.parseInt(reader.readLine());

        while (N -- > 0) {
            long count = 0;
            long sum = 0;
            long profit = 0;

            reader.readLine(); // N drop

            List<Long> values = createValues(new StringTokenizer(reader.readLine()));
            PriorityQueue<Long> maxValues = createMaxValues(values);
            Long maxValue = maxValues.poll();
            for (Long value : values) {
                if (value < maxValue) {
                    count++;
                    sum += value;
                } else if (Objects.equals(value, maxValue)) {
                    profit += value * count - sum;
                    count = 0;
                    sum = 0;

                    if (!maxValues.isEmpty()) {
                        maxValue = maxValues.poll();
                    }
                }
            }

            writer.append(String.valueOf(profit));
            writer.newLine();
        }

        writer.flush();
    }

    private static List<Long> createValues(StringTokenizer tokenizer) {
        List<Long> values = new ArrayList<>();
        while (tokenizer.hasMoreTokens()) {
            values.add(Long.parseLong(tokenizer.nextToken()));
        }
        return values;
    }

    private static PriorityQueue<Long> createMaxValues(List<Long> values) {
        PriorityQueue<Long> maxValues = new PriorityQueue<>(Comparator.reverseOrder());
        maxValues.addAll(values);
        return maxValues;
    }
}

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

[자바] 코드 로직을 한번 봐주실 수 있을까요?

mommoo 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말