[ST]Programmers - 기능개발 - Java

 

https://programmers.co.kr/learn/courses/30/lessons/42586

코딩테스트 연습 - 기능개발

 

스택/큐 문제이다.

그냥 흘러가는데로 풀어서 좋은 코드인지는 모르겠다.

나중에 다시 수정해봐야지

import java.util.*;
class Solution {
  public int[] solution(int[] progresses, int[] speeds) {
    int[] cnt = new int[progresses.length];
    for (int i = 0; i < progresses.length; i++) {
      int pro = progresses[i];
      while (pro < 100) {
        pro += speeds[i];
        cnt[i]++;
      }
    }

    ArrayList<Integer> list = new ArrayList<Integer>();
    list.add(1);
    int start = 0;
    for (int i = 1; i < cnt.length; i++) {
      if (cnt[start] >= cnt[i]) {
        list.set(list.size() - 1, list.get(list.size() - 1) + 1);
      } else {
        list.add(1);
        start = i;
      }
    }
    int[] answer = new int[list.size()];
    for (int i = 0; i < list.size(); i++) {
      answer[i] = list.get(i);
    }
    return answer;
  }
}