Problem p03426 (python)

Fixed submission:
def calc(prepoint, nextpoint):
    return abs(nextpoint[0]-prepoint[0]) + abs(nextpoint[1]-prepoint[1])

def main():
    H,W,D = map(int,input().split())
    index  = [(0,0) for i in range(H*W+1)]
    A = []
    for i in range(H):
        temp = list(map(int, input().split()))
        for t in temp:
            index[t] = (i,temp.index(t))
        A.append(temp)
    Q = int(input())
    L_R_List = [list(map(int, input().split())) for i in range(Q)]

    dp = [0 for i in range(H*W+1)]
    for i in range(D, H*W+1):
        dp[i] = dp[i-D] + calc(index[i], index[i-D])
    for L_R in L_R_List:
        L = L_R[0]
        R = L_R[1]
        print(dp[R]-dp[L])

if __name__ == '__main__':
    main()

Buggy submission (verdict: Runtime Error):
def calc(prepoint, nextpoint):
    return abs(nextpoint[0]-prepoint[0]) + abs(nextpoint[1]-prepoint[1])

def search(L,A,H):
    for h in range(H):
        if L in A[h]:
            return (h,A[h].index(L))
def main():
    H,W,D = map(int,input().split())
    A = [list(map(int, input().split())) for i in range(H)]
    Q = int(input())
    L_R_List = [list(map(int, input().split())) for i in range(Q)]

    temp = [(0,0) for i in range(H*W)]
    for L_R in L_R_List:
        L = L_R[0]
        R = L_R[1]
        ans = 0
        if temp[L] == (0,0):
            temp[L] = search(L,A,H)
            prepoint = temp[L]
        else:
            prepoint = temp[L]
        while(L != R):
            L += D
            if temp[L] == (0,0):
                temp[L] = search(L,A,H)
                nextpoint = temp[L]
            else:
                nextpoint = temp[L]
            ans += calc(prepoint, nextpoint)
            prepoint = nextpoint
        print(ans)
if __name__ == '__main__':
    main()

Buggy submission (verdict: Runtime Error):
def calc(prepoint, nextpoint):
    return abs(nextpoint[0]-prepoint[0]) + abs(nextpoint[1]-prepoint[1])

def search(i,A):
    for h in range(H):
        if L in A[h]:
            return (h,A[h].index(L))
def main():
    H,W,D = map(int,input().split())
    A = [list(map(int, input().split())) for i in range(H)]
    Q = int(input())
    L_R_List = [list(map(int, input().split())) for i in range(Q)]

    temp = [(0,0) for i in range(H*W)]
    for L_R in L_R_List:
        L = L_R[0]
        R = L_R[1]
        ans = 0
        if temp[L] == (0,0):
            temp[L] = search(L,A)
            prepoint = temp[L]
        else:
            prepoint = temp[L]
        while(L != R):
            L += D
            if temp[L] == (0,0):
                temp[L] = search(L,A)
                nextpoint = temp[L]
            else:
                nextpoint = temp[L]
            ans += calc(prepoint, nextpoint)
            prepoint = nextpoint
        print(ans)
if __name__ == '__main__':
    main()

Buggy submission (verdict: Runtime Error):
def calc(prepoint, nextpoint):
    return abs(nextpoint[0]-prepoint[0]) + abs(nextpoint[1]-prepoint[1])

def search(i,A):
    for h in range(H):
        if L in A[h]:
            return (h,A[h].index(L))

H,W,D = map(int,input().split())
A = [list(map(int, input().split())) for i in range(H)]
Q = int(input())
L_R_List = [list(map(int, input().split())) for i in range(Q)]

temp = [(0,0) for i in range(H*W)]
for L_R in L_R_List:
    L = L_R[0]
    R = L_R[1]
    ans = 0
    if temp[L] == (0,0):
        temp[L] = search(L,A)
        prepoint = temp[L]
    else:
        prepoint = temp[L]
    while(L != R):
        L += D
        if temp[L] == (0,0):
            temp[L] = search(L,A)
            nextpoint = temp[L]
        else:
            nextpoint = temp[L]
        ans += calc(prepoint, nextpoint)
        prepoint = nextpoint
    print(ans)