[백준/Java] 카드 놓기 18115

 

 

 

문제

https://www.acmicpc.net/problem/18115

18115번: 카드 놓기

 

 

 

 

 

 

 

 

 

풀이 

처음에 ArrayList를 썼는데 시간 초과가 났다

다음은 배열 순회해서 풀었는데도 시간 초과

다음은 index로 접근하는 걸로 풀었다. 

그랬더니 통과

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        int N = Integer.parseInt(br.readLine());
        int[] list = new int[N];
        int[] input = new int[N];

        StringTokenizer st = new StringTokenizer(br.readLine());
        for (int i = 0; i < N; i++) {
            input[i] = Integer.parseInt(st.nextToken());
        }

        int index1 = 0;
        int index2 = 1;
        int index3 = N - 1;

        for (int i = 0, j = N; i < N; i++, j--) {
            switch (input[i]) {
                case 1:
                    list[index1] = j;
                    index2++;
                    index1 = index2 - 1;
                    break;
                case 2:
                    list[index2] = j;
                    index2++;
                    break;
                case 3:
                    list[index3] = j;
                    index3--;
                    break;
            }
        }


        for (int i = 0; i < list.length; i++) {
            bw.write(String.valueOf(list[i]) + " ");
        }

        bw.flush();
        bw.close();
    }
}

 

 

 

 

 

 

정답