기본 콘텐츠로 건너뛰기

프로그래머스 - 야근 지수

좀 지저분하긴해도 시간초과 까지 개선한 코드.. N^2으로 풀어도 5초가 넘어서(정답 처리는 됨) 좀더 최적화 해봄
import java.util.Arrays;

public class NoOvertime {
    public int noOvertime(int no, int[] works) {
        int result = 0;
        Arrays.sort(works);
        int i = works.length-1;
        Loop1: while(i!=0 && no!=0) {
            if(works[i]>works[i-1]) {
                for(int j=i;j<works.length;j++) {
                    if(no==0) {
                        break Loop1;
                    }
                    works[j]--;
                    no--;
                }            
            continue Loop1;
            }
            i--;
        }

        while(no!=0) {
            works[no--%works.length]--;         
        }
        for(int j=0;j<works.length;j++) {
            result+=works[j]*works[j];
        }

        return result;
    }
    public static void main(String[] args) {
        NoOvertime c = new NoOvertime();
        int []test = {4,3,3};
        System.out.println(c.noOvertime(4,test));
    }
}

댓글

이 블로그의 인기 게시물

프로그래머스 - 최대값과 최소값

import java.util.Vector ; public class GetMinMaxString { public String getMinMaxString( String str) { Integer min= Integer .MAX_VALUE,max= Integer .MIN_VALUE; String [] sepstr = str.split( " " ); Vector < Integer > vec = new Vector < Integer >(); for ( int i= 0 ;i<sepstr.length;i++) { vec.addElement( Integer .parseInt(sepstr[i])); } for ( int i= 0 ;i<vec.size();i++) { if (vec.get(i)<min) { min = vec.get(i); } if (vec.get(i) > max) { max = vec.get(i); } } return min+ " " +max; } public static void main( String [] args) { String str = " 1 2 3 4 " ; GetMinMaxString minMax = new GetMinMaxString(); //아래는 테스트로 출력해 보기 위한 코드입니다. System .out.println( " 최대값과 최소값은? " + minMax.getMinMaxString(str)); } } 아래와...