[프로그래머스] 알고리즘 연습 문제 : 체육복 (자바/JAVA)

https://programmers.co.kr/learn/courses/30/lessons/42862?language=java

알고리즘 연습 - 체육복 | 프로그래머스

 

import org.junit.Test;
 
import java.util.Arrays;
import java.util.Optional;
 
import static org.hamcrest.CoreMatchers.*;
import static org.hamcrest.MatcherAssert.assertThat;
 
public class SolutionTest {
 
    @Test
    public void 체육복문제_그리디알고리즘() {
 
        //given
        int n = 5;
        int[] lost = {2,4};
        int[] reserve = {3};
 
        //when
        int result = solution(n, lost, reserve);
 
        //then
        assertThat(result, is(4));
 
    }
 
    public int solution(int n, int[] lost, int[] reserve) {
        int unLuckyPeople = 0;
        Arrays.sort(lost);
        Arrays.sort(reserve);
 
        //여분 유니폼을 가져온 애꺼 훔쳐간 경우
        for(int i = 0; i < lost.length; i++) {
            for (int j = 0; j < reserve.length; j++) {
                if(lost[i] == reserve[j]) {
                    lost[i] = -1;
                    reserve[j] = -1;
                    break;
                }
            }
        }ㅋ
        
        for(int i = 0; i < lost.length; i++) {
            for(int j = 0; j < reserve.length; j++) {
                if(hasNotReserveUniform(lost[i],reserve[j])) {        //여분 유니폼 있는지 확인 안해도 되면 다음 번호로
                    continue;
                }
                if(hasReserveUniform(lost[i], reserve[j])) {        //여분 유니폼 빌렸을 경우
                    lost[i] = 0;
                    reserve[j] = 0;
                    break;
                }
            }
            if(lost[i] > 0) {
                unLuckyPeople++;
            }
        }
 
        return n - unLuckyPeople;
    }
 
    private boolean hasNotReserveUniform(int lost, int reserve) {
        if(lost-1 > reserve || lost+1 < reserve) {
            return true;
        }
        return false;
    }
 
    private boolean hasReserveUniform(int lost, int reserve) {
        if(lost-1 == reserve || lost+1 == reserve){
            return true;
        }
        return false;
    }
 
}

 

짧게! 짧게!!! 보다는 알아보기 쉽게! 가독성 최대로!

를 목표로 했는데... 둘다 실패했다.

잃어버린 학생 번호의 앞뒤만 확인하려고 과몰입하다보니 시야가 좁아진 것일까...

반복문 중첩은 O(N^2)이어서 최대한 이중 반복문은 피하려고도 했는데,

이것도 고민하다가 그냥 써버렸다.

좋아요를 많이 받은 코드는 훨씬 간단했다.

    public int solution(int n, int[] lost, int[] reserve) {
        int answer = 0;
        answer = n;
 
        for(int i = 0; i < lost.length; i++) {
            boolean rent = false;
            int j = 0;
            while(!rent) {
                if(j == reserve.length)                   break;
                if(lost[i] == reserve[j])                {reserve[j] = -1; rent=true;}
                else if(lost[i] - reserve[j] == 1 )      {reserve[j] = -1; rent=true;}
                else if(lost[i] - reserve[j] == -1)      {reserve[j] = -1; rent=true;}
                else                                     {j++;                      }
            }
            if(!rent) answer--;
        }
        return answer;
    }

잃어버린 학생 - 여분 가져온 학생

해서 확인하면 되는 것을 미련했다.