Solve 004-subset-sum-problem in python