Output is incorrect, please check

Problem Link:

Solution link: