Machine Learning Leetcode Java