The pythagoreans challenge

#1

SUGGESTION NEEDED TO REDUCE THE TIME LIMIT O THE IRST TEST CASE
AND NOT ABLE TO UNDERSTAND WHY SECOND TEST CASE IS COMING WRONG