Leetcode 848. 字母移位 C++

题目描述

有一个由小写字母组成的字符串 S，和一个整数数组 shifts。

我们将字母表中的下一个字母称为原字母的 移位（由于字母表是环绕的， ‘z’ 将会变成 ‘a’）。

例如·，shift(‘a’) = ‘b’， shift(‘t’) = ‘u’,， 以及 shift(‘z’) = ‘a’。

对于每个 shifts[i] = x ， 我们会将 S 中的前 i+1 个字母移位 x 次。

返回将所有这些移位都应用到 S 后最终得到的字符串。

示例：

输入：S = “abc”, shifts = [3,5,9]
输出：“rpl”
解释：
我们以 “abc” 开始。
将 S 中的第 1 个字母移位 3 次后，我们得到 “dbc”。
再将 S 中的前 2 个字母移位 5 次后，我们得到 “igc”。
最后将 S 中的这 3 个字母移位 9 次后，我们得到答案 “rpl”。
提示：

1 <= S.length = shifts.length <= 20000
0 <= shifts[i] <= 10 ^ 9

解答

时间复杂度$O(N)$，空间复杂度$O(N)$,原因是原来的vector溢出

class Solution {
public:
    string shiftingLetters(string S, vector<int>& shifts) {
        vector<long> move(shifts.size());//使用int类型计算，下面计算总位数会溢出
        for(int i=0;i<move.size();++i) move[i]=shifts[i];//先将shifts复制到long型vector
        for(int i=move.size()-2;i>=0;i--)
        {
            move[i]=move[i+1]+move[i];//计算每一个字母需要移动的总位数
        }
        
        for(int i=0;i<S.size();++i)
        {
            char c=S[i];
            int cur=c-'a';//cur表示字母在'a','b',...'z'中的位置，0到25
            int n=move[i]%26;//每26次位移回到原来位置，所以取余，计算实际需要移动次数
            if(cur+n>25)//如果需要转头，计算实际位置
            {
                int k=(cur+n)%26;
                S[i]='a'+k;
            }
            else//如果移动没有到达'z',不需要转头，直接加上移动步数n
            {
                S[i]=c+n;
            }
        }
        return S;
    }
};

其实可以简化为如下的代码。执行速度也会稍快一点。

class Solution {
public:
    string shiftingLetters(string S, vector<int>& shifts) {
        vector<long> move(shifts.size());
        for(int i=0;i<move.size();++i) move[i]=shifts[i];
        for(int i=move.size()-2;i>=0;i--)
        {
            move[i]=move[i+1]+move[i];
        }
        
        for(int i=0;i<S.size();++i)
        {
            char c=S[i];
            int cur=c-'a';
            int n=move[i]%26;
            //直接计算实际位置
            int k=(cur+n)%26;
            S[i]='a'+k;
        }
        
        return S;
    }
};