Bjak interview question

2 Leetcode problems, 2 tests on Machine Learning and Pytorch knowledge