코딩테스트 문제를 풀다보면 위치문제가 자주 나온다. 즉 input으로 어떤 위치정보들이 들어가고 그걸 기반으로 시뮬레이션같은 것을 진행하여 그 결과의 위치들의 상태를 output으로 내는 것이다.
이때 유용한게 리스트 내포 (list comprehension) 이다. 예를 들어, 3*3 바둑판이 있다고 하자.
이때 0이 흰돌, 1이 검은돌이라고 한다면 d = [[0 for i in range(3)] for j in range(3)] 과 같이 선언하면 3*3 개의 흰돌이 바둑판위에 있는 모습을 저장할 수 있다. 즉, d는 [[0,0,0],[0,0,0],[0,0,0]] 과 같이 생겼을텐데 첫번째 [0,0,0]을 바둑판 첫줄 이라고 생각하고 두번째 [0,0,0]을 바둑판의 두번째 줄, 세번째 [0,0,0]을 바둑판의 세..........
원문 링크 : [코딩테스트] 파이썬으로 위치정보 기록하기.