Time complexity in simple c program and trying to use clean code with some additional logic