Aman interview question

Code for decision tree algorithm