Cerebras (CA) Interview Question

2d array manipulation using hackerrank