算法基礎:整數排序問題(連續整數只保留首尾兩個數字)

描述:
實現輸入一組大於等於0的整數,根據從小到大的順序排序後輸出,排序後有連續數時,只輸出連續數中最小和最大的兩個數。

輸入:
一組大於等於0的整數,不考慮非法輸入,各個整數之間以逗號(“,”)分隔, 輸入字符串的總長度小於等於100個字節。
輸出:
排序後的值,各個整數之間以空格分隔。

樣例輸入:
1,4,3,110,2,90,7

樣例輸出:
1 4 7 90 110

代碼實現:

package huawei

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

func Test3Base() {
	res := "1,4,3,2,90,7,6,17,8,9"
	sortNums(res)
}

func sortNums(resString string) {
	numsStr := strings.Split(resString, ",")
	numsInt := convertToIntSlice(numsStr)

	//排序
	sort.Ints(numsInt)
	//連續數字只保留首尾
	result := removeCenterItems(numsInt)

	fmt.Println(result)
}

//將[]string轉爲[]int
func convertToIntSlice(numsStr []string) []int {
	numsInt := make([]int, len(numsStr))
	for i, v := range numsStr {
		n, err := strconv.Atoi(v)
		checkError(err, "string to integer")

		numsInt[i] = n
	}

	return numsInt
}

//連續數字只保留首尾
func removeCenterItems(nums []int) []int {
	//需要移除的索引位置
	removeList := make([]int, 0, len(nums))
	for i := 1; i < len(nums)-1; i++ {
		if nums[i] == nums[i-1]+1 && nums[i] == nums[i+1]-1 {
			removeList = append(removeList, i)
		}
	}

	//移除
	for i := len(nums) - 2; i > 0; i-- {
		if checkContains(removeList, i) {
			nums = append(nums[:i], nums[i+1:]...)
		}
	}

	return nums
}

//判斷slice中是否存在某個元素
func checkContains(nums []int, value int) bool {
	for _, v := range nums {
		if v == value {
			return true
		}
	}

	return false
}


發佈了114 篇原創文章 · 獲贊 24 · 訪問量 49萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章