Problem 1 and 2 using python & c++