代码训练营第2天打卡【数组part2】 | 977.有序数组的平方 209. 长度最小的子数组 59. 螺旋矩阵2

数组part2 主要是为了理解数组中双指针的操作和实现方法。

一、977 有序数组的平方

题目
问题要点：使用暴力解法，将数组中的所有元素平方后，使用C++ Vector自带的sort()排序，复杂度为 O(nlogn)。而本题目的最优策略是使用双指针，时间复杂度O(n)。由于该数组已经是预先排序好的数组，因此对于平方值，最大的值只会出现在原数组的最左(绝对值最大的负数)和最右(最大的正数)，因此用start 和 end 两个指针分别指向头尾，依次比较他们的最大值放入新的数组中即可。

class Solution {
public:
    int power(int a) {
        return a * a;
    }
    
    vector<int> sortedSquares(vector<int>& nums) {

        int start = 0; 
        int end = nums.size() - 1;
        vector<int> ans;

        // while循环的边界条件很重要
        // start == end时，该元素也要放进数组
        // 当 start = end + 1时，说明已经不合法了
        while(start < end + 1) {
            if(power(nums[start]) > power(nums[end])) {
                ans.insert(ans.begin(), power(nums[start++]));
            } else {
                ans.insert(ans.begin(), power(nums[end--]));
            }
        }

        return ans;

    }
};

二、 209 长度最小的子数组

题目
思路：本题采用滑动窗口思想，但实现的方式其实也是通过双指针。需要注意的是为了保证时间复杂度为On，外圈的判断条件 因该是 滑动窗口的尾节点End，在每次循环内处理时，处理挪动滑动窗口的头节点Start

class Solution {
public:
    int minSubArrayLen(int target, vector<int>& nums) {
        int start = 0;
        int end = 0;
        int sum = 0;
        int ans = INT32_MAX;

        
        for(; end < nums.size(); end++) {
            sum += nums[end];
            while(sum >= target) {
                ans = min(ans, (end - start + 1));
                sum -= nums[start++];
            }
            //printf("%d %d %d %d\n", start, end, sum, ans);
        }
        return ans == INT32_MAX ? 0:ans;
    }
};

滑动窗口扩展题目1：904. 水果成篮

题目
思路：本题也是采用了滑动窗口的思想，不过根据题目中的判断条件，需要用Hashmap来存键值对，代表两个篮子中装有的水果数量。
注意：熟悉C++ 的unordered_map的相关操作和实现细节。

class Solution {
public:
    int totalFruit(vector<int>& fruits) {
        // 使用滑动窗口思想
        int start = 0;
        int ans = 0;

        // 用哈希表存 代表两个水果篮子
        // key:   该种水果类型
        // value: 该种水果出现的次数 
        unordered_map <int, int> basket;

        // tips:
        // 要熟悉下哈希表中对元素的操作如何编写
        // 迭代器iterator的second为该key的value
        for(int end = 0; end < fruits.size(); end++) {
            basket[fruits[end]]++;
            while(basket.size() > 2) {
                auto it = basket.find(fruits[start]);
                it->second--;
                if(it->second == 0) {
                    basket.erase(it);
                }
                start++;
            }
            ans = max(ans, end - start + 1);
        }
        return ans;
    }

};

三、59 螺旋矩阵 2

题目
思路：主要就是用代码实现题目中的操作逻辑，将四个位移方向定义为4个 {x, y} 的数组即可轻松实现。

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {

        vector<vector<int>> m(n, vector<int>(n, 0));
        int length = n*n; 
        int x = 0, y = 0;
        int startDir = 0;
        int nextX, nextY;        

        // 定义四个方向，代表 右，下，左，上
        vector<vector<int>> dirs = {{0,1}, {1, 0}, {0, -1}, {-1, 0}};

        for(int i = 1; i <= length; i++) {
            m[x][y] = i;
            nextX = x + dirs[startDir][0];
            nextY = y + dirs[startDir][1];

            if(nextX >= n || nextX < 0 || nextY >= n || nextY < 0 || m[nextX][nextY] != 0) {
                startDir = (startDir + 1) % 4;
            }

            x += dirs[startDir][0];
            y += dirs[startDir][1];
        }
     
        return m;
    }
};

总结：数组part2的前两道题，主要为了让我熟悉双指针的方法。双指针的时间复杂度一般都为O(n)，注意滑动窗口的循环条件是要以窗口的尾节点为准。