0基础力扣 DAY2
今天依然是数组。
977. 有序数组的平方
题目链接:977. 有序数组的平方 - 力扣(LeetCode)
思路:由于题目要求时间复杂度控制在o(n),而排序暴力解需要o(nlogn),不够优秀。找到正负分界处,对数组进行平方,从分界处,双指针分别向前向后遍历,采用类归并的思想对数组进行排序即可。
代码如下:
class Solution {
public:
vector<int> sortedSquares(vector<int>& nums) {
int mid = 0;
int tmp = 0;
int r = 0; //归并指针
int a[nums.size()];
if(nums.size()==1){
nums[0]=nums[0]*nums[0];
return nums;
}
while(mid<nums.size()&&nums[mid]<0) //找到正负分界线
mid++;
if(mid == 0){
for(int l=0;l<nums.size();l++)
nums[l] = nums[l] *nums[l];
return nums;
}
if(mid == nums.size()){
for(int i = 0; i<mid/2;i++){
tmp = nums[i];
nums[i] = nums[mid-1-i];
nums[mid-1-i] = tmp;
}
for(int j=0;j<mid;j++){
nums[j] = abs(nums[j]);
}
for(int n=0;n<nums.size();n++)
nums[n] = nums[n] * nums[n];
return nums;
}
int p = 0;
int q = mid;
for(int i = 0; i<(mid+1)/2;i++){
tmp = nums[i];
nums[i] = nums[mid-1-i];
nums[mid-1-i] = tmp;
}
for(int j=0;j<mid;j++){
nums[j] = abs(nums[j]);
}
for(int k=0;k<nums.size();k++)
a[k] = nums[k];
while(p < mid && q < nums.size()){ //归并
if(a[p]<=a[q]){
nums[r++] = a[p]*a[p];
p++;
}
else{
nums[r++] = a[q]*a[q];
q++;
}
}
while(p<mid){
nums[r++] = a[p]*a[p];
p++;
}
while(q<nums.size()){
nums[r++] = a[q]*a[q];
q++;
}
return nums;
}
};
优化:将双指针从头尾向中间遍历,按照由大到小插入会更为简洁。
参考文章:代码随想录 (programmercarl.com)
209. 长度最小的子数组
题目链接:209. 长度最小的子数组 - 力扣(LeetCode)
思路:同样是双指针遍历数组,子数组和小于target时,移动尾指针;大于target时,移动首指针,同步记录子数组最小长度,遍历完成时返回最小长度即可。
代码如下:
class Solution {
public:
int minSubArrayLen(int target, vector<int>& nums) {
int j=0;
int min = INT_MAX;
int sum = 0;
int count = 0;
for(int i=0;i<nums.size();i++){
sum+=nums[i];
count++;
while(sum>=target){
if(count<min){
min = count;
}
sum-=nums[j++];
count--;
}
}
if(min<=nums.size())
return min;
else
return 0;
}
};
59. 螺旋矩阵Ⅱ
题目链接:59. 螺旋矩阵 II - 力扣(LeetCode)
思路:将一圈作为一轮循环,圈内分成四段进行填充,由外向内逐步进行。主要需要注意每一轮起始位置坐标,以及圈内四段需要全部左闭右开或左开又闭,防止思维混乱。
代码如下:
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
int i=0;
int j=0;
int startx=0;
int starty=0;
int count=1;
vector<vector<int>> a(n,vector<int>(n));
if(n==1){
a[0][0] = 1;
return a;
}
for(int offset=1;offset<=n/2;offset++){
for(;j<n-offset;j++){
a[i][j] = count++;
}
for(;i<n-offset;i++){
a[i][j] = count++;
}
for(;j>offset-1;j--){
a[i][j] = count++;
}
for(;i>offset-1;i--){
a[i][j] = count++;
}
startx++;
starty++;
i=startx;
j=starty;
}
if(n%2==1){
a[n/2][n/2] = count;
}
return a;
}
};
总结
今天内容主要是双指针遍历数组的思想,以及模拟(较为考察代码能力,对边界情况的处理)。继续努力