Plz correct code

#1
#2

hello @Lovishbansal
your logics looks correct.
generalise it and submit ,it will work.