/ SeriousOJ /

Record Detail

Runtime Error


  
# Status Time Cost Memory Cost
#1 Runtime Error Traceback (most recent call last): File "foo.py", line 22, in <module> File "foo.py", line 14, in max_number IndexError: list index out of range 36ms 2.871 MiB

Code

def max_number(arr, k):
    n = len(arr)
    dp = [[[-float('inf')] * (k+1) for _ in range(n)] for _ in range(n)]

    # Base cases
    for i in range(n):
        dp[i][i][0] = arr[i]

    for l in range(2, n+1):
        for i in range(n-l+1):
            j = i + l - 1
            for k in range(1, k+1):
                for p in range(k+1):
                    dp[i][j][k] = max(dp[i][j][k], dp[i][i+p-1][p] * 10**(j-i-p+1) + dp[i+p][j][k-p])

    return dp[0][n-1][k]

t = int(input())
for _ in range(t):
    n, k = map(int, input().split())
    arr = list(map(int, input().split()))
    print(max_number(arr, k))

Information

Submit By
Type
Pretest
Problem
P1083 Number concatenation
Language
Python 3 (Python 3.12.3)
Submit At
2024-08-16 15:58:13
Judged At
2024-08-16 15:58:13
Judged By
Score
0
Total Time
36ms
Peak Memory
2.871 MiB