lintcode 1602 · 索引处的解码字符串【中等 模拟法 逆向】

题目

https://www.lintcode.com/problem/1602

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

如果所读的字符是字母，则将该字母写在磁带上。
如果所读的字符是数字（例如 d），则整个当前磁带总共会被重复写 d-1 次。
现在，对于给定的编码字符串 S 和索引 K，查找并返回解码字符串中的第 K 个字母。


2 <= S.length <= 100
S 只包含小写字母与数字 2 到 9 。
S 以字母开头。
1 <= K <= 10^9
解码后的字符串保证少于 2^63 个字母。
样例
示例 1：

输入：S = "lint2code3", K = 10
输出："o"
解释：
解码后的字符串为 "lintlintcodelintlintcodelintlintcode"。
字符串中的第 10 个字母是 "o"。
示例 2：

输入：S = "ha22", K = 5
输出："h"
解释：
解码后的字符串为 "hahahaha"。第 5 个字母是 "h"。
示例 3：

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

思路

    //思路
    //如果我们有一个像 appleappleappleappleappleapple 这样的解码字符串和一个像 K=24 这样的索引，
    // 那么如果 K=4，答案是相同的。
    //一般来说，当解码的字符串等于某个长度为 size 的单词重复某些次数
    //（例如 apple 与 size=5 组合重复6次）时，索引 K 的答案与索引 K % size 的答案相同。

    //我们可以通过逆向工作，跟踪解码字符串的大小来使用这种洞察力。
    //每当解码的字符串等于某些单词 word 重复 d 次时，我们就可以将 k 减少到 K % (Word.Length)。

    //首先计算最终字符串的长度。再从后向前处理，如果当前位是数字，
    // 就模前面字符串的总长度，就可以得到在前面哪一位找字符了

答案

public class Solution {
    /**
     * @param s: An encoded string.
     * @param k: An index.
     * @return: Return the K-th letter (1 indexed) in the decoded string.
     */
    public String decodeAtIndex(String s, int k) {
        //思路
        //如果我们有一个像 appleappleappleappleappleapple 这样的解码字符串和一个像 K=24 这样的索引，
        // 那么如果 K=4，答案是相同的。
        //一般来说，当解码的字符串等于某个长度为 size 的单词重复某些次数
        //（例如 apple 与 size=5 组合重复6次）时，索引 K 的答案与索引 K % size 的答案相同。

        //我们可以通过逆向工作，跟踪解码字符串的大小来使用这种洞察力。
        //每当解码的字符串等于某些单词 word 重复 d 次时，我们就可以将 k 减少到 K % (Word.Length)。

        //首先计算最终字符串的长度。再从后向前处理，如果当前位是数字，
        // 就模前面字符串的总长度，就可以得到在前面哪一位找字符了
        long size = 0;
        int n = s.length();
        for (int i = 0; i <n ; i++) {
            char c = s.charAt(i);
            if(c>='0' && c<='9'){
                size*=c-'0';
            }else size++;
        }

        for (int i = n-1; i >=0 ; i--) {
            char c = s.charAt(i);
            k%=size;
            if(k==0 && !(c>='0' && c<='9')){
                return c+"";
            }

            if(c>='0' &&c<='9'){
                size/=c-'0';
            }else
                size--;
        }
        return "";
    }
}