Given an array which is an original arithmetic progression with one element changed. The task is to make it an arithmetic progression again. If there are many such possible sequences, return any one of them. The length of the array will always be greater than 2.
Input : arr = [1, 3, 4, 7]
Output : arr = [1, 3, 5, 7]
The common difference for the arithmetic progression is 2, so the resulting
arithmetic progression is [1, 3, 5, 7].
Input : arr = [1, 3, 7]
Output : arr = [-1, 3, 7]
The common difference can be either 2 or 3 or 4.
Hence the resulting array can be either [1, 3, 5], [-1, 3, 7] or [1, 4, 7].
As any one can be chosen, [-1, 3, 7] is returned as the output.
The key components of an Arithmetic Progression are initial term and the common difference. So our task is to find these two values to know the actual Arithmetic Progression.
- If the length of the array is 3, take the common difference as the difference between any two elements.
- Otherwise, try to find the common difference using the first four elements.
- Check if the first three elements are in arithmetic progression using the formula a – a = a – a. If they are in arithmetic progression, the common difference d = a – a and the initial term would be a
- Check if the 2nd, 3rd and 4th elements are in arithmetic progression using the formula a – a = a – a. If they are in arithmetic progression, the common difference d = a – a, which means that the first element has been changed, so initial term is a = a – d
- In both of the above cases, we have checked if the first element or fourth element has been changed. If both the cases are false, then it means that first or fourth element was not changed. So the common difference can be taken as d = (a – a)/3 and the initial term = a
- Print all the elements using the initial term and common difference.
Below is the implementation of the above approach:
-1 3 7
Time Complexity: O(N), where N is the number of elements in the array.
Attention reader! Don’t stop learning now. Get hold of all the important mathematical concepts for competitive programming with the Essential Maths for CP Course at a student-friendly price. To complete your preparation from learning a language to DS Algo and many more, please refer Complete Interview Preparation Course.