99클럽 코테 스터디 5일차 TIL (챌린저): [백준][Java] 2457 공주님의 정원 - 골드3

 

문제 보기

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

 

 

풀이

1. 정렬 방법

    flowers.sort(new Comparator<int[]>() {
      @Override
      public int compare(int[] o1, int[] o2) {
        if(o1[2] == o2[2]) {
          return o2[3] - o1[3];
        }
        return o2[2] - o1[2];
      }
    });

시작, 끝이 주어지기 때문에 정렬 기준을 무엇으로 해야 할지 감이 안 잡힐 수 있다.

나는 끝나는 날짜를 기준으로 내림차순 정렬을 했다.

유효기한이 긴 순으로 살펴보며 개화 시기가 이전 꽃이 진 날 보다 작거나 같은 것을 찾았다.

 

 

2. 비교 방법

int prevMonth = 3;
int prevDay = 1;
int count = 0;
while (prevMonth < 12) {
  for(int i = 0; i < flowers.size(); i++) {
    if(prevMonth > flowers.get(i)[0] || (prevMonth == flowers.get(i)[0] && prevDay >= flowers.get(i)[1])) {
      count++;
      prevMonth = flowers.get(i)[2];
      prevDay = flowers.get(i)[3];
      flowers.remove(flowers.get(i));
      break;
    }
    if(i == flowers.size() - 1) {
      count = 0;
    }
  }
  if(count == 0 || (prevMonth < 12 && flowers.isEmpty())) {
    count = 0;
    break;
  }
}

끝나는 날짜가 그날까지 핀다가 아니라 그전 날까지 핀다기 때문에 유의해야 한다.

11월 30일 까지가 조건이기 때문에 마지막 지는 날은 무조건 12월이 넘어야 한다.

이전 꽃이 진 시기를 저장해놓고 계속 비교해 가며 카운트하였다.

 

꽃을 마지막까지 순회했는데도 다음 꽃을 찾지 못했거나

모든 꽃을 다 소비했을 경우 count를 0으로 하고 빠져나오도록 했다.

 

 

전체 코드

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Objects;

public class Main {

  public static void main(String[] args) throws IOException {

    BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

    int n = Integer.parseInt(br.readLine());
    List<int[]> flowers = new ArrayList();
    while (n-- > 0) {
      String[] inputs = br.readLine().split(" ");
      flowers.add(new int[]{
          Integer.parseInt(inputs[0]), Integer.parseInt(inputs[1]), Integer.parseInt(inputs[2]), Integer.parseInt(inputs[3])});
    }

    flowers.sort(new Comparator<int[]>() {
      @Override
      public int compare(int[] o1, int[] o2) {
        if(o1[2] == o2[2]) {
          return o2[3] - o1[3];
        }
        return o2[2] - o1[2];
      }
    });

    int prevMonth = 3;
    int prevDay = 1;
    int count = 0;
    while (prevMonth < 12) {
      for(int i = 0; i < flowers.size(); i++) {
        if(prevMonth > flowers.get(i)[0] || (prevMonth == flowers.get(i)[0] && prevDay >= flowers.get(i)[1])) {
          count++;
          prevMonth = flowers.get(i)[2];
          prevDay = flowers.get(i)[3];
          flowers.remove(flowers.get(i));
          break;
        }
        if(i == flowers.size() - 1) {
          count = 0;
        }
      }
      if(count == 0 || (prevMonth < 12 && flowers.isEmpty())) {
        count = 0;
        break;
      }
    }

    bw.write(count + "\n");

    bw.flush();
    bw.close();

  }
}

 

 

GitHub 링크

https://github.com/MetroDefro/CodingTest_AutoSave/tree/main/%EB%B0%B1%EC%A4%80/Gold/2457.%E2%80%85%EA%B3%B5%EC%A3%BC%EB%8B%98%EC%9D%98%E2%80%85%EC%A0%95%EC%9B%90

CodingTest_AutoSave/백준/Gold/2457. 공주님의 정원 at main · MetroDefro/CodingTest_AutoSave