算法训练Day 2

LeetCode 977. 有序数组的平方

题目链接：977. 有序数组的平方

思路：观察了一下示例，发现平方之后数值最大的一定在数组的两侧，而平方之后数值最小的一定在数组的中间，天才在左，疯子在右？(bushi 。 所以考虑使用双指针，分别指向数组的首尾两端。但写代码的时候发现还需要一个指针，因为我们最一开始循环比较的时候，得到的是最大的数（应该放在结果的尾端），所以这个指针一开始应该指向结果数组的尾端。

Python版本：

class Solution:
    def sortedSquares(self, nums: List[int]) -> List[int]:
        n = len(nums)
        res = [0]*n
        k = n-1
        l, r = 0, n-1
        while l <= r:
            if nums[l]**2 > nums[r]**2:
                res[k] = nums[l]**2
                l += 1
            else:
                res[k] = nums[r]**2
                r -= 1
            k -= 1

        return res

时间复杂度：$O(n)$，空间复杂度：$O(1)$

go版本：

func sortedSquares(nums []int) []int {
    n := len(nums)
    res := make([]int, n)
    l := 0
    r := n-1
    for k := n-1; l<=r; k-- {
        if nums[l]*nums[l] > nums[r]*nums[r] {
            res[k] = nums[l]*nums[l]
            l++
        } else {
            res[k] = nums[r]*nums[r]
            r--
        }
    }
    return res
}

go写两数平方不如python方便，但大括号yyds

LeetCode 209. 长度最小的子数组

题目链接：209. 长度最小的子数组

思路：题目要求连续子数组，看到连续想到使用滑动窗口。大体思路就是先扩展窗口（右移右边界right），直到窗口内数组和大于等于target了，再缩小窗口（右移左边界left），遍历完整个数组寻找其中满足要求且长度最小的窗口。

Python版本：

class Solution:
    def minSubArrayLen(self, target: int, nums: List[int]) -> int:
        n = len(nums)
        # 这个地方只要把res设置成比n大的数即可
        res = float('inf')
        left, right = 0, 0
        cur = 0
        while right < n:
            cur += nums[right]
            while cur >= target:
                res = min(res, right-left+1)
                cur -= nums[left]
                left += 1
            right += 1

        return res if res != float('inf') else 0

时间复杂度：$O(n)$，空间复杂度：$O(1)$

go版本：

func minSubArrayLen(target int, nums []int) int {
    n := len(nums)
    res := n+1
    r, l, cur := 0, 0, 0
    for r < n {
        cur += nums[r]
        for cur >= target {
            if res > r-l+1 {
                res = r-l+1
            }
            cur -= nums[l]
            l++
        }
        r++
    }
    if res == n+1 {
        return 0
    } else {
        return res
    }
}

————————分割线————————
20220509：第二次写的时候，内循环的判断写成了cur>target，当时的想法是数组元素全是正整数，那当cur==target的时候，再减掉nums[l]肯定就小于target了。这里问题出在忽略了if判断，当cur == target的时候，if会首先判断r-l+1与res的长度，之后才会相减。

LeetCode 59. 螺旋矩阵 II

题目链接：59. 螺旋矩阵 II

思路：按顺时针顺序螺旋排列的单层逻辑为：右移、下移、左移、上移，如此往复。

tips： 如果循环的边界条件不好想的话，可以直接思考第一次循环的各个边界值。例如，第一步是从$matrix[0][0]$的位置走到$matrix[0][n-1]$的位置，第二步从$matrix[1][n-1]$走到$matrix[n-1][n-1]$，第三步从$matrix[n-1][n-2]$走到$matrix[n-1][0]$，最后一步从$matrix[n-2][0]$走到$matrix[1][0]$。这个过程可以辅助我们来确定循环的边界值。

Python版本：

class Solution:
    def generateMatrix(self, n: int) -> List[List[int]]:
        left, top = 0, 0
        right, bottom = n-1, n-1
        matrix = [[0]*n for _ in range(n)]
        i = 1
        while left<=right and top<=bottom:
            for col in range(left, right+1):
                matrix[top][col] = i
                i += 1

            for row in range(top+1, bottom+1):
                matrix[row][right] = i
                i += 1

            for col in range(right-1, left, -1):
                matrix[bottom][col] = i
                i += 1

            for row in range(bottom, top, -1):
                matrix[row][left] = i
                i += 1

            left += 1
            right -= 1
            top += 1
            bottom -= 1

        return matrix

时间复杂度：$O(n^2)$，空间复杂度：$O(1)$

go版本：

func generateMatrix(n int) [][]int {
    left, right := 0, n-1
    top, bottom := 0, n-1
    i := 1
    matrix := make([][]int, n)
    for i := range matrix {
        matrix[i] = make([]int, n)
    }

    for left<=right && top<=bottom {
        for col:=left;col<=right;col++ {
            matrix[top][col] = i
            i++
        }
        for row:=top+1;row<=bottom;row++ {
            matrix[row][right] = i
            i++
        }
        for col:=right-1;col>=left;col-- {
            matrix[bottom][col] = i
            i++
        }
        for row:=bottom-1;row>=top+1;row-- {
            matrix[row][left] = i
            i++
        }
        left++
        right--
        top++
        bottom--
    }
    return matrix
}

————————分割线————————
20220509：上文的tips写的真好啊，严格按照tips说的走一轮，判断条件和赋值对象才不容易错。