문제 풀이/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