글 읽기 - [Go] 자꾸 틀립니다.. 어디가 틀렸는지 알 수있을까요??

tizm423 3년 전 0

병합정렬을 응용해서 푸는 문제라고 알 고있고, 이를 적용했습니다.

틀렸다는것을 보면 어딘가 잘못됐다는건데 혹시 어디가 잘못된건지 설명해주실 수 있을까요..?

package main

import (
	"bufio"
	"os"
	"strconv"
	"strings"
)

var (
	a              []int
	b              []int
	n, m           int
	aIndex, bIndex int
	s              = bufio.NewScanner(os.Stdin)
	w              = bufio.NewWriter(os.Stdout)
)

func main() {
	defer w.Flush()
	s.Scan()
	nm := strings.Split(s.Text(), " ")
	n, _ = strconv.Atoi(nm[0])
	m, _ = strconv.Atoi(nm[1])
	a = make([]int, n)
	b = make([]int, m)
	s.Scan()
	for i, at := range strings.Split(s.Text(), " ") {
		a[i], _ = strconv.Atoi(at)
	}
	s.Scan()
	for i, bt := range strings.Split(s.Text(), " ") {
		b[i], _ = strconv.Atoi(bt)
	}
	mergeSort()
}

func mergeSort() {
	aIndex = 0
	bIndex = 0
	for aIndex < n && bIndex < m {
		if a[aIndex] < b[bIndex] {
			w.WriteString(strconv.Itoa(a[aIndex])+" ")
			aIndex++
		} else {
            w.WriteString(strconv.Itoa(b[bIndex])+" ")
			bIndex++
		}
	}
    for i := aIndex; i < m; i++ {
        w.WriteString(strconv.Itoa(a[i])+" ")
    }
    for i := bIndex; i < n; i++ {
        w.WriteString(strconv.Itoa(b[i])+" ")
    }
	w.WriteString("\n")
}