[Baekjoon] # 5430: AC

AC

문제

선영이는 주말에 할 일이 없어서 새로운 언어 AC를 만들었다.
AC는 정수 배열에 연산을 하기 위해 만든 언어이다.
이 언어에는 두 가지 함수 R(뒤집기)과 D(버리기)가 있다.

함수 R은 배열에 있는 수의 순서를 뒤집는 함수이고, 
D는 첫 번째 수를 버리는 함수이다. 
배열이 비어있는데 D를 사용한 경우에는 에러가 발생한다.

함수는 조합해서 한 번에 사용할 수 있다. 
예를 들어, "AB"는 A를 수행한 다음에 바로 이어서 B를 수행하는 함수이다.
예를 들어, "RDD"는 배열을 뒤집은 다음 처음 두 수를 버리는 함수이다.

배열의 초기값과 수행할 함수가 주어졌을 때, 최종 결과를 구하는 프로그램을 작성하시오.

예제

• 입력

첫째 줄에 테스트 케이스의 개수 T가 주어진다. T는 최대 100이다.

각 테스트 케이스의 첫째 줄에는 수행할 함수 p가 주어진다.

p의 길이는 1보다 크거나 같고, 100,000보다 작거나 같다.

다음 줄에는 배열에 들어있는 수의 개수 n이 주어진다. (0 ≤ n ≤ 100,000)

다음 줄에는 [x1, …, xn]과 같은 형태로 배열에 들어있는 정수가 주어진다. (1 ≤ xi ≤ 100)

전체 테스트 케이스에 주어지는 p의 길이의 합과 n의 합은 70만을 넘지 않는다.

4
RDD
4
[1,2,3,4]
DD
1
[42]
RRD
6
[1,1,2,3,5,8]
D
0
[]

• 출력

각 테스트 케이스에 대해서, 입력으로 주어진 정수 배열에 함수를 수행한 결과를 출력한다.

만약, 에러가 발생한 경우에는 error를 출력한다.

[2,1]
error
[1,2,3,5,8]
error

---

Solution

def get_inputs():
    import sys

    input_data = sys.stdin.read()

    input_lines = input_data.splitlines()

    inputs = []
    for i in range(1, len(input_lines), 3):
        inputs.append(
            [
                # functions
                input_lines[i],
                # number of data
                int(input_lines[i + 1]),
                # data array
                (
                    []
                    if input_lines[i + 2] == "[]"
                    else list(map(int, input_lines[i + 2][1:-1].split(",")))
                ),
            ]
        )

    return inputs


def solution(n: list[str, int, list]):
    func = n[0]
    length_of_data = n[1]
    data = n[2]

    # `Reverse` 상태를 저장
    is_reverse = False
    for f in func:
        if f == "R":
            is_reverse = not is_reverse
        else:
            # 삭제할 data가 없으면 "error"
            if not data:
                return "error"
            
            # 뒤집어져있으면 끝을 삭제
            if is_reverse:
                data.pop()
            # 뒤집어져있지 않으면 앞을 삭제
            else:
                data.pop(0)

    if is_reverse:
        data.reverse()

    return "[" + ",".join(map(str, data)) + "]"


def main():
    inputs = get_inputs()

    result = [solution(n) for n in inputs]

    print(*result, sep="\n", end="")


if __name__ == "__main__":
    main()

---

해결 방법

두 연산에 대한 이해가 필요했다.

Reverse를 하여도 Delete가 되는 위치가 달라질 뿐,

굳이 Reverse를 할 필요가 없었다.

따라서 Reverse가 되었는지 안 되었는지 변수에 상태를 저장하고,

뒤집어져있는 상태라면 마지막 요소를 제거,

뒤집어져있지 않다면 첫 요소를 제거하여

모든 연산을 진행한다.

그리고 마지막으로 상태에 따라 뒤집는 연산을 한 번 수행하면 된다.

의사 코드

함수 solution(입력값):
    뒤집기_상태 = False

    for 각 명령어 in 명령어 배열:
        만약 명령어가 "R"이면:
            뒤집기_상태 = 반전

        그렇지 않으면:
            만약 데이터_배열이 비어있으면:
                "error" 반환
            
            만약 뒤집기_상태가 True이면:
                데이터_배열의 마지막 요소 제거
            그렇지 않으면:
                데이터_배열의 첫번째 요소 제거

    만약 뒤집기_상태가 True이면:
        데이터_배열 뒤집기

---

URL

• AC