remove-k-digits.go

package problem0402

func removeKdigits(num string, k int) string {
	// 返回值的长度
	digits := len(num) - k
	stack := make([]byte, len(num))
	top := 0

	for i := range num {
		// 在还能删除的前提下
		// 从上往下，删除 stack 中所有比 num[i] 大的数
		for top > 0 && stack[top-1] > num[i] && k > 0 {
			top--
			k--
		}
		stack[top] = num[i]
		top++
	}

	// 处理开头的　'0'
	i := 0
	for i < digits && stack[i] == '0' {
		i++
	}

	if i == digits {
		return "0"
	}
	return string(stack[i:digits])
}