Pythagoras Triplets

Not able to pass 2 testcases .

@G_rahil,
Why are you taking two numbers in input, only N is given as input?