代码随想录算法训练营day02 | 977.有序数组的平方 ,209.长度最小的子数组 ,59.螺旋矩阵II
977.有序数组的平方
教程视频:https://www.bilibili.com/video/BV1QB4y1D7ep/?vd_source=ddffd51aa532d23e6feac69924e20891
注意最后一行:请你设计时间复杂度为 O(n) 的算法解决本问题
解法一:从两侧逼近中心(while循环)
public int[] sortedSquares(int[] nums) {
//从两侧逼近中心
//创建空数组及其待更新的下标
int[] arr = new int[nums.length];
int index = nums.length -1;
//定义左右指针
int l = 0;
int r = nums.length -1;
//循环遍历,左右指针重合时结束
while(l<=r){ //最后一个元素也要存入结果数组,此处需写等号
if(nums[l]*nums[l]>nums[r]*nums[r]){
arr[index] = nums[l]*nums[l];
index-=1;
l+=1;
}else{
arr[index] = nums[r]*nums[r];
index-=1;
r-=1;
}
}
return arr;
}
解法二:从两侧逼近中心(for循环)
public int[] sortedSquares(int[] nums) {
//创建空数组及其待更新的下标
int[] result = new int[nums.length];
int k = nums.length -1;
for(int i=0, j=nums.length -1; i<=j; ){
if(nums[i]*nums[i]>nums[j]*nums[j]){
result[k--] = nums[i]*nums[i];
i++;
}else{
result[k--] = nums[j]*nums[j];
j--;
}
}
return result;
}
209.长度最小的子数组
教程视频:https://www.bilibili.com/video/BV1tZ4y1q7XE/?vd_source=ddffd51aa532d23e6feac69924e20891
解法一:双重for循环
public int minSubArrayLen(int target, int[] nums) {
int length = nums.length+1;
for(int i=0;i<nums.length;i++){
int count=0;
//如果加到最后一位仍未找到子串,强制退出循环
int j;
for(j=i;j<nums.length;j++){
count+=nums[j];
//如果找到子串,判断最小字串长度是否更新后退出for循环
if(count>=target){
if(length>j-i+1){
length = j-i+1;
}
break;
}
}
//如果有一次累加到尾部仍然未找到子串,则后面不可能出现子数组,无需继续遍历
if(count<target && j==nums.length){
break;
}
}
//如果不存在符合条件的子数组,返回0
if(length == nums.length+1){
length = 0;
}
return length;
}
解法二(最优解):滑动窗口(双指针)
public int minSubArrayLen(int target, int[] nums) {
int i=0, j=0;
int count = 0;
int length = nums.length+1;
for(i=0; i<nums.length; i++){ //外层循环i为快指针
count+=nums[i];
//若找到子串,计算子串长度,更新慢指针j
while(count>=target){ //这里要用while,因为不确定慢指针要移动多少位
if(length>i-j+1){
length = i-j+1;
}
count-=nums[j];
j++;
}
}
if(length == nums.length+1){
length = 0;
}
return length;
}
977.有序数组的平方
教程视频:https://www.bilibili.com/video/BV1SL4y1N7mV/
解法
- 把握循环条件,遍历起始位置 和 每条边长度。
- 注意中间点位置是否赋值。
public int[][] generateMatrix(int n) {
int[][] result = new int[n][n];
int x=0,y=0; //设置初始位置
int offset = 1; //控制每条边的偏移量
int count=1; // 用于传入数组的变量值
//创建循环变量
int circle=0;
int i,j;
while(circle<n/2){ //以旋转的圈数作为判断条件
//采用左闭右开的思路
for(j=y;j<n-offset;j++){ //上边界
result[x][j] = count++;
}
for(i=x;i<n-offset;i++){ //右边界
result[i][j] = count++;
}
for(j=j;j>x;j--){
result[i][j] = count++;
}
for(i=i;i>y;i--){
result[i][j] = count++;
}
x+=1;
y+=1;
offset++;
//循环变量更新
circle++;
}
//判断中间位置是否需要单独赋值
if(n%2==1){
result[n/2][n/2] = n*n;
}
return result;
}
总结
1、双指针移动方向根据题中数组变化趋势设置
2、快慢指针外层循环使用快指针作为终止条件,否则和暴力解法没有区别
3、注意确定边界条件为左闭右开/左闭右闭,同时每圈的起始位置也很重要