1. 题目分析
1.1. 错误理解
- 大多数人看到这道题的第一反应就是把这个字符串整个还原,再遍历还原后的字符串,就能得到结果了。这是所谓的暴力解法,但是请看清楚,这道题是一道中等题,怎么可能那么简单。
- 若使用暴力解法,提交肯定不会通过,原因是内存溢出,当还原后的字符串很长的时候,就容易造成内存溢出
- 那可能这又有人说了,为了避免这个问题,我们可以多方位思考,因为本体只需要我们去第K个字母,那么我们需要多少就还原多少,没有必要把整个字符串都还原。
- 这样想就太天真了,当K值很大的时候,我们依然需要很长的字符串,依然会溢出,那我现在给出这道题的正确思路。
2.2. 正确思路
- 举个例子,当S = “abcabcabc” 的时候,我们取其中第2个数和第5个数的结果是一样的。那么这道题的解题思路就有了,S="abc3"的时候,若我们此时要取第5个数,就直接5%3 = 2,实际上就是第2个数,也就是b。
- 倒着遍历这个字符串,可以更早遇到数字,之后再用K与这个数字进行取模运算。
- 说了这么多晦涩的对象,还是不如来一套图解。
为了方便画图,我将这几个字母分隔了远了一些。
- 第一步,虽然我们不能将字符串还原,但是我们可以将还原后的字符串的长度计算出来,计为size,从后往前遍历,指针先指向a,K=10
- 第二步,K%size = 10(我们的目的是要让这个数字等于0,当这个数字为0的时候,说明K的位置就是我们想要的值),指针左移,szie -= 1
- 第三步,我们继续移动指针,继续第二步的操作,一直到移动的到3为止,3是个整数,而这个时候size = 36,根据我们上面的分析,我们可以用size = size/3
- 第四步,虽然我们移动了这么多次,但是K%size = 10这个结果还是没有变,所以我们继续移动,每移动一位size还是要减1,我们发现当我们移动两次之后,size变成了10,而K%size=0,这就说明,结果就是e向左移动两次指向的o
这其实就是一个逆向的操作,有了这些思想,我们来用代码实现。
2. 代码实现
2.1. Python代码
class Solution(object):
def decodeAtIndex(self, S, K):
"""
:type S: str
:type K: int
:rtype: str
"""
size = 0
for i in S:
if i.isdigit():
size = size * int(i)
else:
size += 1
for i in reversed(S):
K %= size
if(K == 0 and i.isalpha()):
return i
else:
if(i.isdigit()):
size /= int(i)
else:
size -= 1
2.2. 代码实现
class Solution {
public String decodeAtIndex(String S, int K) {
long size = 0;
int N = S.length();
for(int i = 0;i < N;i++){
char c = S.charAt(i);
if(Character.isDigit(c)){
size *= c-'0';
}else{
size++;
}
}
for(int i = N - 1;i >= 0;--i){
char c = S.charAt(i);
K %= size;
if(K == 0 && Character.isLetter(c)){
return String.valueOf(c);
}
else{
if(Character.isDigit(c)){
size/=(c - '0');
}else{
size--;
}
}
}
throw null;
}
}