算法基礎:排列組合問題-全排列(Golang實現)

【排列組合問題】
一共N輛火車(0<N<10),每輛火車以數字1-9編號,要求以字典序排序輸出火車出站的序列號。

輸入:
包括N個正整數(0<N<10),範圍爲1到9,數字之間用空格分割,字符串首位不包含空格。
輸出:
輸出以字典序排序的火車出站序列號,每個編號以空格隔開,每個輸出序列換行。

樣例輸入:
1 2 3

樣例輸出:
1 2 3
1 3 2
2 1 3
2 3 1
3 1 2
3 2 1

代碼實現:

package huawei

import (
    "fmt"
    "sort"
    "strings"
)

func Test7Base() {
    in := "5 2 3"
    result := outOrder(strings.Fields(in))
    dictSort(result)
    s := format(result)
    fmt.Println(s)

}

//輸入trainsNums,返回全部排列
//如輸入[1 2 3],則返回[123 132 213 231 312 321]
func outOrder(trainsNums []string) []string {
    COUNT := len(trainsNums)
    //檢查
    if COUNT == 0 || COUNT > 10 {
        panic("Illegal argument. trainsNums size must between 1 and 9.")
    }

    //如果只有一個數,則直接返回
    if COUNT == 1 {
        return []string{trainsNums[0]}
    }

    //否則,將最後一個數插入到前面的排列數中的所有位置(遞歸)
    return insert(outOrder(trainsNums[:COUNT-1]), trainsNums[COUNT-1])
}

func insert(res []string, insertNum string) []string {
    //保存結果的slice
    result := make([]string, len(res)*(len(res[0])+1))

    index := 0
    for _, v := range res {
        for i := 0; i < len(v); i++ {
            //在v的每一個元素前面插入
            result[index] = v[:i] + insertNum + v[i:]
            index++
        }

        //在v最後面插入
        result[index] = v + insertNum
        index++
    }

    return result
}

//按字典順序排序
func dictSort(res []string) {
    sort.Strings(res)
}

//按指定格式輸出
func format(res []string) string {
    strs := make([]string, len(res))
    for i := 0; i < len(res); i++ {
        strs[i] = addWhiteSpace(res[i])
    }

    return strings.Join(strs, "\n")
}

//添加空格
func addWhiteSpace(s string) string {
    var retVal string
    for i := 0; i < len(s); i++ {
        retVal += string(s[i])

        if i != len(s)-1 {
            retVal += " "
        }
    }

    return retVal
}
發佈了114 篇原創文章 · 獲贊 24 · 訪問量 49萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章