数组(二)

前言

代码随想录day2，皆为LeetCode题。

共3题，知识点如下。

双指针：
977. 有序数组的平方
209. 长度最小的子数组

模拟:
59. 螺旋矩阵 II

模版

双指针

for (int i = 0, j = 0; i < n; i ++ )
{
    while (j < i && check(i, j)) j ++ ;

    // 具体问题的逻辑
}
常见问题分类：
    (1) 对于一个序列，用两个指针维护一段区间
    (2) 对于两个序列，维护某种次序，比如归并排序中合并两个有序序列的操作

977. 有序数组的平方

前面递减，后面递增，两个有序数组归并，

两个指针分别从两端开始走，每次比较一下哪个元素更大，大的放后面，直到两指针相遇

res存答案，从后往前存【大的放后面】，res[k]

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
        int n = nums.size();
        vector<int> res(n);

        for (int i = 0, j = n - 1, k = n - 1; i <= j; ) {
            if (nums[i] * nums[i] > nums[j] * nums[j]) {
                res[k] = nums[i] * nums[i];
                i++, k--;
            } else {
                res[k] = nums[j] * nums[j];
                j--, k--;
            }
        }
        return res;
    }
};

随想录代码

class Solution {
public:
    vector<int> sortedSquares(vector<int>& A) {
        for (int i = 0; i < A.size(); i++) {
            A[i] *= A[i];
        }
        sort(A.begin(), A.end()); // 快速排序
        return A;
    }
};
// 暴力O(n + nlog n)

class Solution {
public:
    vector<int> sortedSquares(vector<int>& A) {
        int k = A.size() - 1;
        vector<int> result(A.size(), 0);
        for (int i = 0, j = A.size() - 1; i <= j;) { // 注意这里要i <= j，因为最后要处理两个元素
            if (A[i] * A[i] < A[j] * A[j])  {
                result[k--] = A[j] * A[j];
                j--;
            }
            else {
                result[k--] = A[i] * A[i];
                i++;
            }
        }
        return result;
    }
};
// O(n)

209. 长度最小的子数组

暴力做法：枚举两个端点，然后求一下和

优化：考虑有没有单调性，然后用双指针

class Solution {
public:
    int minSubArrayLen(int target, vector<int>& nums) {
        int res = INT_MAX; // (1)
        for (int i = 0, j = 0, sum = 0; i < nums.size(); i++) {
            sum += nums[i];
            while (sum - nums[j] >= target) sum -= nums[j++]; // (2)
            if (sum >= target) res = min(res, i - j + 1); // (3)
        }
        if (res == INT_MAX) res = 0; // (4)
        return res;
    }
};

代码解析
（1）：初始化为一个最大值
（2）：遍历到满足语句时，逐个判断每个j是否符合

（3）：求出长度的最小值，如最小范围下标为[2,5]，则长度为4
（4）：没求出最小长度，即无解

随想录代码

class Solution {
public:
    int minSubArrayLen(int s, vector<int>& nums) {
        int result = INT32_MAX;
        int sum = 0; // 滑动窗口数值之和
        int i = 0; // 滑动窗口起始位置
        int subLength = 0; // 滑动窗口的长度
        for (int j = 0; j < nums.size(); j++) {
            sum += nums[j];
            // 注意这里使用while，每次更新 i（起始位置），并不断比较子序列是否符合条件
            while (sum >= s) {
                subLength = (j - i + 1); // 取子序列的长度
                result = result < subLength ? result : subLength;
                sum -= nums[i++]; // 这里体现出滑动窗口的精髓之处，不断变更i（子序列的起始位置）
            }
        }
        // 如果result没有被赋值的话，就返回0，说明没有符合条件的子序列
        return result == INT32_MAX ? 0 : result;
    }
};

扩展
904.水果成篮
76.最小覆盖子串

59. 螺旋矩阵 II

定义四个方向为数组，到边界了转方向，右下左上

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> res(n, vector<int>(n)); // (1)

        int dx[] = {0, 1, 0, -1}, dy[] = {1, 0, -1, 0}; // (2)

        for (int i = 1, x= 0, y = 0, d = 0; i <= n * n; i++) { 
            res[x][y] = i; 
            int a = x + dx[d], b = y + dy[d];
            if (a < 0 || a >= n || b < 0 || b >= n || res[a][b]) { // (3)
                d = (d + 1) % 4; 
                a = x + dx[d], b = y + dy[d];  // (4)
            }
            x = a, y = b;
        }
        return res;
    }
};

代码解析
（1）：初始化结果数组
（2）：定义x，y的偏移量，方向分别为右下左上

如图，右下左上偏移量分别对应，矩阵：x行，y列
dx[] = {0, 1, 0, -1}, dy[] = {1, 0, -1, 0}

（3）：超出范围或者填过就拐弯

（填过：撞到上一层的了。比如8往上是1，填过，拐弯，9填右边)

（4）：更新x y的值，用d+1模4来改变各方向的偏移量实现坐标的增加

随想录代码

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> res(n, vector<int>(n, 0)); // 使用vector定义一个二维数组
        int startx = 0, starty = 0; // 定义每循环一个圈的起始位置
        int loop = n / 2; // 每个圈循环几次，例如n为奇数3，那么loop = 1 只是循环一圈，矩阵中间的值需要单独处理
        int mid = n / 2; // 矩阵中间的位置，例如：n为3， 中间的位置就是(1，1)，n为5，中间位置为(2, 2)
        int count = 1; // 用来给矩阵中每一个空格赋值
        int offset = 1; // 需要控制每一条边遍历的长度，每次循环右边界收缩一位
        int i,j;
        while (loop --) {
            i = startx;
            j = starty;

            // 下面开始的四个for就是模拟转了一圈
            // 模拟填充上行从左到右(左闭右开)
            for (j = starty; j < n - offset; j++) {
                res[startx][j] = count++;
            }
            // 模拟填充右列从上到下(左闭右开)
            for (i = startx; i < n - offset; i++) {
                res[i][j] = count++;
            }
            // 模拟填充下行从右到左(左闭右开)
            for (; j > starty; j--) {
                res[i][j] = count++;
            }
            // 模拟填充左列从下到上(左闭右开)
            for (; i > startx; i--) {
                res[i][j] = count++;
            }

            // 第二圈开始的时候，起始位置要各自加1， 例如：第一圈起始位置是(0, 0)，第二圈起始位置是(1, 1)
            startx++;
            starty++;

            // offset 控制每一圈里每一条边遍历的长度
            offset += 1;
        }

        // 如果n为奇数的话，需要单独给矩阵最中间的位置赋值
        if (n % 2) {
            res[mid][mid] = count;
        }
        return res;
    }
};

扩展
54.螺旋矩阵
剑指Offer 29.顺时针打印矩阵

数组总结