算法基础:整数排序有关问题(连续整数只保留首尾两个数字)

算法基础:整数排序问题(连续整数只保留首尾两个数字)
描述:
实现输入一组大于等于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
}


版权声明:本文为博主原创文章,未经博主允许不得转载。