《代码随想录》Day2: 977. 有序数组的平方 + 209. 长度最小的子数组 + 59. 螺旋矩阵Ⅱ

《代码随想录》Day2

1. Q977. 有序数组的平方

给定一个非递减顺序的整数数组nums，返回由该数组每个数字的平方组成的新数组，且也为非递减顺序

暴力解法

直接平方后用sorted函数，但时间复杂度比较高

class Solution:
    def sortedSquares(self, nums: List[int]) -> List[int]:
        return sorted(num*num for num in nums)

双指针法：两端向中间找

利用数组有序（非递减顺序）的条件，原数组平方的最大值只能来自原数组的两端（考虑负数平方后可能为最大值的情况），则可利用双指针，a从数组前端开始，b从数组后端开始，比较出较大值后放入新数组末尾空位。

class Solution:
    def sortedSquares(self, nums: List[int]) -> List[int]:
        n = len(nums)
        a, b, i = 0, n-1, n-1
        ans = [0] * n #由于是从后往前依次填充，所以需要先把list结构和大小先定义好
        while a <= b :  #可等于
            if nums[a]*nums[a] <= nums[b]*nums[b]: #则先将b放入新数组
                ans[i] = nums[b]*nums[b]
                b -= 1 #b从最右端向中间移
            elif nums[a]*nums[a] > nums[b]*nums[b]: #则先将a放入新数组
                ans[i] = nums[a]*nums[a]
                a += 1 #从最左端向中间移
            i -= 1  #i可往前移一位,且不管跳入哪个循环都能往前一位
        return ans

双指针法：从正负交接往两边找

可理解为从中间向两边找，中间为正负组交界处。先通过enumerate遍历函数找到边界neg，在从neg出发向前找，neg+1出发向后找，比较出较小值后依次填充入新数组中

• enumerate(nums): 遍历函数，可返回数组nums的下标和对应数据
• 该方法要考虑边界问题，若负数组到边界，后续就是正数组往后加，反之亦然
• 且情况讨论时，需要先把边界情况定义好，否则可能会出现IndexError: list index out of range的报错

class Solution:
    def sortedSquares(self, nums: List[int]) -> List[int]:
        #先确定边界
        neg = -1 ##若没有负数时
        for i, num in enumerate(nums): #先确定边界
            if num < 0:
                neg = i
            else:
                break
        n = len(nums)
        a, b = neg, neg + 1
        ans = list() #因是从前往后填充，可直接list.append
        while a >= 0 or b < n:  #只要满足其中一个条件，就可以接着填数；不满足某个条件是因为该组以达到边界
            if a < 0: #负数组已达到边界，后续就是正数组往后加
                ans.append(nums[b]*nums[b])  
                b += 1
            elif b >= n: #正数组已达到边界，后续就是负数组往后加
                ans.append(nums[a]*nums[a])
                a -= 1
            #需要先把边界情况定义好，否则可能会出现IndexError: list index out of range的报错
            elif nums[a]*nums[a] <= nums[b]*nums[b]: #则先将a放入新数组
                ans.append(nums[a]*nums[a])
                a -= 1
            elif nums[a]*nums[a] > nums[b]*nums[b]: #则先将b放入新数组
                ans.append(nums[b]*nums[b])  
                b += 1
        return ans

2. Q209. 长度最小的子数组

给定一个含有 n 个正整数的数组和一个正整数 s ，找出该数组中满足其和 ≥ s 的长度最小的连续子数组，并返回其长度。如果不存在符合条件的子数组，返回 0。

暴力解法

用两个for循环嵌套，第一个设定起点i，第二个设定终点j，若满足和>target，则ans为满足条件的跨度区间最小值

class Solution:
    def minSubArrayLen(self, target: int, nums: List[int]) -> int:
        n = len(nums)
        ans = n+1 #ans的最大值比较限制，也为若找不到满足条件和的判断标准
        for i in range(n): #设定起点
            total = 0
            for j in range(i,n): #设定终点
                total += nums[j]
                if total >= target:
                    ans = min (ans, j-i+1) #筛选跨度区间最小值
                    break
        return 0 if ans == n+1 else ans

滑动窗口

先固定滑动窗口终止位置b，以数组开头为滑动窗口起始位置a，若a到b和小于target则继续向后移动终止位置b，若大于target则后移a，一点点缩小窗口范围，看是否还符合条件

class Solution:
    def minSubArrayLen(self, target: int, nums: List[int]) -> int:
        n = len(nums)
        ans = n+1
        a = 0
        total = 0
        for b in range(n):
            total += nums[b]
            while total >=target: #若满足条件，则移动a缩小范围
                ans = min(ans, b-a+1)
                total -= nums[a] #剔除一位数字
                a += 1
        return 0 if ans == n+1 else ans

3. Q59. 螺旋矩阵Ⅱ

给一个正整数 n ，生成一个包含 1 到 $n^2$所有元素，且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。

模拟法

画螺旋式可以上行从左到右，右列从上到下，下行从右到左，左列从下到上，一圈圈往内画，且最好用左闭右开区间

• 4种情况的讨论比较繁琐，需要注意i. j的增减
• range(start, end, n) 函数是左闭右开区间，不包括end， n 为每次移动步距，默认为1，若从大到小 range 需设定 n=-1

class Solution:
    def generateMatrix(self, n: int) -> List[List[int]]:
        left, right, top, bottom = 0, n-1, 0, n-1 #划分边界，左右为列数，上下为行数
        count, stop = 1, n*n #设定起始值和终点值
        res = [[0] * n for _ in range(n)]
        while count <= stop:
            #从左向右填充最上列
            for j in range(left,right+1):
                res[top][j] = count
                count +=1
            top += 1 #上边界已填好一行
            #从上向下填充最右列
            for i in range(top,bottom+1):
                res[i][right] = count
                count +=1  
            right -= 1 #右边已填好一列
            #从右向左填充最下列
            for j in range(right,left-1,-1):  #range为从大到小
                res[bottom][j] = count
                count +=1
            bottom -= 1 #下边界已填好一行
            #从下往上填充最左列
            for i in range(bottom,top-1,-1):
                res[i][left] = count
                count +=1 
            left += 1 #右边已填好一列
        return res