[python] List padding

청천·2022년 8월 12일

python

목록 보기
1/3

누적합 문제를 풀면 인덱스 에러에 주의 해야한다.
이때 리스트 패딩을 이용하면 쉽게 문제에 접근 할 수 있다.

여기서 이야기 하는 리스트 패딩은 2차원 리스트 가로축 혹은 세로축에 값을 추가하는 것을 말한다.
대게 리스트 인덱스 0에 0을 더해준다.( 0은 누적합에 영향을 끼치지 않기 때문이다.)

N = int(input())
arr = [list(map(int, input().split()) for _ in range(N)]
'''이차원 리스트 패딩, 세로축'''
arr = [[0] + list(map(int, input().split()) for _ in range(N)]
'''이차원 리스트 패딩, 세로축 + 가로축'''
arr = [[0]*(N+1)] + [[0] + list(map(int, input().split()) for _ in range(N)]

2차원 리스트 패딩 연습하기
https://www.acmicpc.net/problem/11659
https://www.acmicpc.net/problem/11660

0개의 댓글