华为机考:多个数组按顺序合并(Golang实现)

华为笔试题 – 多个数组按顺序合并(Golang代码实现)
题目描述:
现在有多组整数数组,需要将他们合并成一个新的数组。合并规则,从每个数组里按顺序取出固定长度的内容合并到新的数组中,取完的内容会删除掉,如果该行不足固定长度或者已经为空,则直接取出剩余部分的内容放到新的数组中,继续下一行。如样例1,获得长度3,先遍历第一行,获得2,5,6;再遍历第二行,获得1,7,4;再循环回到第一行,获得7,9,5;再遍历第二行,获得3,4;再回到第一行,获得7,按顺序拼接成最终结果。

输入描述:
第一行是每次读取的固定长度,长度>0;
第2-n行是需要合并的数组,不同的数组用回车换行分隔,数组内部用逗号分隔。

输出描述:
输出一个新的数组,用逗号分隔。

-------------示例---------------------------------------------------------------------------
输入:
3
2,5,6,7,9,5,7
1,7,4,3,4
4,5,7,1,3,8

输出:
2,5,6,1,7,4,4,5,7,7,9,5,3,4,1,3,8,7

package main

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

func solve(n int, m [][]int) []int {
    i, j := 0, n
    nums := make([]int,0)
    for {
        var flag = false
        for t :=0; t < len(m); t++ {
            if i <= len(m[t]) && j <= len(m[t]) {
                flag = true
                nums = append(nums, m[t][i:j]...)
            } else if i <= len(m[t]) && j > len(m[t]) {

                flag = true
                nums = append(nums, m[t][i:]...)
            }
        }
        if !flag {
            break
        }
        i += n
        j += n
    }
    return nums
}
func main() {
    var n int
    m := make([][]int, 0)
    scanner := bufio.NewScanner(os.Stdin)
    //scanner.Scan()
    fmt.Scanln(&n)
    for scanner.Scan() {
        tmp := strings.Split(scanner.Text(),",")
        arr := make([]int,0)
        for i :=0; i 

你可能感兴趣的:(华为机考:多个数组按顺序合并(Golang实现))