feat(sorting): add even-only merge sort strategy
This commit is contained in:
@@ -0,0 +1,128 @@
|
|||||||
|
package ru.project.sorting;
|
||||||
|
|
||||||
|
import ru.project.collection.MyList;
|
||||||
|
|
||||||
|
import java.util.Comparator;
|
||||||
|
import java.util.function.ToIntFunction;
|
||||||
|
|
||||||
|
public class EvenOnlyMergeSortStrategy<T> implements SortStrategy<T> {
|
||||||
|
private final ToIntFunction<T> valueExtractor;
|
||||||
|
|
||||||
|
public EvenOnlyMergeSortStrategy(ToIntFunction<T> valueExtractor) {
|
||||||
|
this.valueExtractor = valueExtractor;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void sort(MyList<T> list, Comparator<T> comparator) {
|
||||||
|
|
||||||
|
if (list == null || list.size() <= 1) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
int evenCount = 0;
|
||||||
|
|
||||||
|
for (int i = 0; i < list.size(); i++) {
|
||||||
|
T element = list.get(i);
|
||||||
|
|
||||||
|
if (valueExtractor.applyAsInt(element) % 2 == 0) {
|
||||||
|
evenCount++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (evenCount <= 1) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
@SuppressWarnings("unchecked")
|
||||||
|
T[] evenElements = (T[]) new Object[evenCount];
|
||||||
|
|
||||||
|
int[] evenIndexes = new int[evenCount];
|
||||||
|
|
||||||
|
int evenIndex = 0;
|
||||||
|
|
||||||
|
for (int i = 0; i < list.size(); i++) {
|
||||||
|
T element = list.get(i);
|
||||||
|
|
||||||
|
if (valueExtractor.applyAsInt(element) % 2 == 0) {
|
||||||
|
evenElements[evenIndex] = element;
|
||||||
|
evenIndexes[evenIndex] = i;
|
||||||
|
evenIndex++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
mergeSort(evenElements, 0, evenElements.length - 1, comparator);
|
||||||
|
|
||||||
|
for (int i = 0; i < evenIndexes.length; i++) {
|
||||||
|
list.set(evenIndexes[i], evenElements[i]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void mergeSort(
|
||||||
|
T[] elements,
|
||||||
|
int left,
|
||||||
|
int right,
|
||||||
|
Comparator<T> comparator
|
||||||
|
) {
|
||||||
|
if (left >= right) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
int mid = (left + right) / 2;
|
||||||
|
|
||||||
|
mergeSort(elements, left, mid, comparator);
|
||||||
|
mergeSort(elements, mid + 1, right, comparator);
|
||||||
|
|
||||||
|
merge(elements, left, mid, right, comparator);
|
||||||
|
}
|
||||||
|
private void merge(
|
||||||
|
T[] elements,
|
||||||
|
int left,
|
||||||
|
int mid,
|
||||||
|
int right,
|
||||||
|
Comparator<T> comparator
|
||||||
|
) {
|
||||||
|
int leftSize = mid - left + 1;
|
||||||
|
int rightSize = right - mid;
|
||||||
|
|
||||||
|
@SuppressWarnings("unchecked")
|
||||||
|
T[] leftArray = (T[]) new Object[leftSize];
|
||||||
|
|
||||||
|
@SuppressWarnings("unchecked")
|
||||||
|
T[] rightArray = (T[]) new Object[rightSize];
|
||||||
|
|
||||||
|
for (int i = 0; i < leftSize; i++) {
|
||||||
|
leftArray[i] = elements[left + i];
|
||||||
|
}
|
||||||
|
|
||||||
|
for (int j = 0; j < rightSize; j++) {
|
||||||
|
rightArray[j] = elements[mid + 1 + j];
|
||||||
|
}
|
||||||
|
|
||||||
|
int i = 0;
|
||||||
|
int j = 0;
|
||||||
|
int k = left;
|
||||||
|
|
||||||
|
while (i < leftSize && j < rightSize) {
|
||||||
|
if (comparator.compare(leftArray[i], rightArray[j]) <= 0) {
|
||||||
|
elements[k] = leftArray[i];
|
||||||
|
i++;
|
||||||
|
} else {
|
||||||
|
elements[k] = rightArray[j];
|
||||||
|
j++;
|
||||||
|
}
|
||||||
|
|
||||||
|
k++;
|
||||||
|
}
|
||||||
|
|
||||||
|
while (i < leftSize) {
|
||||||
|
elements[k] = leftArray[i];
|
||||||
|
i++;
|
||||||
|
k++;
|
||||||
|
}
|
||||||
|
|
||||||
|
while (j < rightSize) {
|
||||||
|
elements[k] = rightArray[j];
|
||||||
|
j++;
|
||||||
|
k++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
Reference in New Issue
Block a user