代码随想录算法训练营Day2 | 977.有序数组的平方 、209.长度最小的子数组、59.螺旋矩阵II
977.有序数组的平方 题目链接
方法:双指针
由于0的存在,使得数组的平方不有序,但最大的在两端,最小的在中间,因此借鉴二路归并排序的思想,采用双指针法,对数组的平方进行排序。
class Solution {
public:
vector<int> sortedSquares(vector<int>& nums) {
int n = nums.size();
vector<int> ret(n);
int i = 0;
int j = n-1;
int cur = n-1;
for(;i<=j;){
if(nums[i]*nums[i]<nums[j]*nums[j]){
ret[cur] = nums[j]*nums[j];
j--;
}
else{
ret[cur] = nums[i]*nums[i];
i++;
}
cur--;
}
return ret;
}
};
209.长度最小的子数组 题目链接
本题需要取区间,暴力解法需要两个for循环来实现,时间复杂度为O(n^2)
考虑本题采用滑动窗口法,本质是根据当前子序列的大小和长度再调整序列的长度,核心在于窗口的起止及其移动。
class Solution {
public:
int minSubArrayLen(int s, vector<int>& nums) {
// 滑动窗口
int result = INT32_MAX;
int sum = 0;
int i = 0;
int subLength = 0;
int n = nums.size();
for(int j = 0; j < n; j++){ // 遍历的是滑动窗口的终点
sum+=nums[j];
while(sum>=s){
subLength = j-i+1;
result = result<subLength ? result:subLength;
sum-=nums[i++]; // 滑动窗口的核心,根据序列情况移动窗口起点
}
}
// 如果result没有被赋值的话,就返回0,说明没有符合条件的子序列
return result == INT32_MAX ? 0 : result;
}
};
时间复杂度为O(n),每个元素都被操作2次,入窗口和出窗口各一次,因此时间复杂度为2×n=O(n)
59.螺旋矩阵II 题目链接
本题不涉及具体的算法,是一个模拟过程
在涉及到循环的题目时,需要注意循环不变量,按照左闭右开的规则,每条边从第一个节点开始处理到倒数第二个节点,不处理最后一个节点,而是留作下一次处理的起点,使得4条边的遍历规则统一。
以n=6为例,共转了6/2=3圈,对应每条边的边长为6、4、2。
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
vector<vector<int>> res(n, vector<int>(n, 0));
int startx = 0, starty = 0;
int loop = n / 2;
int mid = n / 2;
int count = 1;
int offset = 1;
int i,j;
while (loop --) {
i = startx;
j = starty;
for (j = starty; j < n - offset; j++) {
res[startx][j] = count++;
}
for (i = startx; i < n - offset; i++) {
res[i][j] = count++;
}
for (; j > starty; j--) {
res[i][j] = count++;
}
for (; i > startx; i--) {
res[i][j] = count++;
}
startx++;
starty++;
offset += 1;
}
if (n % 2) {
res[mid][mid] = count;
}
return res;
}
};