题解 | #输入整型数组和排序标识

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

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

package main

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

func main() {
	inputs := bufio.NewScanner(os.Stdin)
	inputs.Scan()
	inputs.Scan()
	arr := strings.Split(inputs.Text(), " ")
	inputs.Scan()
	order := inputs.Text()

	var numbers []int
	for _, number := range arr {
		num, _ := strconv.Atoi(number)
		numbers = append(numbers, num)
	}

	sort.Slice(numbers, func(i, j int) bool {
		if order == "0" {
			return numbers[i] < numbers[j]
		} else {
			return numbers[i] > numbers[j]
		}
	})
	for _, number := range numbers {
		fmt.Printf("%v ", number)
	}
}

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务