During an interview, I've been asked the following Question:
You're given an array of integer numbers.
Find the maximum difference between two elements
arr[j] - arr[i]
for any sub array in the array, so thatj>i
.For example:
array =
{20,18,45,78,3,65,55}
, max diff is65 - 3 = 62
.array =
{20,8,45,78,3,65,55}
, max diff is78 - 8 = 70
.
Here is the solution I come up with:
private static int calculateProfit() {
int[] arr = {20, 18, 45, 78, 3, 65, 55};
int maxProfit = 0;
for (int i = 0; i < arr.length; i ) {
for (int j = arr.length - 1; j > 0; j--) {
if (arr[i] < arr[j] && i < j) {
maxProfit = Math.max(arr[j] - arr[i], maxProfit);
}
}
}
return maxProfit; // ans: (65 - 3) = 62
}
The problem is that it runs in O(n^2). How it can be done with a better time complexity?
CodePudding user response:
This problem can be solved in a linear time O(n), with a single run through the given array.
We need to declare only a couple of local variables, no data additional data structures required, space complexity is O(1).
These are the variables we need to track:
min
- the lowest value encountered so far;max
- the highest encountered value;maxProfit
- maximal profit that can be achieved at the moment.
While declaring these variables, we can either initialize min
to Integer.MAX_VALUE
and max
to Integer.MIN_VALUE
, or initialize both with the value of the first element in the array (this element should be present because the array needs to have at least two elements, otherwise the task has no sense).
And here is a couple of caveats:
Since
max
element can not precede themin
element, when a newmin
element is encountered (when the current element is less thanmin
) themax
element also needs to be reinitialized (withInteger.MIN_VALUE
or with the value of the current element depending on the strategy you've chosen at the beginning).maxProfit
should be checked against the difference betweenmax
andmin
each time when a newmax
has been encountered.
That's how it might be implemented:
public static int calculateProfit(int[] arr) {
if (arr.length < 2) return -1; // incorrect input
int max = arr[0];
int min = arr[0];
int maxProfit = 0;
for (int i = 1; i < arr.length; i ) {
int next = arr[i];
if (next > max) {
max = next;
maxProfit = Math.max(max - min, maxProfit);
} else if (next < min){
min = next;
max = next;
}
}
return maxProfit;
}
main()
public static void main(String[] args) {
System.out.println(calculateProfit(new int[]{1, 2, 3, 4, 10}));
System.out.println(calculateProfit(new int[]{1, 10, -10, 4, 8}));
System.out.println(calculateProfit(new int[]{5, 8, 12, 1, 9}));
System.out.println(calculateProfit(new int[]{20, 18, 45, 78, 3, 65, 55}));
}
Output:
9 // [1, 2, 3, 4, 10] -> 10 - 1 = 9
18 // [1, 10, -10, 4, 8] -> 8 - (-10) = 18
8 // [5, 8, 12, 1, 9] -> 9 - 1 = 8
62 // [20, 18, 45, 78, 3, 65, 55] -> 65 - 3 = 62
CodePudding user response:
Start by finding the first possible solution. This is easy. Scan left-to-right. Track p for the index of the lowest value seen so far. As soon as you see a new non-descending value, set q to that index and goto
step 2
:[8, 5, 2, 2, 4, ...] [ D D -, I ] # Look for first increasing value [ p q ] # arr[p] < arr[q] and p < q
Now continue looping over the array, looking for an improvement to the solution. The new wrinkle is tracking a u, the index of a value smaller than
arr[p]
but is unused because it is to the right of q:[8, 5, 2, 2, 4, 3, 1, ...] [ p q u ] # arr[u] < arr[p] but u > q
For each new value in the loop, update the solution. If the new i/u difference is bigger than the p/q solution, it becomes the new best:
[8, 5, 2, 2, 4, 3, 1, 2, X, ...]
[ p q u i, ...]
\...../ Previous best solution
\-----/ Potential new best