백준 16967 배열 복원하기 / python

문제 : 16967

풀이 point

B배열은 A배열을 겹처서 만든 것이다. 겹치는 부분은 공식에 의해 해결될 수 있다.

풀이 방법

겹치는 부분은
겹치는 부분 - 원래 A배열의 값을 하면 된다.

원래 A배열의 값을 어떻게 아냐? 이미 A배열은 밑으로 X만큼 내려오기 때문에 X만큼 차이가 난다. 때문에 B배열의 X위치 전 값은 A배열과 동일하다.

코드

import sys
input = sys.stdin.readline

H, W, X, Y = map(int, input().rstrip().split())

B = [list(map(int, input().rstrip().split())) for _ in range(H+X)]

for i in range(H):
    for j in range(W):
        B[i+X][j+Y] -= B[i][j]
        print(B[i][j], end=' ')
    print()