Test case 1 and 2 failing

#1
#2

hello @varun.saxena
declare ur vector and function return type as long long .
and then it should work fine