[SWEA/d2] 1940. 가랏! RC카! (Python)

📚 문제 링크

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV5PjMgaALgDFAUq&categoryId=AV5PjMgaALgDFAUq&categoryType=CODE&problemTitle=1940&orderBy=FIRST_REG_DATETIME&selectCodeLang=ALL&select-1=&pageSize=10&pageIndex=1

SW Expert Academy

 

물1 물2 한 사람으로서 좀 .. 문제가 물리 법칙에 안 맞긴 하다.

그래도 어쩌겠나 풀어야지.

풀이

 

---

 

📒 코드

# 가랏! RC카!
import sys
sys.stdin = open("input.txt", "r")

T = int(input())
for tc in range(1, T+1):
    N = int(input()) # 커맨드의 수
    v = 0 # 속도
    d = 0 # 거리
    
    for i in range(N):
        arr = list(map(int, input().split())) # 1 2 / 2 2 / 0
        if arr[0] == 1: # 가속
            v += arr[1]
        elif arr[0] == 2: # 감속
            v -= arr[1]
            if v < 0:
                v = 0
        d += v
        
    print("#{} {}" .format(tc, d))

 

한 커맨드에서 두 변수를 입력받을 때도 있고, 한 변수를 입력받을 때도 있으니(0, 속도 유지인 경우)

변수를 지정해서 입력을 받을 수 없음. 따라서 배열로 입력을 받아줬다. (원하는만큼 입력받기)

 

속도를 v라 한 다음, 가속도(?)에 따라 속도를 지정해줌.

그러고는 거리 d 변수에 저장해주기.

 

그런데 9, 10번 테스트 케이스가 틀렸다고 나와서 다시 확인해보니

제약사항3 (3. 현재 속도보다 감속할 속도가 더 클 경우, 속도는 0 m/s 가 된다.)을 적용해주지 않아서였다.

감속할 경우, v가 음수되면 0으로 바꿔주는 조건문을 추가하여 해결했다.

 

---

💡 새롭게 알게 된 지식

• 입력을 원하는 만큼, 유동적으로 받고 싶다면 배열로 받아주자 !

arr = list(map(int, input().split()))

 

 

---

🔎 결과