package mergesort

import (
	"reflect"
	"testing"
)

// MergeSort returns a sorted copy of nums using merge sort.
func MergeSort(nums []int) []int {
	if len(nums) <= 1 {
		return append([]int(nil), nums...)
	}

	mid := len(nums) / 2
	left := MergeSort(nums[:mid])
	right := MergeSort(nums[mid:])

	return merge(left, right)
}

// merge combines two sorted slices into one sorted slice.
func merge(left, right []int) []int {
	result := make([]int, 0, len(left)+len(right))
	i, j := 0, 0

	for i < len(left) && j < len(right) {
		if left[i] <= right[j] {
			result = append(result, left[i])
			i++
		} else {
			result = append(result, right[j])
			j++
		}
	}

	result = append(result, left[i:]...)
	result = append(result, right[j:]...)
	return result
}

func TestMergeSort(t *testing.T) {
	tests := []struct {
		name string
		in   []int
		want []int
	}{
		{"empty", []int{}, []int{}},
		{"single", []int{7}, []int{7}},
		{"unsorted", []int{5, 2, 9, 1, 5, 6}, []int{1, 2, 5, 5, 6, 9}},
		{"negatives", []int{3, -1, 0, -8, 3}, []int{-8, -1, 0, 3, 3}},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got := MergeSort(tt.in)
			if !reflect.DeepEqual(got, tt.want) {
				t.Fatalf("MergeSort(%v) = %v, want %v", tt.in, got, tt.want)
			}
		})
	}
}

func TestMergeSortDoesNotMutateInput(t *testing.T) {
	in := []int{4, 1, 3, 2}
	original := append([]int(nil), in...)

	_ = MergeSort(in)

	if !reflect.DeepEqual(in, original) {
		t.Fatalf("MergeSort mutated input: got %v, want %v", in, original)
	}
}