Program to find most occurring number after k increments in python
Suppose we have a list of numbers called nums and another value k. Let us consider an operation where we increase some element by one. We can perform at most k times, we have to find the value of the most frequently occurring number we can obtain. If there are more than one solution, select the smallest possible number.
So, if the input is like nums = [1, 0, 0, 0, 8, 8, 8, 8] k = 8, then the output will be 8, as we can increase 1, 7 times to get 8, and increase any 0 to 1, so, we get [8, 1, 0, 0, 8, 8, 8, 8]. So the result is 8.
To solve this, we will follow these steps:
- sort the list nums
- low := 0, high := 0
- dist := 0, best := 0
- ret := -1
- while high < size of nums, do
- if high > 0 and nums[high] is not same as nums[high - 1], then
- dist := dist +(high - low) *(nums[high] - nums[high - 1])
- high := high + 1
- while dist > k, do
- dist := dist - nums[high - 1] - nums[low]
- low := low + 1
- if high - low > best, then
- best := high - low
- ret := nums[high - 1]
- return ret
Let us see the following implementation to get better understanding:
Example Code
Live Demo
class Solution:
def solve(self, nums, k):
nums.sort()
low, high = 0, 0
dist = 0
best = 0
ret = -1
while high < len(nums):
if high > 0 and nums[high] != nums[high - 1]:
dist += (high - low) * (nums[high] - nums[high - 1])
high += 1
while dist > k:
dist -= nums[high - 1] - nums[low]
low += 1
if high - low > best:
best = high - low
ret = nums[high - 1]
return ret
ob = Solution()
nums = [1, 0, 0, 0, 8, 8, 8, 8]
k = 8
print(ob.solve(nums, k))
Input
[1, 0, 0, 0, 8, 8, 8, 8], 8
Output
8
Published on 25-Nov-2020 17:12:25
