leetcode-443. 压缩字符串

61 阅读2分钟

题目要求:

给你一个字符数组 chars ,请使用下述算法压缩:

从一个空字符串 s 开始。对于 chars 中的每组 连续重复字符 :

  • 如果这一组长度为 1 ,则将字符追加到 s 中。
  • 否则,需要向 s 追加字符,后跟这一组的长度。

压缩后得到的字符串 s 不应该直接返回 ,需要转储到字符数组 chars 中。需要注意的是,如果组长度为 10 或 10 以上,则在 chars 数组中会被拆分为多个字符。

请在 修改完输入数组后 ,返回该数组的新长度。

你必须设计并实现一个只使用常量额外空间的算法来解决此问题。

示例 1:
输入: chars = ["a","a","b","b","c","c","c"]
输出: 返回 6 ,输入数组的前 6 个字符应该是: ["a","2","b","2","c","3"]
解释: "aa" 被 "a2" 替代。"bb" 被 "b2" 替代。"ccc" 被 "c3" 替代。

示例 2:
输入: chars = ["a"]
输出: 返回 1 ,输入数组的前 1 个字符应该是:["a"]
解释: 唯一的组是“a”,它保持未压缩,因为它是一个字符。

示例 3:
输入: chars = ["a","b","b","b","b","b","b","b","b","b","b","b","b"]
输出: 返回 4 ,输入数组的前 4 个字符应该是:["a","b","1","2"]。
解释: 由于字符 "a" 不重复,所以不会被压缩。"bbbbbbbbbbbb" 被 “b12” 替代。

提示
1 <= chars.length <= 2000
chars[i] 可以是小写英文字母、大写英文字母、数字或符号

链接leetcode.cn/problems/st…

解题思路

使用双指针,这道题难在数字和字符的格式转换

代码

func compress(chars []byte) int {
	// 使用双指针
	n := len(chars)
	if n <= 1 {
		return n
	}
	w, r := 0, 0
	for r < n {
		cur := chars[r]
		count := 0
		for r < n && cur == chars[r] {
			r++
			count++
		}
		chars[w] = cur
		w++
		if count > 1 {
			for _, c := range strconv.Itoa(count) {
				chars[w] = byte(c)
				w++
			}
		}
	}
	return w
}