题解 | #输入整型数组和排序标识,对其元素升序或降序排序#
输入整型数组和排序标识,对其元素按照升序或降序进行排序
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) }