Matrix-Free Matvec Count

Problem

In a matrix-free iterative solver, count total matrix-vector products with A.

Given a list of iteration counts for consecutive solves, each solve uses one matvec per iteration plus one extra to form the initial residual. Return the total count.

Hint

Each solve pays one residual computation plus one product per iteration.

Test cases

case 1

in: [[10,5]]

expected: 17

case 2

in: [[0]]

expected: 1

case 3

in: [[3]]

expected: 4

case 4

in: [[1,1,1]]

expected: 6

Starter codePython

def matvec_count(iterations_per_solve):
    # Your code here
    pass

4 test cases · run your code against them in the editor