[백준(BOJ)] 20055번 : 컨베이어 벨트 위의 로봇 - PYTHON[파이썬]

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

20055번: 컨베이어 벨트 위의 로봇

 

---

⊙ 문제

⊙ 입력

⊙ 출력

⊙ 예제 입출력

⊙ 알고리즘 분류

• 구현
• 시뮬레이션

 

---

 

⊙ 문제 접근 과정

 

파이썬에서 회전 -> deque 라이브러리에 있는 rotate 함수 사용

 

1. 무한 반복문 생성

2. rotate(1) : 한 칸 회전

3. 로봇 내리는 위치라면 내리기

4. 이동 가능 여부 체킹

5. 가능하다면 이동 후 내구도 감소

6. 로봇 내리는 위치라면 내리기

7. 로봇 올리는 위치 내구도 체킹 후 로봇 올리기

8. 만약 내구도 0의 개수가 K개 이상일 시 STOP

9. 아니라면 2부터 다시 반복

 

---

 

⊙ 문제 풀이

 

import sys
from collections import deque

input = sys.stdin.readline
N, K = map(int, input().split())
A = deque(map(int, input().split()))
robots = deque([0 for _ in range(N)])
result = 0

while True:
    # 회전!
    A.rotate(1)
    robots.rotate(1)

    # 로봇 내리는 위치
    robots[N-1] = 0

    # 이동 여부 확인!
    for i in range(N-2, -1, -1):
        # 이동할 수 있다면?
        if (robots[i] and not robots[i+1] and A[i+1] >= 1):
            # 다음칸으로 이동하면서 내구도 -1
            A[i+1] -= 1
            robots[i+1] = robots[i]
            robots[i] = robots[0]

    # 로봇 내리는 위치
    robots[N-1] = 0

    # 로봇 올리는 위치
    if A[0] >= 1:
        A[0] -= 1
        robots[0] = 1
    result += 1

    # 내구도 0인 칸 개수 체킹
    if A.count(0) >= K:
        break

print(result)

 

---

⊙ 결과

 

---

⊙ 마무리

 

파이썬 회전문제 = rotate

 

 

좋아요는 로그인하지 않아도 누를 수 있습니다!