【数据结构-差分】力扣2381. 字母移位 II

给你一个小写英文字母组成的字符串 s 和一个二维整数数组 shifts ，其中 shifts[i] = [starti, endi, directioni] 。对于每个 i ，将 s 中从下标 starti 到下标 endi （两者都包含）所有字符都进行移位运算，如果 directioni = 1 将字符向后移位，如果 directioni = 0 将字符向前移位。

将一个字符 向后 移位的意思是将这个字符用字母表中 下一个 字母替换（字母表视为环绕的，所以 ‘z’ 变成 ‘a’）。类似的，将一个字符 向前 移位的意思是将这个字符用字母表中 前一个 字母替换（字母表是环绕的，所以 ‘a’ 变成 ‘z’ ）。

请你返回对 s 进行所有移位操作以后得到的最终字符串。

示例 1：
输入：s = “abc”, shifts = [[0,1,0],[1,2,1],[0,2,1]]
输出：“ace”
解释：首先，将下标从 0 到 1 的字母向前移位，得到 s = “zac” 。
然后，将下标从 1 到 2 的字母向后移位，得到 s = “zbd” 。
最后，将下标从 0 到 2 的字符向后移位，得到 s = “ace” 。

示例 2:
输入：s = “dztz”, shifts = [[0,0,0],[1,1,1]]
输出：“catz”
解释：首先，将下标从 0 到 0 的字母向前移位，得到 s = “cztz” 。
最后，将下标从 1 到 1 的字符向后移位，得到 s = “catz” 。

差分

class Solution {
public:
    string shiftingLetters(string s, vector<vector<int>>& shifts) {
       int n = s.size(), m = shifts.size();
        vector<int> a(n+1);
        for(int i = 0; i < m; i++){
            int tem = 1;
            if(shifts[i][2] == 0){
                tem = -1;
            }
            a[shifts[i][0]]+=tem;
            a[shifts[i][1]+1]-=tem;
        }

        int sum = a[0];         
        for(int i = 0; i < n; i++){
            int k = (s[i] - 'a' + sum) % 26;
            if(k < 0){
                k += 26;
            }
            s[i] = 'a' + k;
            sum += a[i+1];
        }
        return s;
    }
};

用差分的思想，首先判断差分数组到底是+1还是-1由shift[2]的是0还是1来决定。我们规定了向右移动时候为正，向左移动为负。根据shift构造出差分数组后，就可以知道每个字符是如何位移的。我们初始化sum为a[0]，然后定义k为偏移量，当k小于0的时候，就令他加上26变为正数，然后我们计算每个字符的偏移结果s[i] = 'a' + k;。每次循环结束，sum+=a[i+1]，用于计算下个字符的偏移量。