[알고리즘] 수열 편집

no.13501

package swea.solution5122;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;
import java.util.StringTokenizer;


public class Solution {
	static StringBuilder sb = new StringBuilder();
	
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st;
		int T = Integer.parseInt(br.readLine());
		for (int testCase = 1; testCase <= T; testCase++) {
			st = new StringTokenizer(br.readLine());
			int N = Integer.parseInt(st.nextToken());
			int M = Integer.parseInt(st.nextToken());
			int L = Integer.parseInt(st.nextToken());
			
			st = new StringTokenizer(br.readLine());
			List<Integer> numbers = new ArrayList<>();
			for (int i = 1; i <= N; i++) {
				numbers.add(Integer.parseInt(st.nextToken()));
			}
			
			for (int i = 0; i < M; i++) {
				st = new StringTokenizer(br.readLine());
				int idx = 0;
				int number = 0;
				switch (st.nextToken()) {
				case "I":
					idx = Integer.parseInt(st.nextToken());
					number = Integer.parseInt(st.nextToken());
					numbers.add(idx, number);
					break;
				case "D":
					idx = Integer.parseInt(st.nextToken());
					numbers.remove(idx);
					break;
				case "C":
					idx = Integer.parseInt(st.nextToken());
					number = Integer.parseInt(st.nextToken());
					numbers.set(idx, number);
					break;
				
				}
			}
			sb.append("#").append(testCase).append(" ");
			if (L >= numbers.size()) {
				sb.append(-1);
			} else {
				sb.append(numbers.get(L));
			}
			sb.append("\n");
		}
		
		System.out.println(sb);
	}
}

나의 풀이

- ArrayList를 쓰면 되는 문제

- ArrayList에는 다음과 같은 기능들이 들어 있다.

- 특정 인덱스에 원소 추가

- 특정 인덱스 삭제

- 특정 인덱스 검색