Pythagorean triplet

still not clearing all the test cases