[프로그래머스] 야근지수 Java

[프로그래머스] level3 야근지수

 

소요시간 : 15분

 

우선.. n이 너무 커서 반복문으로 밖에 못하겠다고 생각했고,

정렬하고 하나씩 -1을 하면 너무 오래 걸릴 것 같았다.

 

제곱을 더하는 것이니 최대한 큰 숫자를 줄여야한다고 생각했다.

그래서 우선 배열을 정렬하여 가장 큰 값 여러개를 다 -1씩하고

다시 정렬하는 방법을 생각했다.

가장 큰 값 여러개를 -1씩 하다가 n 이 다끝나면 그냥 종료하고

그게 아니라면 다시 정렬해서 또 가장 큰 값 여러개들을 다 -1 한다.

 

isZero 변수는 단지 모두 0일 때 연산을 좀 줄여주자일 뿐 별 의미 x

 

import java.util.Arrays;

class Solution {
  public long solution(int n, int[] works) {
    long answer = 0;
    boolean isZero = false;
out:while(true) {
        Arrays.sort(works);
        int large = works[works.length-1];
        if(large == 0) {
          isZero = true;
          break;
        }
        int idx = works.length-1;
        while(true) {
          if(n == 0) break out;
          if(idx < 0) break;
          if(large == works[idx]) {
          works[idx--]--;
          n--;
          }
          else break;
      	}

	}
    if(!isZero) {
      for (int i = 0; i < works.length; i++) {
      	answer += works[i] * works[i];
      }
    }
    return answer;
  }
}