summaryrefslogtreecommitdiff
path: root/ds/arraylist.go
blob: 8e9938491943d2e5f4d3450e8a24118551c3486d (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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
package ds

import (
	"fmt"
	"math/rand"
	"strings"
)

type ArrayList []int

func NewRandomArrayList(l, max int) ArrayList {
	a := make(ArrayList, l)
	for i := 0; i < l; i++ {
		if max > 0 {
			a[i] = rand.Intn(max)
			continue
		}
		a[i] = rand.Int()
	}
	return a
}

func NewAscendingArrayList(l int) ArrayList {
	a := make(ArrayList, l)
	for i := 0; i < l; i++ {
		a[i] = i
	}
	return a
}

func NewDescendingArrayList(l int) ArrayList {
	a := make(ArrayList, l)
	j := l - 1
	for i := 0; i < l; i++ {
		a[i] = j
		j--
	}
	return a
}

func (a ArrayList) FirstN(n int) string {
	var sb strings.Builder
	j := n

	l := len(a)
	if j > l {
		j = l
	}

	for i := 0; i < j; i++ {
		fmt.Fprintf(&sb, "%v ", a[i])
	}

	if j < l {
		fmt.Fprintf(&sb, "... ")
	}

	return sb.String()
}

func (a ArrayList) Sorted() bool {
	for i := len(a) - 1; i > 0; i-- {
		if a[i] < a[i-1] {
			return false
		}
	}
	return true
}

func (a ArrayList) Swap(i, j int) {
	aux := a[i]
	a[i] = a[j]
	a[j] = aux
}