[문제 풀이] 프로그래머스 - 체육복(java)

프로그래머스 - 수열과 구간 쿼리2(java)

---

프로그래머스 문제 풀이 입니다 !

오류가 있다면 댓글로 남겨주세요 :)

 

1. 처음 제출한 코드

2. 다음 제출한 코드

---

1. 처음 제출한 코드

import java.util.Arrays;

class Solution {
    public int solution(int n, int[] lost, int[] reserve) {
        int answer = n - lost.length;
        
        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] - 1 == reserve[j] || lost[i] + 1 == reserve[j]) {
                    answer++;
                    reserve[j] = -1;
                    break;
                }
            }
        }
        
        return answer;
    }
}

 

 

문제를 보고 앞 사람과 뒷 사람에게 빌릴 수 있는 경우를 생각하여 위와 같이 코드를 작성했다.

테스트 실행했을 때는 잘 작동이 되는데 제출을 했을 때는 테스트 결과에서 실패했다.

어떤 경우를 고려하지 않은건지에 대해 생각하고, 조건도 다시 살펴봤다.

그랬는데도 잘 모르겠어서 계속 고민하다가 검색을 해봤더니 답이 나왔다.

 

2. 다음 제출한 코드

import java.util.Arrays;

class Solution {
    public int solution(int n, int[] lost, int[] reserve) {
        int answer = n - lost.length;
        
        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]) {
					answer++;
					lost[i] = -1;
					reserve[j] = -1;
                    break;
				}
			}
		}
        
        for (int i = 0; i < lost.length; i++) {
            for (int j = 0; j < reserve.length; j++) {    
                if (lost[i] - 1 == reserve[j] || lost[i] + 1 == reserve[j]) {
                    answer++;
                    reserve[j] = -1;
                    break;
                }
            }
        }
        
        return answer;
    }
}

 

"여분이 있는 사람도 잃어버릴 수가 있다." 라는 조건이 있었는데, 2번 학생이 여분의 체육복을 가지고 있는데 2번 학생이 가져온 여분의 체육복을 잃어버린 경우에 대해서 생각치 못했다. 그래서 그 경우에 대한 코드를 작성해주니까 성공적으로 문제를 해결할 수 있었다. 예상치 못한 곳에서 오류가 발생하는데 꼼꼼하게 생각해야겠다.