差分数组和经典例题

差分数组

差分数组https://labuladong.github.io/algo/2/20/25/

例题1

1109. 航班预订统计 - 力扣（LeetCode）

代码：

class Solution {
public:
//差分数组的话那必然在[l,r] 都加上一个数c  <=> nums[l] += c && nums[r + 1] -= c
//如果求出了差分数组，则通过相加可得原数组   c为(差分值) = ret[i] - ret[i - 1], 若要得原数组，则 c(差分值) + ret[i - 1]  (准确的说是求差分的前缀和)
    vector<int> corpFlightBookings(vector<vector<int>>& bookings, int n) {
        vector<int>ret(n,0);                //差分数组
        for (int i = 0; i < bookings.size(); i++) {
            int a = bookings[i][0], b = bookings[i][1], c = bookings[i][2];
            ret[a - 1] += c;
            if (b < n) {
                ret[b] -= c;
            }
            
        }
        for (int i = 1; i < n; i++) {    //求差分的前缀和
            ret[i] += ret[i - 1];
        }
        return ret;
    }
};

例题2

6158. 字母移位 II - 力扣（LeetCode）

代码

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

        for (int i = 1; i < f.size(); i++) {
            f[i] = f[i] + f[i - 1];
        }

        string ans;
        for (int i = 0; i < n; i++) {
            // cout << f[i] << "";
            char c = s[i];
            int d = (c - 'a' + f[i]) % 26;
            char m = (d + 26) % 26 + 'a';             //因为会出现负数，所以才让d 加 26 ，否则只有上一步就行。
            ans.push_back(m);
        }

        return ans;
    }
};