Walks on a Triangle — Solution & Editorial

Back to the Problem

Model the count as a linear state transition (a fixed matrix acting on the current state vector) and raise the matrix to the N-th power in O(log N).

Complexity: O(log N)

Watch out: Collapse the two non-start vertices by symmetry into one state.