Leetcode 每日一题 2022-09-16 880.索引处的解码字符串
这道题我首先的思路当然是遍历一遍字符串,然后求出目标字符串以后,直接输出s[i].但是无奈字符串过长,超时了。
string decodeAtIndex(string s, int k) {
int len=s.size();
int i=0;
string ans="";
while(i<len)
{
char t=s[i];
if( s[i]<='9' && s[i]>='1')
{
string temp=ans;
int n=s[i]-'0'-1;
for(int i=0;i<n;i++)
ans+=temp;
}
else
{
ans=ans+s[i];
}
i++;
}
return string("") + ans[k-1];
}
所以就得另辟蹊径。
我们考虑:目标是要返回第k个字符,所以没必要求出所有字符串,只要动态监控字符即可。
所以可以得到思路:
首先定义len来得到总长度,然后每次用add来表示要增加的量。
如果加入add后没到k,说明还得往上加。
如果刚好是k或者 k%len==0,说明是最后一个字符。
否则,我们求的字符一定在当前「解码字符串」的若干次循环中,问题规模可以缩小为:在长度为len的「解码字符串」中寻找第k % len个字符,递归解决即可。
class Solution {
public:
string decodeAtIndex(string s, int k) {
long long len = 0;
for(int i = 0; i < s.size(); ++i){
long long add = isalpha(s[i]) ? 1 : (s[i] - '1') * len;
if(len + add < k){
len += add;
}else{
if(len + add == k || k % len == 0){
while(!isalpha(s[i])){
--i;
}
return string("") + s[i];
}
return decodeAtIndex(s.substr(0,i),k % len);
}
}
return "";
}
};