The program finds the maximum number of "A"s that can be printed on the screen given a set of keyboard commands and a number of times the keyboard can be pressed.

Problem #1.py

# THis code Finds the max 'A' count with given keyboard presses.

def max_a(n):
    # initialize dp array with size n + 1
    dp = [0] * (n + 1)
    
    # loop through the dp array
    for i in range(1, n + 1):
    
        # assign the maximum number of 'A' for dp[i]
        dp[i] = dp[i - 1] + 1
        j = 2
        while i - j >= 0:
            dp[i] = max(dp[i], dp[i - j] * (j + 1))
            j += 1
    
    # return the maximum number of 'A'
    return dp[n]