LeetCode 880 索引处的解码字符串 HERODING的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 个字母。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/decoded-string-at-index
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路：
首先我想到的是栈的结构，因为与计算器的实现类似，字母相当于计算器中的数字，数字相当于计算器中的运算符，但是由于我的时间复杂度为O(n3)，所以超时了。。。究其原因还是我把整个序列都存下来了，这大可不必，因为最后输出的结果只在初始的S中，所以更换方法（但是不得不说这个思路还是很有意思的）。其实O(n)的方法就可以解决该问题，正向遍历获取整个index的长度，反向遍历不断回退，直到找到K的位置，代码如下：

class Solution {
public:
    string decodeAtIndex(string S, int K) {
        // 如果是第一个字母
        int len = S.length();
        string res = "";
        if(K == 1) {
            return res + S[0];
        }
        stack<char> s;
        int index = 0;
        for(int i = 0; i < len; i ++) {
            if(S[i] - '0' > 1 && S[i] - '0' <= 9) {
                int num = S[i] - '0';
                index *= num;
                vector<char> temp;
                while(!s.empty()) {
                    char c = s.top();
                    s.pop();
                    temp.push_back(c);
                }
                while((--num) >= 0) {
                    for(int j = temp.size() - 1; j >= 0; j --) {
                        s.push(temp[j]);
                    }
                }
            } else {
                index ++;
                s.push(S[i]);
            }
            if(index > K) {
                break;
            }
        }
        int length = index - K;
        while(length--) {
            s.pop();
        } 
        return res + s.top();
    }
};

class Solution {
public:
    string decodeAtIndex(string S, int K) {
        long size = 0;
        int N = S.size();
        // 存储最后的长度
        for (int i = 0; i < N; ++i) {
            // 遇到数字按照数字面值翻倍
            if (isdigit(S[i]))
                size *= S[i] - '0';
            else // 遇到字符++
                size++;
        }
        // 从后往前遍历
        for (int i = N-1; i >=0; --i) {
            K %= size;
            // 如果当前位置就是想要输出的位置且不是数字
            if (K == 0 && isalpha(S[i]))
                return (string) "" + S[i];
            // 如果是数字那么降低倍数
            if (isdigit(S[i]))
                size /= S[i] - '0';
            else // 字符的话就--
                size --;
        }
        return "";
    }
};

最后我还进行了小小的优化，就是在遍历到大于K的时候直接break，掉头查询，代码如下：

class Solution {
public:
    string decodeAtIndex(string S, int K) {
        long size = 0;
        int N = S.size();
        // 存储最后的长度
        int i;
        for (i = 0; i < N; ++i) {
            // 遇到数字按照数字面值翻倍
            if (isdigit(S[i]))
                size *= S[i] - '0';
            else // 遇到字符++
                size++;
            if(size > K) break;
        }
        // 从后往前遍历
        i = min(i, N - 1);
        for (int j = i; j >=0; --j) {
            K %= size;
            // 如果当前位置就是想要输出的位置且不是数字
            if (K == 0 && isalpha(S[j]))
                return (string) "" + S[j];
            // 如果是数字那么降低倍数
            if (isdigit(S[j]))
                size /= S[j] - '0';
            else // 字符的话就--
                size --;
        }
        return "";
    }
};