Leetcode 每日一题 2022-09-16 880.索引处的解码字符串

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 "";
    }
};