[프로그래머스] 다리를 지나는 트럭 - 파이썬 (Python)

2021. 4. 6. 00:01· 알고리즘/문제풀이
목차
  1. 문제 설명
  2. 문제 풀이
  3. 코드 설명
728x90

문제 설명

트럭 여러 대가 강을 가로지르는 일 차선 다리를 정해진 순으로 건너려 합니다. 모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 알아내야 합니다. 트럭은 1초에 1만큼 움직이며, 다리 길이는 bridge_length이고 다리는 무게 weight까지 견딥니다.
※ 트럭이 다리에 완전히 오르지 않은 경우, 이 트럭의 무게는 고려하지 않습니다.

예를 들어, 길이가 2이고 10kg 무게를 견디는 다리가 있습니다. 무게가 [7, 4, 5, 6]kg인 트럭이 순서대로 최단 시간 안에 다리를 건너려면 다음과 같이 건너야 합니다.

 

경과 시간 다리를 지난 트럭 다리를 건너는 트럭 대기 트럭
0 [] [] [7,4,5,6]
1~2 [] [7] [4,5,6]
3 [7] [4] [5,6]
4 [7] [4,5] [6]
5 [7,4] [5] [6]
6~7 [7,4,5] [6] []
8 [7,4,5,6] [] []

따라서, 모든 트럭이 다리를 지나려면 최소 8초가 걸립니다.

solution 함수의 매개변수로 다리 길이 bridge_length, 다리가 견딜 수 있는 무게 weight, 트럭별 무게 truck_weights가 주어집니다. 이때 모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 return 하도록 solution 함수를 완성하세요.


제한 조건

  • bridge_length는 1 이상 10,000 이하입니다.
  • weight는 1 이상 10,000 이하입니다.
  • truck_weights의 길이는 1 이상 10,000 이하입니다.
  • 모든 트럭의 무게는 1 이상 weight 이하입니다.

입출력 예

 

bridge_length weight truck_weights return
2 10 [7,4,5,6] 8
100 100 [10] 101
100 100 [10,10,10,10,10,10,10,10,10,10] 110

 

문제 풀이

def solution(bridge_length, weight, truck_weights):
	answer = 0
    queue = [0] * bridge_length
    
    while queue:
    	queue.pop(0)
        answer += 1
        
        if truck_weights:
        	if sum(queue) + truck_weights[0] <= weight:
            	queue.append(truck_weights.pop(0))
            else:
            	queue.append(0)
                
	return answer

 

코드 설명

해당 문제의 핵심은 다리를 다음과 같이 표현하는 것이다.

 [0] * bridge_length 

1초마다 다리의 맨 앞의 값을 pop해주고, 다리에 새로운 트럭이 올라올 수 있으면 해당 트럭의 값을 다리의 맨 뒤의 값에, 그렇지 않다면 0을 맨 뒤의 값에 append해준다.

 

그렇게 모든 트럭이 다리를 지나게 되면 if truck_weights: 의 값이 False가 되고, 마지막 트럭이 다리를 지나면 자연스레 queue의 값도 모두 pop되어 while문이 종료되고 answer가 리턴된다.

 

코드에서는 collections모듈의 deque를 사용하지 않고 리스트를 사용해 풀었지만, 사실 맨 앞의 값을 pop하는 작업을 주로 수행하기 때문에 효율성 측면에서는 deque를 import해 사용하는 것이 더욱 효율적이라고 할 수 있다.

저작자표시 (새창열림)

'알고리즘 > 문제풀이' 카테고리의 다른 글

[백준] 1697번 숨바꼭질 - 파이썬(Python)  (0) 2021.04.20
[프로그래머스] 모의고사 - 파이썬(Python)  (0) 2021.04.19
[프로그래머스] 기능개발 - 파이썬(Python)  (0) 2021.02.19
[백준] 7576번 토마토 - 파이썬(Python)  (0) 2021.02.07
[백준] 7562번 나이트의 이동 - 파이썬(Python)  (0) 2021.02.06
  1. 문제 설명
  2. 문제 풀이
  3. 코드 설명
'알고리즘/문제풀이' 카테고리의 다른 글
  • [백준] 1697번 숨바꼭질 - 파이썬(Python)
  • [프로그래머스] 모의고사 - 파이썬(Python)
  • [프로그래머스] 기능개발 - 파이썬(Python)
  • [백준] 7576번 토마토 - 파이썬(Python)
SeongOnion
SeongOnion
서버는 꺼지지 않아요
SeongOnion
조무래기 코딩
SeongOnion
전체
오늘
어제
  • 분류 전체보기 (167)
    • 알고리즘 (81)
      • 이론 (8)
      • 문제풀이 (73)
    • 언어 (15)
      • Python (9)
      • JavaScript (1)
      • JAVA (5)
    • 데이터베이스 (5)
    • 프레임워크 (15)
      • Django (7)
      • Spring (8)
    • 그 외 공부 (38)
      • 운영체제 (1)
      • 자료구조 (14)
      • 네트워크 (5)
      • CS (2)
      • 기타 (7)
      • 트러블 슈팅 (9)
    • 프로젝트 (0)
    • 개발자취 (8)
    • 회고 (3)
    • 주저리주저리 (1)
    • 기타 (비개발) (1)

블로그 메뉴

  • 홈
  • 태그
  • 방명록

공지사항

인기 글

태그

  • 컨트리뷰트
  • 자바
  • 정렬 알고리즘
  • 소수
  • 백준
  • 코딩테스트
  • 이진탐색
  • 스택
  • DRF
  • 코딩
  • 장고
  • 투 포인터 알고리즘
  • 큐
  • 데이터베이스
  • 트러블 슈팅
  • 프로그래머스
  • 그리디알고리즘
  • DP
  • 웹
  • 브루트포스
  • 파이썬
  • BFS/DFS
  • spring
  • BFS
  • 오픈소스
  • 알고리즘
  • 에라토스테네스의 체
  • 개발자
  • Django
  • 회고

최근 댓글

최근 글

hELLO · Designed By 정상우.v4.2.2
SeongOnion
[프로그래머스] 다리를 지나는 트럭 - 파이썬 (Python)
상단으로

티스토리툴바

단축키

내 블로그

내 블로그 - 관리자 홈 전환
Q
Q
새 글 쓰기
W
W

블로그 게시글

글 수정 (권한 있는 경우)
E
E
댓글 영역으로 이동
C
C

모든 영역

이 페이지의 URL 복사
S
S
맨 위로 이동
T
T
티스토리 홈 이동
H
H
단축키 안내
Shift + /
⇧ + /

* 단축키는 한글/영문 대소문자로 이용 가능하며, 티스토리 기본 도메인에서만 동작합니다.