summaryrefslogtreecommitdiff
path: root/sort/parallelmerge.go
blob: 5186632b14d449c6a1d9dfcf76918fbf326686c9 (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"

	"github.com/snonux/algorithms/ds"
)

func ParallelMerge(a ds.ArrayList) ds.ArrayList {
	parallelMerge(a, make(ds.ArrayList, len(a)))
	return a
}

func parallelMerge(a, aux ds.ArrayList) {
	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)
}