백준 문제 해결 시 이용할 파이썬 입력 코드!

SamSim·2021년 8월 11일
0

알고리즘문제해결

목록 보기
1/1

거진 2년만에 백준에서 문제를 풀었다.
문제 해결과정보다는, 백준의 코드 제출 및 채점방식에 애를 먹었다.
백준에서는 input값이 사용자 입력으로 주어지고, 출력 또한 print해주는 방식으로 해야한다.
반면 프로그래머스에서는 주어진 함수로 input값이 매개변수로 입력되며, return해주는 방식으로 결과값을 전달한다.
특히 최근 해결한 RGB거리 문제는, 입력값이 줄로 함께 주어지는데, 이를 한줄입력 받을 함수를
별도로 만들어두어 추후에 사용하고자 하였다!

def arrayinput():
    n=int(input())
    costs=[]
    costs.append(n)
    for r in range(n):
        temp=list(map(int,input().split()))
        costs.append(temp)
    return costs

먼저 가장 앞에 주어지는 행 수를 받고, 각 행을 input으로 받는데, 문자열의 형태로 받게 되므로 split()해준 뒤, list의 형태로 map해주었다. (첫번째 줄 행 수를 받을 때에도 문자의 형태로 받아지므로 int로 바꾸어 주어야한다) 그리고 이걸 가장 첫줄과 함께 list에 넣어주면... 끝!
이 코드를 사용하면 [행 수 ,[1행],[2행].....]의 형태로 나오게 된다!

여담

최근에 해결한 RGB거리 문제에 대한 내 해결법은
rgb거리 문제 해결 깃허브에 있다! readme를 이용해 내 해결법과 문제 해결과정에서 겪은 어려운 점, 고민한 점들을 설명해 두었다.(프로그래머스 문제를 풀 때 부터 늘 해왔지만...)

profile
개발새발

0개의 댓글