0880. 索引处的解码字符串【中等】

• 1. 📝 题目描述
• 2. 🎯 s.1 - 逆向工作

1. 📝 题目描述

• leetcode

给定一个编码字符串 s。请你找出 解码字符串 并将其写入磁带。解码时，从编码字符串中 每次读取一个字符，并采取以下步骤：

• 如果所读的字符是字母，则将该字母写在磁带上。
• 如果所读的字符是数字（例如 d），则整个当前磁带总共会被重复写 d-1 次。

现在，对于给定的编码字符串 s 和索引 k，查找并返回解码字符串中的第 k 个字母。

---

示例 1：

输入：s = "leet2code3", k = 10
输出："o"
解释：
解码后的字符串为 "leetleetcodeleetleetcodeleetleetcode"。
字符串中的第 10 个字母是 "o"。

示例 2：

输入：s = "ha22", k = 5
输出："h"
解释：
解码后的字符串为 "hahahaha"。第 5 个字母是 "h"。

示例 3：

输入：s = "a2345678999999999999999", k = 1
输出："a"
解释：
解码后的字符串为 "a" 重复 8301530446056247680 次。第 1 个字母是 "a"。

---

提示：

• 2 <= s.length <= 100
• s 只包含小写字母与数字 2 到 9。
• s 以字母开头。
• 1 <= k <= 10^9
• 题目保证 k 小于或等于解码字符串的长度。
• 解码后的字符串保证少于 2^63 个字母。

2. 🎯 s.1 - 逆向工作

c

char* decodeAtIndex(char* s, int k) {
    long long size = 0;
    int n = strlen(s);
    for (int i = 0; i < n; i++) {
        if (s[i] >= '2' && s[i] <= '9') size *= (s[i] - '0');
        else size++;
    }
    char* res = (char*)malloc(2);
    res[1] = '\0';
    for (int i = n - 1; i >= 0; i--) {
        k %= size;
        if (k == 0 && s[i] >= 'a' && s[i] <= 'z') { res[0] = s[i]; return res; }
        if (s[i] >= '2' && s[i] <= '9') size /= (s[i] - '0');
        else size--;
    }
    return res;
}

js

/**
 * @param {string} s
 * @param {number} k
 * @return {string}
 */
var decodeAtIndex = function (s, k) {
  let size = 0
  for (const c of s) {
    if (c >= '2' && c <= '9') size *= Number(c)
    else size++
  }
  for (let i = s.length - 1; i >= 0; i--) {
    k %= size
    if (k === 0 && s[i] >= 'a' && s[i] <= 'z') return s[i]
    if (s[i] >= '2' && s[i] <= '9') size /= Number(s[i])
    else size--
  }
  return ''
}

py

class Solution:
    def decodeAtIndex(self, s: str, k: int) -> str:
        size = 0
        for c in s:
            if c.isdigit():
                size *= int(c)
            else:
                size += 1
        for i in range(len(s) - 1, -1, -1):
            k %= size
            if k == 0 and s[i].isalpha():
                return s[i]
            if s[i].isdigit():
                size //= int(s[i])
            else:
                size -= 1

• 时间复杂度：$O(n)$，其中 n 是编码字符串长度
• 空间复杂度：$O(1)$

算法思路：

• 先正向计算解码后的总长度
• 再从后向前遍历，用取模缩小 k，当 k 为 0 且当前是字母时即为答案