[BOJ 9019] DSLR(Python)
문제
문제 해설
문제에서 요구한대로 구현을 시작했고 단순하게 bfs를 이용하여 계산된 결과가 답과 같으면 종료되는 식으로 구현을 했지만 예상대로 시간초과가 났다. 구현하면서도 깊이가 깊어질수록 중복된 값이 당연히 많아지고 그것에 대한 처리가 필요했다. dp를 이용하여 이미 저장되어있는 값이면 넘어가는 형태로 구현했다. 그래도 비정상적으로 시간이 오래걸렸는데 bfs함수에서 계산처리를 동시에하면 시간이 더 단축될 것이다.
시간초과 코드
from collections import deque
q = deque()
resultset = []
command = ["D", "S", "L", "R"]
T = int(input())
for i in range(T):
start, result = input().split()
result = int(result)
resultset.append([start, result])
def calculate(string, com):
temp = string.zfill(4)
# D
if com == 0:
temp = int(string)
if temp * 2 > 10000:
return str(temp * 2 - 10000)
return str(temp * 2)
# S
elif com == 1:
temp = int(string)
if temp == 0:
return str(9999)
return str(temp - 1)
# L
elif com == 2:
return temp[1:4] + temp[0]
# R
else:
return temp[3] + temp[:3]
for i in range(T):
answer = ""
# 초기값 큐에 삽입
for j in range(4):
out = calculate(resultset[i][0], j)
q.append([out, command[j]])
while q:
size = len(q)
for j in range(size):
string, com = q.popleft()
if int(string) == resultset[i][1]:
answer += com
q.clear()
break
for k in range(4):
out = calculate(string, k)
q.append([out, com + command[k]])
print(answer)
풀이 코드
from collections import deque
resultset = []
dp = [0] * 10000
command = ["D", "S", "L", "R"]
T = int(input())
for i in range(T):
start, result = input().split()
result = int(result)
resultset.append([start, result])
def calculate(string, com):
temp = string.zfill(4)
# D
if com == 0:
temp = int(string)
if temp * 2 >= 10000:
return str(temp * 2 % 10000)
return str(temp * 2)
# S
elif com == 1:
temp = int(string)
if temp == 0:
return str(9999)
return str(temp - 1)
# L
elif com == 2:
return temp[1:4] + temp[0]
# R
else:
return temp[3] + temp[:3]
def bfs():
q = deque()
answer = ""
for j in range(4):
out = calculate(resultset[i][0], j)
q.append([out, command[j]])
while q:
string, com = q.popleft()
if int(string) == resultset[i][1]:
answer += com
return answer
for j in range(4):
out = calculate(string, j)
if dp[int(out)] > 0:
continue
dp[int(out)] += 1
q.append([out, com + command[j]])
for i in range(T):
print(bfs())
for j in range(10000):
dp[j] = 0
Author And Source
이 문제에 관하여([BOJ 9019] DSLR(Python)), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://velog.io/@qweadzs/BOJ-9019-DSLRPython저자 귀속: 원작자 정보가 원작자 URL에 포함되어 있으며 저작권은 원작자 소유입니다.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)