Merge Sort with O(1) extra space merge and O(n log n) time [Unsigned Integers Only]
Last Updated :
16 May, 2023
We have discussed Merge sort. How to modify the algorithm so that merge works in O(1) extra space and algorithm still works in O(n Log n) time. We may assume that the input values are integers only.
Examples:
Input : 5 4 3 2 1
Output : 1 2 3 4 5
Input : 999 612 589 856 56 945 243
Output : 56 243 589 612 856 945 999
For integer types, merge sort can be made inplace using some mathematics trick of modulus and division. That means storing two elements value at one index and can be extracted using modulus and division.
First we have to find a value greater than all the elements of the array. Now we can store the original value as modulus and the second value as division. Suppose we want to store arr[i] and arr[j] both at index i(means in arr[i]). First we have to find a ‘maxval’ greater than both arr[i] and arr[j]. Now we can store as arr[i] = arr[i] + arr[j]*maxval. Now arr[i]%maxval will give the original value of arr[i] and arr[i]/maxval will give the value of arr[j]. So below is the implementation on merge sort.
Approach: (Euclidean Division)
dividend = divisor * quotient +remainder
ex: 5/3 = q:1, r:2 applying euclidean: 3*1+2 =>5 (dividend)
divisor = maxele (absolute max element in the array)+1 (so that we always get non zero remainder)
quotient = min(first, second)
remainder = original element
Note: (when getting current element, assume the current container already has encoded element hence using % divisor)
first = arr[i] % divisor
second = arr[j] % divisor
encoded element = remainder + quotient*divisor
Possible issues in Merging:
1. If current number is Integer.MAX then new encoded value which is usually greater than current element will cause integer overflow and data corruption (In python there is no limit to number size so this issue will not occur).
2. Doesn’t handle negative numbers (ie, when encoding a -ve number(current) with another -ve number(chosen smallest) the sign can’t be preserved since both numbers have -ve sign. Also absolute values must be used when computing dividend = divisor*quotient+remainder (divisor = maxele, quotient = smallest, remainder = original) and sign must be restored, still it might not work due to sign preservation issue.
3. Only applicable to Unsigned integers, like indexes which are usually non-negative.
4. AUX = O(n) in worst case, assuming in a language like python where there is no limit to word/integer size, when input array elements are almost at Integer.MAX, then encoded value will require possibly 2x bits space to represent new number, the 2x bit space on whole can become +1x array size, which is almost like creating an AUX array but in an indirect way.
5. mod and division operations are the costliest, hence reduces overall performance(upto some extent).
C++
#include <bits/stdc++.h>
using namespace std;
void merge( int arr[], int beg, int mid, int end, int maxele)
{
int i = beg;
int j = mid + 1;
int k = beg;
while (i <= mid && j <= end) {
if (arr[i] % maxele <= arr[j] % maxele) {
arr[k] = arr[k] + (arr[i] % maxele) * maxele;
k++;
i++;
}
else {
arr[k] = arr[k] + (arr[j] % maxele) * maxele;
k++;
j++;
}
}
while (i <= mid) {
arr[k] = arr[k] + (arr[i] % maxele) * maxele;
k++;
i++;
}
while (j <= end) {
arr[k] = arr[k] + (arr[j] % maxele) * maxele;
k++;
j++;
}
for ( int i = beg; i <= end; i++)
arr[i] = arr[i] / maxele;
}
void mergeSortRec( int arr[], int beg, int end, int maxele)
{
if (beg < end) {
int mid = (beg + end) / 2;
mergeSortRec(arr, beg, mid, maxele);
mergeSortRec(arr, mid + 1, end, maxele);
merge(arr, beg, mid, end, maxele);
}
}
void mergeSort( int arr[], int n)
{
int maxele = *max_element(arr, arr+n) + 1;
mergeSortRec(arr, 0, n-1, maxele);
}
int main()
{
int arr[] = { 999, 612, 589, 856, 56, 945, 243 };
int n = sizeof (arr) / sizeof (arr[0]);
mergeSort(arr, n);
cout << "Sorted array \n" ;
for ( int i = 0; i < n; i++)
cout << arr[i] << " " ;
return 0;
}
|
Java
import java.util.Arrays;
class GFG
{
static void merge( int [] arr, int beg,
int mid, int end,
int maxele)
{
int i = beg;
int j = mid + 1 ;
int k = beg;
while (i <= mid && j <= end)
{
if (arr[i] % maxele <=
arr[j] % maxele)
{
arr[k] = arr[k] + (arr[i]
% maxele) * maxele;
k++;
i++;
}
else
{
arr[k] = arr[k] +
(arr[j] % maxele)
* maxele;
k++;
j++;
}
}
while (i <= mid)
{
arr[k] = arr[k] + (arr[i]
% maxele) * maxele;
k++;
i++;
}
while (j <= end)
{
arr[k] = arr[k] + (arr[j]
% maxele) * maxele;
k++;
j++;
}
for (i = beg; i <= end; i++)
{
arr[i] = arr[i] / maxele;
}
}
static void mergeSortRec( int [] arr, int beg,
int end, int maxele)
{
if (beg < end)
{
int mid = (beg + end) / 2 ;
mergeSortRec(arr, beg,
mid, maxele);
mergeSortRec(arr, mid + 1 ,
end, maxele);
merge(arr, beg, mid,
end, maxele);
}
}
static void mergeSort( int [] arr, int n)
{
int maxele = Arrays.stream(arr).max().getAsInt() + 1 ;
mergeSortRec(arr, 0 , n - 1 , maxele);
}
public static void main(String[] args)
{
int [] arr = { 999 , 612 , 589 ,
856 , 56 , 945 , 243 };
int n = arr.length;
mergeSort(arr, n);
System.out.println( "Sorted array " );
for ( int i = 0 ; i < n; i++)
{
System.out.print(arr[i] + " " );
}
}
}
|
Python3
def merge(arr, beg, mid, end, maxele):
i = beg
j = mid + 1
k = beg
while (i < = mid and j < = end):
if (arr[i] % maxele < = arr[j] % maxele):
arr[k] = arr[k] + (arr[i] %
maxele) * maxele
k + = 1
i + = 1
else :
arr[k] = arr[k] + (arr[j] %
maxele) * maxele
k + = 1
j + = 1
while (i < = mid):
arr[k] = arr[k] + (arr[i] %
maxele) * maxele
k + = 1
i + = 1
while (j < = end):
arr[k] = arr[k] + (arr[j] %
maxele) * maxele
k + = 1
j + = 1
for i in range (beg, end + 1 ):
arr[i] = arr[i] / / maxele
def mergeSortRec(arr, beg, end, maxele):
if (beg < end):
mid = (beg + end) / / 2
mergeSortRec(arr, beg, mid, maxele)
mergeSortRec(arr, mid + 1 , end, maxele)
merge(arr, beg, mid, end, maxele)
def mergeSort(arr, n):
maxele = max (arr) + 1
mergeSortRec(arr, 0 , n - 1 , maxele)
if __name__ = = '__main__' :
arr = [ 999 , 612 , 589 , 856 , 56 , 945 , 243 ]
n = len (arr)
mergeSort(arr, n)
print ( "Sorted array" )
for i in range (n):
print (arr[i], end = " " )
|
C#
using System;
using System.Linq;
class GFG
{
static void merge( int []arr, int beg,
int mid, int end,
int maxele)
{
int i = beg;
int j = mid + 1;
int k = beg;
while (i <= mid && j <= end)
{
if (arr[i] %
maxele <= arr[j] % maxele)
{
arr[k] = arr[k] + (arr[i] %
maxele) * maxele;
k++;
i++;
}
else
{
arr[k] = arr[k] +
(arr[j] % maxele) *
maxele;
k++;
j++;
}
}
while (i <= mid)
{
arr[k] = arr[k] + (arr[i] %
maxele) * maxele;
k++;
i++;
}
while (j <= end)
{
arr[k] = arr[k] + (arr[j] %
maxele) * maxele;
k++;
j++;
}
for ( i = beg; i <= end; i++)
arr[i] = arr[i] / maxele;
}
static void mergeSortRec( int []arr, int beg,
int end, int maxele)
{
if (beg < end)
{
int mid = (beg + end) / 2;
mergeSortRec(arr, beg,
mid, maxele);
mergeSortRec(arr, mid + 1,
end, maxele);
merge(arr, beg, mid,
end, maxele);
}
}
static void mergeSort( int []arr, int n)
{
int maxele = arr.Max() + 1;
mergeSortRec(arr, 0, n - 1, maxele);
}
public static void Main ()
{
int []arr = {999, 612, 589,
856, 56, 945, 243};
int n = arr.Length;
mergeSort(arr, n);
Console.WriteLine( "Sorted array " );
for ( int i = 0; i < n; i++)
Console.Write( arr[i] + " " );
}
}
|
Javascript
<script>
function merge(arr,beg,mid,end,maxele)
{
let i = beg;
let j = mid + 1;
let k = beg;
while (i <= mid && j <= end)
{
if (arr[i] % maxele <=
arr[j] % maxele)
{
arr[k] = arr[k] + (arr[i]
% maxele) * maxele;
k++;
i++;
}
else
{
arr[k] = arr[k] +
(arr[j] % maxele)
* maxele;
k++;
j++;
}
}
while (i <= mid)
{
arr[k] = arr[k] + (arr[i]
% maxele) * maxele;
k++;
i++;
}
while (j <= end)
{
arr[k] = arr[k] + (arr[j]
% maxele) * maxele;
k++;
j++;
}
for (i = beg; i <= end; i++)
{
arr[i] = Math.floor(arr[i] / maxele);
}
}
function mergeSortRec(arr,beg,end,maxele)
{
if (beg < end)
{
let mid = Math.floor((beg + end) / 2);
mergeSortRec(arr, beg,
mid, maxele);
mergeSortRec(arr, mid + 1,
end, maxele);
merge(arr, beg, mid,
end, maxele);
}
}
function mergeSort(arr,n)
{
let maxele = Math.max(...arr) + 1;
mergeSortRec(arr, 0, n - 1, maxele);
}
let arr=[999, 612, 589,
856, 56, 945, 243];
let n = arr.length;
mergeSort(arr, n);
document.write( "Sorted array <br>" );
for (let i = 0; i < n; i++)
{
document.write(arr[i] + " " );
}
</script>
|
Output:
Sorted array
56 243 589 612 856 945 999
Time Complexity: O(n*log(n))
Space Complexity: O(1)
Share your thoughts in the comments
Please Login to comment...