문제 풀이/Programmers

181893. 배열 조각하기 자바

공부하고 기억하는 공간 2023. 7. 31. 02:56
728x90
반응형
SMALL

ArrayList를 사용하여 조건에 맞게 add와 remove를 사용한다.

import java.util.ArrayList;
class Solution {
    public int[] solution(int[] arr, int[] query) {
        ArrayList<Integer> list = new ArrayList<Integer>();
	for(int i: arr) list.add(i);	
	for(int i=0; i<query.length;i++) {
		if(i%2==0) {
			int start = query[i]+1;
			int cnt = list.size()-start;
			for(int j=start; cnt!=0;cnt--) {
				list.remove(j);
			}	
					}
		else {
			int end = query[i];
			int cnt = end;
			for(int x=0; cnt!=0;cnt--) list.remove(x);
			}//else
		}//for
	
	
	int[] answer = list.stream().mapToInt(Integer::intValue).toArray();
        
        return answer;
    }
}

728x90
반응형
SMALL