Problem p03013 (python)

Fixed submission:
#%%
import sys
def input():
    return sys.stdin.readline().rstrip()

def main():
    N, M = map(int, input().split())
    a_list = [int(input()) for _ in range(M)]

    p = 0
    steps=[]
    m = 0
    for a in a_list:
        if a != 1:
            steps.append((p, a-1))
            m = max(m, a-p)
        p = a + 1
    else:
        steps.append((p, N))
        m = max(m, N-p+1)

    dp=[0 for _ in range(m+1)]

    divider = 1000000007
    dp[0], dp[1] = 1, 1
    for i in range(2,m+1):
        dp[i]=(dp[i-1]+dp[i-2])%divider

    ans = 1
    for p, n in steps:
        if n < p:
            ans = 0
            break
        else:
            r = dp[n-p]
        ans = (ans*r)%divider 

    print(ans)
        

# %%
if __name__ == '__main__':
    main()

# %%
# from atcoder_test import doTest
# doTest("abc129","c",main)

Buggy submission (verdict: Runtime Error):
#%%
import sys
def input():
    return sys.stdin.readline().rstrip()

def main():
    N, M = map(int, input().split())
    a_list = [int(input()) for _ in range(M)]

    dp=[0 for _ in range(N+1)]

    divider = 1000000007
    dp[0], dp[1] = 1, 1
    for i in range(2,N+1):
        dp[i]=(dp[i-1]+dp[i-2])%divider

    p = 0
    ans = 1
    for a in a_list:
        r = dp[a-1-p] if a-1 >= p else 0
        ans = (ans*r)%divider 
        p = a + 1
    else:
        if a < N:
            r = dp[N-p]%divider
            ans = (ans*r)%divider

    print(ans)
        

# %%
if __name__ == '__main__':
    main()

# %%
# from atcoder_test import doTest
# doTest("abc129","c",main)

Buggy submission (verdict: Runtime Error):
#%%
import sys
def input():
    return sys.stdin.readline().rstrip()

def comb(N):
    if N<0:
        return 0
    if N==0 or N==1:
        return 1

    dp=[[0 for _ in range(N+1)] for _ in range(3)]

    for j in range(1,N+1):
        for i in [1,2]:
            if j == i:
                dp[i][j] = 1
            else:
                dp[i][j] = dp[1][j-i] + dp[2][j-i]
    return dp[1][N]+dp[2][N]

def main():
    N, M = map(int, input().split())
    a_list = [int(input()) for _ in range(M)]

    divider = 1000000007
    p = 0
    ans = 1

    for a in a_list:
        r = comb(a-1-p)%divider
        ans = ans%divider 
        ans = (ans*r)%divider 
        p = a + 1
    else:
        if a < N:
            r = comb(N-p)%divider
            ans = ans%divider
            ans = (ans*r)%divider

    print(ans)
        

# %%
if __name__ == '__main__':
    main()

# %%
# from atcoder_test import doTest
# doTest("abc129","c",main)

Buggy submission (verdict: Runtime Error):
#%%
import sys
def input():
    return sys.stdin.readline().rstrip()

def main():
    N, M = map(int, input().split())
    a_list = [int(input()) for _ in range(M)]

    def comb(N):
        if N<0:
            return 0
        if N==0:
            return 1

        dp=[[0 for _ in range(N+1)] for _ in range(3)]

        for j in range(1,N+1):
            for i in [1,2]:
                if j == i:
                    dp[i][j] = 1
                else:
                    dp[i][j] = dp[1][j-i] + dp[2][j-i]
        return dp[1][N]+dp[2][N]

    divider = 1000000007
    p = 0
    ans = 1

    for a in a_list:
        r = comb(a-1-p)%divider
        ans = ans%divider 
        ans = (ans*r)%divider 
        p = a + 1
    else:
        if a < N:
            r = comb(N-p)%divider
            ans = ans%divider
            ans = (ans*r)%divider

    print(ans)
        

# %%
if __name__ == '__main__':
    main()

# %%
# from atcoder_test import doTest
# doTest("abc129","c",main)

Buggy submission (verdict: Runtime Error):
#%%
import sys
def input():
    return sys.stdin.readline().rstrip()

def main():
    N, M = map(int, input().split())
    a_list = [int(input()) for _ in range(M)]

    def comb(N):
        if N<0:
            return 0
        if N==0:
            return 1

        dp=[[0 for _ in range(N+1)] for _ in range(3)]

        for j in range(1,N+1):
            for i in [1,2]:
                if j == i:
                    dp[i][j] = 1
                else:
                    dp[i][j] = dp[1][j-i] + dp[2][j-i]
        return dp[1][N]+dp[2][N]

    divider = 1000000007
    p = 0
    ans = 1

    for a in a_list:
        r = comb(a-1-p)%divider
        ans = (ans*r)%divider 
        p = a + 1
    else:
        if a < N:
            r = comb(N-p)%divider
            ans = (ans*r)%divider

    print(ans)
        

# %%
if __name__ == '__main__':
    main()

# %%
# from atcoder_test import doTest
# doTest("abc129","c",main)

Buggy submission (verdict: Runtime Error):
#%%
import sys
def input():
    return sys.stdin.readline().rstrip()

def main():
    N, M = map(int, input().split())
    a_list = [int(input()) for _ in range(M)]

    def comb(N):
        if N<0:
            return 0
        if N==0:
            return 1
        dp=[[0 for _ in range(N+1)] for _ in range(3)]
        for j in range(1,N+1):
            dp[0][j] = 1
            for i in [1,2]:
                if j == i:
                    dp[i][j] = dp[i-1][j]
                else:
                    dp[i][j] = dp[1][j-i] + dp[2][j-i]
        return dp[1][N]+dp[2][N]
    p = 0
    divider = 1000000007
    ans = 1
    for a in a_list:
        r = comb(a-1-p)%divider
        ans = ans * r
        p = a + 1
    else:
        if a < N:
            r = comb(N-p)%divider
            ans = ans * r
    ans = ans%divider
    print(ans)
        

# %%
if __name__ == '__main__':
    main()

# %%
# from atcoder_test import doTest
# doTest("abc129","c",main)

Buggy submission (verdict: Runtime Error):
#%%
import sys
def input():
    return sys.stdin.readline().rstrip()

def main():
    N, M = map(int, input().split())
    a_list = [int(input()) for _ in range(M)]

    def comb(N):
        if N<0:
            return 0
        if N==0:
            return 1
        import numpy as np
        dp=np.zeros((3, N+1), dtype=np.int64)
        for j in range(1,N+1):
            dp[0][j] = 1
            for i in [1,2]:
                if j == i:
                    dp[i][j] = dp[i-1][j]
                else:
                    dp[i][j] = dp[1][j-i] + dp[2][j-i]
        return dp[1][N]+dp[2][N]
    p = 0
    divider = 1000000007
    ans = 1
    for a in a_list:
        r = comb(a-1-p)%divider
        ans = ans * r
        p = a + 1
    else:
        if a < N:
            r = comb(N-p)%divider
            ans = ans * r
    ans = ans%divider
    print(ans)
        

# %%
if __name__ == '__main__':
    main()

# %%
# from atcoder_test import doTest
# doTest("abc129","c",main)