码迷,mamicode.com
首页 > 编程语言 > 详细

算法基础:排列组合问题(Golang实现)

时间:2015-07-10 11:24:32      阅读:143      评论:0      收藏: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
}

版权声明:本文为博主原创文章,未经博主允许不得转载。

算法基础:排列组合问题(Golang实现)

标签:golang   序列   组合   算法   

原文地址:http://blog.csdn.net/books1958/article/details/46827453

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!