Solve Matrix Equations (Also for AI/ML Interns :)

Problem Overview

  • Find the 1x2 vector A such that X = A * B * C, given scalar X, 2x2 matrix B, and 2x1 vector C.
  • Implement findA(X, B, C) and return [A[0], A[1]] as floats.
  • Constraint: A[1] = 1 - A[0].
  • Linear algebra context relevant to data and ML workflows.
  • From Snowflake interviews: a coding interview problem and interview question for AI/ML interns.

Example

Unlock to view complete problem details

and practice with sample input/output

Was this article helpful?

View Test Cases & Run Code requires membership

Input Variables
Execution Result: