题解 | #输入整型数组和排序标识,对其元素升序或降序排序#

输入整型数组和排序标识,对其元素按照升序或降序进行排序

https://www.nowcoder.com/practice/dd0c6b26c9e541f5b935047ff4156309

package main

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

func main() {
	var n int
	if _, err := fmt.Scan(&n); err != nil {
		fmt.Printf("input n err:%v\n", err)
		return
	}

	scanner := bufio.NewScanner(os.Stdin)
	var line2Sli = make([]int, 0)
	for scanner.Scan() {
		str := scanner.Text()
		strSli := strings.Fields(str)
		if len(strSli) != n {
			fmt.Printf("line 2 is not valid\n")
			return
		}

		for _, val := range strSli {
			valInt, err := strconv.Atoi(val)
			if err != nil {
				fmt.Printf("current line 2 val err:%v\n", err)
				return
			}

			line2Sli = append(line2Sli, valInt)
		}

		break
	}

	for scanner.Scan() {
		sortFlag := scanner.Text()
		if sortFlag != "0" && sortFlag != "1" {
			fmt.Printf("input sort flag is not 0 or 1!\n")
			return
		}

		if sortFlag == "0" {
			sort.Ints(line2Sli)
		} else {
			sort.Slice(line2Sli, func(i, j int) bool {
				return line2Sli[i] > line2Sli[j]
			})
		}

		break
	}

	var retStr string
	for _, val := range line2Sli {
		retStr += fmt.Sprintf("%d ", val)
	}
	retStr = strings.TrimSuffix(retStr, " ")

	fmt.Printf("%v\n", retStr)
}

全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务