115752024-10-27 12:11:31balintCsúcsokpython3Wrong answer 0/100245ms19528 KiB
def main():
    N, M = map(int, input().split())
    A = list(map(int, input().split()))
    B = list(map(int, input().split()))

    # Create the height matrix H[i][j] = A[i] * B[j]
    # heights = [[A[i] + B[j] for j in range(M)] for i in range(N)]
    peaks = 0
    def height(i, j):
        return (A[i] + B[j] if 0 <= i < N and 0 <= j < M else 0)
    
    def is_peak(a,b,c,d,e,f,g,h,i):
        print("checking: ", e)
        return e > a and e > b and e > c and e > d and e > f and e > g and e > h and e > i

    
    # Iterate through each cell to check for peaks
    a, b, c, d, e, f, g, h, i = 0, 0, 0, 0, height(0,0), height(0,1), 0, height(1,0), height(1,1) 
    peaks += is_peak(a,b,c,d,e,f,g,h,i)

    # Iterate from top to bottom (each iteration left to right, down, right to left, down)
    for N_i in range(0, N, 2):
        # scroll right
        for M_i in range(M-1):
            a,b,c = b,c, height(N_i-1, M_i+2)
            d,e,f, = e,f, height(N_i, M_i+2)
            g,h,i = h,i, height(N_i+1, M_i+2)

            peaks += is_peak(a,b,c,d,e,f,g,h,i)

        # reached end at bottom right
        if height(N_i+2, M_i+2) == 0:
            break

        # scroll down
        a,b,c = d,e,f
        d,e,f = g,h,i
        g,h,i = height(N_i+2, M_i), height(N_i+2, M_i+1), height(N_i+2, M_i+2)
        peaks += is_peak(a,b,c,d,e,f,g,h,i)

        # scroll left
        for M_i in range(M_i, -1, -1):
            a,b,c = height(N_i, M_i-1),a,b
            d,e,f, = height(N_i+1, M_i-1), d,e
            g,h,i = height(N_i+2, M_i-1), g,h

            peaks += is_peak(a,b,c,d,e,f,g,h,i)

        # reached end at bottom left
        if height(N_i+3, M_i-1) == 0:
            break

        # scroll down
        a,b,c = d,e,f
        d,e,f = g,h,i
        g,h,i = height(N_i+3, M_i-1), height(N_i+3, M_i), height(N_i+3, M_i+1)
        peaks += is_peak(a,b,c,d,e,f,g,h,i)

# Call the main function
main()
SubtaskSumTestVerdictTimeMemory
subtask10/0
1Wrong answer17ms3128 KiB
2Wrong answer17ms3128 KiB
subtask20/16
3Wrong answer204ms15308 KiB
4Wrong answer202ms13552 KiB
5Wrong answer193ms6468 KiB
6Wrong answer211ms15264 KiB
7Wrong answer196ms15128 KiB
8Wrong answer200ms14776 KiB
subtask30/33
9Wrong answer18ms3344 KiB
10Wrong answer17ms3384 KiB
11Wrong answer17ms3384 KiB
12Wrong answer18ms3576 KiB
13Runtime error17ms3276 KiB
14Wrong answer16ms3284 KiB
15Wrong answer17ms3384 KiB
subtask40/51
16Wrong answer244ms18848 KiB
17Wrong answer215ms15696 KiB
18Wrong answer238ms19100 KiB
19Wrong answer48ms15392 KiB
20Wrong answer217ms15388 KiB
21Wrong answer245ms18968 KiB
22Wrong answer232ms19424 KiB
23Wrong answer232ms18312 KiB
24Wrong answer244ms19528 KiB