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

문제

https://school.programmers.co.kr/learn/courses/30/lessons/12927

프로그래머스

 

풀이

import java.util.*;

class Solution {
     public long solution(int n, int[] works) {
        long answer = 0;
        long sum = 0;

        for(int i=0; i<works.length; i++){
            sum += works[i];
        }

        sum -= n;

        if(sum < 0){
            return 0;
        }

        if(sum % works.length == 0){
            long a = sum / works.length;
            for(int i=0; i< works.length; i++){
                answer += Math.pow(a, 2);
            }
        }else{
            long a = sum / works.length;
            long b = sum % works.length;

            for(int i=0; i< works.length; i++){
                if(b>0){
                    answer += Math.pow(a+1, 2);
                    b--;
                }else{
                    answer += Math.pow(a, 2);
                }
            }

        }

        return answer;
    }
}