Sobes.tech
Middle — Senior

Kako implementirati spajanje dve sortirane nizove u jedan sortirani niz?

sobes.tech АИ

Одговор од АИ

Za spajanje dva sortirana niza u jedan sortirani niz, možete koristiti algoritam spajanja sličan onom koji se koristi u sortiranju spajanjem (merge sort).

Ideja:

  • Proći kroz oba niza sa dva indeksa.
  • Na svakom koraku, uporediti trenutne elemente i dodati manji u rezultujući niz.
  • Kada jedan od nizova završi, dodajte preostale elemente drugog niza.

Primer u Go:

func mergeSortedArrays(a, b []int) []int {
    result := make([]int, 0, len(a)+len(b))
    i, j := 0, 0

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

    // Dodajte preostale elemente
    result = append(result, a[i:]...)
    result = append(result, b[j:]...)

    return result
}

Ovaj algoritam radi u vremenu O(n + m), gde su n i m veličine ulaznih nizova.