Check whether it is possible to make both arrays equal by modifying a single element in Python

Suppose we have two arrays nums1 and nums2 and another value k. We have to check whether both arrays can be made equal by modifying any one element from the nums1 in the following way (only once): We can add any value from the range [-k, k] to any element of nums1.

So, if the input is like nums1 = [5,7,11] nums2 = [5,5,11] k = 8, then the output will be True as we can add -2 (in range [-8,8]) with nums1[1] to make it 5 then it will be same as nums2.

To solve this, we will follow these steps −

• sort the list nums1 and nums2
• temp := False
• idx := -1
• for i in range 0 to size of nums1 - 1, do
  • if nums1[i] is not same as nums2[i], then
    • if temp is true, then
      • return False
    • temp := True
    • idx := i
• if idx is -1 or |nums1[idx]-nums2[idx]| <= k, then
  • return True
• return False

Let us see the following implementation to get better understanding −

Example Code

Live Demo

def solve(nums1, nums2, k):
   nums1.sort()
   nums2.sort()

   temp = False

   idx = -1
   for i in range(len(nums1)):
      if nums1[i] != nums2[i]:
         if temp:
           return False
         temp = True
         idx = i

   if idx == -1 or abs(nums1[idx]-nums2[idx]) <= k:
      return True
   return False

nums1 = [5,7,11]
nums2 = [5,5,11]
k = 8
print(solve(nums1, nums2, k))

Input

[5,7,11], [5,5,11], 8

Output

True

Arnab Chakraborty

Updated on 16-Jan-2021 04:35:12

• Related Questions & Answers
• Python Program to check whether it is possible to make a divisible by 3 number using all digits in an array
• Java Program to check whether it is possible to make a divisible by 3 number using all digits in an array
• C/C++ Program to check whether it is possible to make a divisible by 3 number using all digits in an array?
• Check if it is possible to make two matrices strictly increasing by swapping corresponding values only in Python
• C/C++ Program to check whether it is possible to make the divisible by 3 number using all digits in an array?
• Check if it possible to partition in k subarrays with equal sum in Python
• Is it possible to make a case-insensitive query in MongoDB?
• Is it possible to hack a phone by Texting?
• Check if it is possible to survive on Island in Python
• Check if it is possible to sort the array after rotating it in Python
• Program to check whether one string swap can make strings equal or not using Python
• Check if it is possible to reach a number by making jumps of two given length in Python
• Check if it is possible to create a polygon with a given angle in Python
• Program to sqeeze list elements from left or right to make it single element in Python
• Check if it is possible to transform one string to another in Python