[JAVA] 프로그래머스: 체육복 빌리기

keep it simple 2023. 3. 27. 00:39

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

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 public int solution(int n, int[] lost, int[] reserve) {
          int answer = 0;
          int[] arr = new int[n];
          for(int i=0;i<n;i++) {
               arr[i] = 1;
          }

          for(int l:lost) {
              arr[l-1]--;
          }
          for(int r:reserve) {
              arr[r-1]++;
          }


          for(int i=0;i<n;i++) {
             if(arr[i] == 0) {
                 if(i >0&&arr[i-1] == 2) {
                     arr[i-1]--;
                     arr[i]++;
                 }
                 else if(i < n-1&&arr[i+1] == 2) {
                     arr[i]++;
                     arr[i+1]--;
                 }

             }
          }
          for(int i : arr) {
              if(i > 0) {
                  answer++;
              }
          }
        return answer;
}