Amazon Interview Question Software Engineer / Developers
0of 0 votesQ4. Written Exam Amazon(Bangalore)
Given an array of integers A[1....n-1] where 'N' is the length of array A[ ]. Construct an array B such that B[i] = min(A[i], A[i+1], ......., A[i-K+1]), where K will be given.
Array B will have N-K+1 elements.
Constraint: Extra space allowed O(K) and time complexity allowed O(N.K) or lower.
Country: India
Interview Type: Written Test
the algo could go on like this....
1.scan the array in reverse order and create a min heap of k elements.
2.the element B[i] would be the element at the root of min heap.
3. now delete the element at A[K+i -1] and insert the element A[i-1]
4.reheapify the heap.
TC-O(nlogk)
deletion of element A[K+i-1] is O(k) operation. So, time complexity of solution is O(nk).
How this algo will work for following input :
arr[] = {9,8,7};
N=3;
k=2;
So, as per step (i),starting from last K elements we will maintain MIN heap.
So, at first, heap will be {7,8} with 7 sitting at top. << b[1] = 7
But as per step (iii), 8 will be replaced by 9
So next time b[0] will be 7 as 7<9, and resulted output will be {7,7} instead of {8,7}
the algo could go on like this....
1.scan the array in reverse order and create a min heap of k elements.
2.the element B[i] would be the element at the root of min heap.
3. now delete the element at A[K+i -1] and insert the element A[i-1]
4.reheapify the heap.
TC-O(nlogk)
Solution: O(n) time and excluding the result space + O(1) space if there are no duplicates (if there are duplicates we can move them to the end of the array and use extra HashMap to count the duplicates)
Here is the java code for nonDuplicate version (can be easyly modified for duplicates) :
public int[] getFirstElements(int[] arr, int k) {
int[] result = new int[k];
int currK = 0;
for (int i = 0; i < arr.length && currK < k; i++)
if (arr[i] != 0)
result[currk++] = arr[i];
return result
}
//O(n) time
private void sortArr(int[] arr) {
for (int i = 0; i < arr.length; i++)
while (arr[i] != i || arr[i] != 0)
swap(arr, arr[i], i);
}
can be done in O(N)
1.b[0] = the minimum of (0,1,K-1)
2. b[1] = min(b[0], a[k]
3. b[2] = min(b[1], a[k+1]
Looks like below is not correct
- Anonymous on May 13, 2012 Edit | Flag ReplyB[i] = min(A[i], A[i+1], ......., A[i-K+1]), where K will be given.
here A[i-K+1] should be A[i+k-1]
if i=0, and k=4 then i-K+1 will be negative, which is not correct