summaryrefslogtreecommitdiff
path: root/sort/parallelmerge.go
blob: 1f5030c064ba45701564f3e7c690afc7a46bd44d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
package sort

import (
	"sync"

	"codeberg.org/snonux/algorithms/ds"
)

func ParallelMerge[V ds.Number](a ds.ArrayList[V]) ds.ArrayList[V] {
	parallelMerge[V](a, make(ds.ArrayList[V], len(a)))
	return a
}

func parallelMerge[V ds.Number](a, aux ds.ArrayList[V]) {
	l := len(a)

	if l < 1000 {
		mergeSort(a, aux)
		return
	}

	mi := l / 2
	var wg sync.WaitGroup
	wg.Add(2)

	go func() {
		parallelMerge(a[0:mi], aux[0:mi])
		wg.Done()
	}()

	go func() {
		parallelMerge(a[mi:], aux[mi:])
		wg.Done()
	}()

	wg.Wait()
	merge(a, aux, 0, mi, l-1)
}