https://www.acmicpc.net/problem/5430
시간 1초, 메모리 256MB
input :
- 테스트 케이스의 개수 T가 주어진다. (1 <= T <= 100)
- 수행할 함수 p(1 <= p의 길이 <= 100000)
- 수의 개수 n(0 ≤ n ≤ 100,000)
- [x1,...,xn]과 같은 형태로 배열에 들어있는 수
output :
- 함수를 수행한 결과를 출력한다. 만약, 에러가 발생한 경우에는 error를 출력
조건 :
- AC는 정수 배열에 연산을 하기 위해 만든 언어이다. 이 언어에는 두 가지 함수 R(뒤집기)과 D(버리기)가 있다.
- R은 배열에 있는 숫자의 순서를 뒤집는 함수이고, D는 첫 번째 숫자를 버리는 함수이다. 배열이 비어있는데 D를 사용한 경우에는 에러
수행해야 하는 것이 우선 배열을 수로 나누어야 한다. 이때 그냥 split(',')를 할 경우에 괄호가 포함 되니까 이를 제외 하기 위해 [1:-1]로 슬라이싱을 해야 한다.
그리고 strip()을 하지 않으면 제일 뒤에 개행문자가 존재해서 안된다.
그렇다면 인제 함수에 맞게 행동을 하게 하면 되는데 이 때 reverse()와 del을 이용해 할 경우 시간 복잡도가 많이 소요 된다.
그래서 r = 짝수 -> 뒤집지 않은 상태, r = 홀수 -> 뒤집은 상태로 생각을 하고 이 때에 맞춰 앞에서 제거를 하는지 뒤에서 제거를 하는지를 기록을 한다.
그리고 데이터의 수 보다 제거하는 횟수가 작다면 여전히 배열에 값이 존재하니까 이 값을 출력한다. 이때 r의 값이 짝, 홀 인지 체크를 한 후 뒤집어서 출력을 할 지 판단한다.
그것이 아니라면 error을 출력한다.
import sys
t = int(sys.stdin.readline())
for i in range(t):
command = sys.stdin.readline().strip().replace('RR', '')
n = int(sys.stdin.readline())
data = sys.stdin.readline().strip()[1:-1].split(",")
r, f, b = 0, 0, 0
for item in command:
if item == 'R':
r += 1
else:
if r % 2 == 0:
f += 1
else:
b += 1
if f + b <= n:
data = data[f: n - b]
if r % 2 == 1:
print('[' + ','.join(data[::-1]) + ']')
else:
print('[' + ','.join(data) + ']')
else:
print('error')
