欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Golang排列组合算法问题之全排列实现方法

程序员文章站 2022-07-04 23:44:44
本文实例讲述了golang排列组合算法问题之全排列实现方法。分享给大家供大家参考,具体如下: 【排列组合问题】 一共n辆火车(0

本文实例讲述了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
}

希望本文所述对大家go语言程序设计有所帮助。