Sobes.tech
Junior — Middle

Hogyan lehet két rendezett tömböt egyesíteni úgy, hogy megőrizzük az elemek sorrendjét?

sobes.tech MI

Válasz az MI-től

Két rendezett tömb egyesítéséhez egy rendezett tömbbé, használhatunk egy összeolvasztási algoritmust, amely hasonló a merge sortban alkalmazott módszerhez. Az ötlet az, hogy egyidejűleg végigmenjünk mindkét tömbön, összehasonlítva az aktuális elemeket, és hozzáadva a kisebbet az eredmény tömbhöz.

Példa Go nyelven:

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++
        }
    }

    // Hozzáadjuk a maradék elemeket
    result = append(result, a[i:]...)
    result = append(result, b[j:]...)

    return result
}

Ez a módszer időben O(n + m) működik, ahol n és m az eredeti tömbök mérete.