nums[i+1] must be at least equal to nums[i] + 1.

Think greedily. You don't have to increase nums[i+1] beyond nums[i]+1.

Iterate on i and set nums[i] = max(nums[i-1]+1, nums[i]) .