function in c hackerrank solution