AI013. 单通道卷积特征图

# 朴素卷积:先零补边构造新矩阵,再按尺寸公式 (H+2P-K)/S+1 跑四层循环
# i、j 是输出格坐标,a、b 是核内偏移,取 padded[i*S+a][j*S+b]
import sys


def main():
    data = sys.stdin.buffer.read().split()
    at = 0
    H = int(data[at]); at += 1
    W = int(data[at]); at += 1
    K = int(data[at]); at += 1
    S = int(data[at]); at += 1
    P = int(data[at]); at += 1
    nums = list(map(int, data[at:]))
    X = [nums[i * W:(i + 1) * W] for i in range(H)]
    ker = [nums[H * W + i * K:H * W + (i + 1) * K] for i in range(K)]
    PH, PW = H + 2 * P, W + 2 * P
    pad = [[0] * PW for _ in range(PH)]
    for i in range(H):
        pad[i + P][P:P + W] = X[i]
    OH = (PH - K) // S + 1
    OW = (PW - K) // S + 1
    lines = []
    for i in range(OH):
        base_r = i * S
        row = []
        for j in range(OW):
            base_c = j * S
            acc = 0
            for a in range(K):
                pr = pad[base_r + a]
                kr = ker[a]
                for b in range(K):
                    acc += pr[base_c + b] * kr[b]
            row.append(acc)
        lines.append(" ".join(map(str, row)))
    sys.stdout.write("\n".join(lines) + "\n")


main()