一 有序数组的平方
题目链接:977.有序数组的平方
思路
1.暴力解法
先遍历所有元素,平方后重新排序
时间复杂度是 O(n + nlogn)
排序的写法:
Python:
nums.sort()
2.双指针法
数组其实是有序的, 只不过负数平方之后可能成为最大数了。
那么数组平方的最大值就在数组的两端,不是最左边就是最右边,不可能是中间。
此时可以考虑双指针法了,i指向起始位置,j指向终止位置。
定义一个新数组result,和A数组一样的大小,让k指向result数组终止位置。
如果A[i] *A[i] < A[j] * A[j] 那么result[k–] = A[j] * A[j]; 。
如果A[i]* A[i] >= A[j] * A[j] 那么result[k–] = A[i] * A[i]; 。
如动画所示:
时间复杂度为O(n)
代码
class Solution(object):
# 暴力法
# def sortedSquares(self, nums):
# """
# :type nums: List[int]
# :rtype: List[int]
# """
# return sorted(x*x for x in nums)
# 双指针法
def sortedSquares(self,nums):
l,r,k = 0,len(nums)-1,len(nums)-1
results = []
for i in range(len(nums)):
results.append(0)
while l <= r :
if nums[l] * nums[l] <= nums[r] * nums[r]:
results[k] = nums[r] * nums[r]
k-=1
r-=1
else:
results[k] = nums[l] * nums[l]
k-=1
l+=1
return results
二 长度最小的子数组
题目链接:209.长度最小的子数组
思路
1.暴力解法
两个for循环,不断寻找符合条件的子序列
时间复杂度O(n^2) 超时
2.滑动窗口
滑动窗口,就是不断的调节子序列的起始位置和终止位置,从而得出我们要想的结果。
只用一个for循环,这个循环的索引,表示 滑动窗口的终止位置。
滑动窗口也可以理解为双指针法的一种,只不过这种解法更像是一个窗口的移动,所以叫做滑动窗口更适合一些。
在本题中实现滑动窗口,主要确定如下三点:
窗口内是什么?
如何移动窗口的起始位置?
如何移动窗口的结束位置?
窗口就是 满足其和 ≥ s 的长度最小的 连续 子数组。
窗口的起始位置如何移动:如果当前窗口的值大于等于s了,窗口就要向前移动了(也就是该缩小了)。
窗口的结束位置如何移动:窗口的结束位置就是遍历数组的指针,也就是for循环里的索引。
解题的关键在于 窗口的起始位置如何移动,如图所示:
滑动窗口的精妙之处在于根据当前子序列和大小的情况,不断调节子序列的起始位置。从而将O(n^2)暴力解法降为O(n)。
时间复杂度:O(n)
代码
class Solution(object):
def minSubArrayLen(self, target, nums):
"""
:type target: int
:type nums: List[int]
:rtype: int
"""
### 滑动窗口法
i,sum = 0,0
l = len(nums)
result = float('inf')
for j in range(l):
sum += nums[j]
while (sum >= target):
length = j - i + 1
sum -= nums[i]
i += 1
if result > length:
result = length
return result if result != float('inf') else 0
# ### 暴力求解法 超时
# result = float('inf')
# l = len(nums)
# for i in range(l):
# sum = 0
# for j in range(i,l):
# sum += nums[j]
# if (sum>=target):
# lenth = j - i + 1
# if lenth < result:
# result = lenth
# return result if result != float('inf') else 0
三 螺旋矩阵 II
题目链接:59.螺旋矩阵 II
思路
本题并不涉及到什么算法,就是模拟过程,但却十分考察对代码的掌控能力。
二分法,提到如果要写出正确的二分法一定要坚持循环不变量原则。
而求解本题依然是要坚持循环不变量原则。
模拟顺时针画矩阵的过程:
- 填充上行从左到右
- 填充右列从上到下
- 填充下行从右到左
- 填充左列从下到上
由外向内一圈一圈这么画下去。
一圈下来,我们要画每四条边,这四条边怎么画,每画一条边都要坚持一致的左闭右开,或者左开右闭的原则,这样这一圈才能按照统一的规则画下来。
按照左闭右开的原则,来画一圈:
时间复杂度 O(n^2): 模拟遍历二维矩阵的时间
代码
class Solution(object):
def generateMatrix(self, n):
"""
:type n: int
:rtype: List[List[int]]
"""
startx , starty = 0, 0
count= 1
loop = n // 2
mid = n // 2
nums = [[0]*n for _ in range(n)]
for offset in range(1,loop+1):
for i in range(starty, n - offset):
nums[startx][i] = count
count += 1
for i in range(startx, n - offset):
nums[i][n - offset] = count
count += 1
for i in range(n - offset, starty, -1):
nums[n - offset][i] = count
count += 1
for i in range(n - offset, startx, -1):
nums[i][starty] = count
count += 1
startx += 1
starty += 1
if n%2 == 1:
nums[mid][mid] = n**2
return nums