【Leetcode】880. Decoded String at Index

记录算法题解

于 2023-01-08 10:00:01 发布

阅读量412

点赞数

分类专栏： LC 二分、位运算与数学文章标签： leetcode 算法 c++ 数据结构

本文链接：https://blog.csdn.net/qq_46105170/article/details/128598456

版权

LC 二分、位运算与数学专栏收录该内容

385 篇文章 2 订阅

订阅专栏

该文介绍了一道LeetCode的编程题，涉及字符串解码问题。给定一个含有字母和数字的编码字符串，需要找出解码后字符串的第k个字符。通过从后向前遍历，递归地解码并计算每个阶段的字符串长度，最终找到目标字符。时间复杂度为O(n)，空间复杂度为O(1)。

摘要由CSDN通过智能技术生成

题目地址：

https://leetcode.com/problems/decoded-string-at-index/description/

给定一个长 $n$ 的编码过的字符串 $s$ ，其只含英文小写字母和 $2\sim 9$ 的数字，其还原方式如下：读到字符则记录，读到数字 $c$ 则将记录好的字符串重复 $c$ 遍。问还原后的字符串的第 $k$ 个字符是什么。题目保证 $s$ 以字母开头，且 $k$ 一定有效。计数从 $1$ 开始。

先求还原后的总长度 $m$ ，然后我们一步步“加密”回来寻找答案，加密的过程即为从后向前遍历 $s$ 的过程，也是递归的思想。从后向前遍历 $s$ ，如果遇到数字 $c$ ，则说明加密一次后字符串变为了 $m / c$ ，并且第 $k$ 个字符即为加密后的第 $k\mod (m/c)$ 个字符（如果余数是 $0$ 则说明是最后一个字符），所以我们考虑加密 $1$ 次后的新字符串，其长度即为 $m / c$ ，我们要求新字符串的第 $k\mod (m/c)$ 个字符（余数是 $0$ 则是最后一个字符）；如果遇到的是字符，此时如果 $m = k$ ，则当前字符就是我们要找的字符，否则我们考虑少了最后那个字符的新字符串，其长度为 $m - 1$ ，我们还是要找第 $k$ 个字符。无论怎样，问题的规模都会缩小。代码如下：

class Solution {
 public:
  string decodeAtIndex(string s, int k) {
    long n = 0;
    for (char ch : s) 
      if (isdigit(ch)) n *= ch - '0';
      else n++;
    
    for (int i = s.size() - 1; i >= 0; i--) {
      char ch = s[i];
      if (isdigit(ch)) {
        int x = ch - '0';
        // 我们要在新的长n/x的串里找第k%n个字符 
        n /= x;
        k %= n;
        if (!k) k = n;
      } else {
        if (n == k) return string(1, ch);
        n--;
      }
    }

    return "";
  }
};