Program to find length of longest sublist containing repeated numbers by k operations in Python

Suppose we have a list called nums and a value k, now let us consider an operation by which we can update the value of any number in the list. We have to find the length of the longest sublist which contains repeated numbers after performing at most k operations.

So, if the input is like nums = [8, 6, 6, 4, 3, 6, 6] k = 2, then the output will be 6, because we can change 4 and 3 to 6, to make this array [8, 6, 6, 6, 6, 6, 6], and the length of sublist with all 6s is 6.

To solve this, we will follow these steps −

• if nums is empty, then

  • return 0

• num_count := an empty map

• max_count := 0

• start := 0

• for each index end and value num in nums, do

  • num_count[num] := num_count[num] + 1

  • max_count := maximum of max_count and num_count[num]

  • if end - start + 1 > max_count + k, then

    • num_count[nums[start]] := num_count[nums[start]] - 1

    • start := start + 1

• return end - start + 1

Example

Let us see the following implementation to get better understanding

from collections import defaultdict
def solve(nums, k):
   if not nums:
      return 0

   num_count = defaultdict(int)
   max_count = 0
   start = 0

   for end, num in enumerate(nums):
      num_count[num] += 1
      max_count = max(max_count, num_count[num])
      if end - start + 1 > max_count + k:
         num_count[nums[start]] -= 1
         start += 1
   return end - start + 1

nums = [8, 6, 6, 4, 3, 6, 6]
k = 2
print(solve(nums, k))

Input

[8, 6, 6, 4, 3, 6, 6], 2

Output

6

Arnab Chakraborty

Published on 11-Oct-2021 08:12:58

• Related Questions & Answers
• Program to find length of longest distinct sublist in Python
• Program to find length of longest alternating inequality elements sublist in Python
• Program to find length of longest sublist with given condition in Python
• Program to find longest equivalent sublist after K increments in Python
• Program to find length of longest strictly increasing then decreasing sublist in Python
• Program to find length of longest sublist whose sum is 0 in Python
• Program to find length of longest consecutive sublist with unique elements in Python
• Program to find length of longest sublist with value range condition in Python
• Program to find length of longest set of 1s by flipping k bits in Python
• Program to find length of longest sublist where difference between min and max smaller than k in Python
• Program to find length of longest contiguously strictly increasing sublist after removal in Python
• Program to find length of longest contiguous sublist with same first letter words in Python
• Program to find length of smallest sublist that can be deleted to make sum divisible by k in Python
• Program to find length of the longest path in a DAG without repeated nodes in Python
• Program to find length of longest matrix path length in Python