数组part02
977 有序数组的平方
题目
思路
因为数组是有序的,且其中可能有负数,所以求平方最大的元素一定是开头的元素或者结尾的元素,第二大的元素是除去最大的元素之后的开头元素和结尾元素之间,第三大的元素…。综上,使用双指针left和right,left指向开头元素,right指向结尾元素,将其中求平方较大的放入结果数组的后方(从后向前开始放),比如left下标的元素求平方比较大,则把nums[left]放到结果数组的后方,然后left++,继续和nums[right]比较,直到left>right。
代码
vector<int> sortedSquares(vector<int>& nums) {
vector<int> result(nums.size(), 0);
int left=0, right=nums.size()-1;
int index=right;
while(left<=right){ //注,这里一定要是left<=right,而不是left<right,否则最后一个元素会被遗漏
int leftNum=pow(nums[left], 2);
int rightNum=pow(nums[right], 2);
if(leftNum>rightNum){
result[index--]=leftNum;
left++;
}else{
result[index--]=rightNum;
right--;
}
}
return result;
}
209 长度最小的子数组
题目
思路
这是一道典型的滑动窗口题目,题目中要求的是最小窗口,决定窗口大小的是窗口中元素的总和和target的关系。设窗口起始和结尾的下标分别为i和j,i和j都从0开始,让j开始移动,移动时计算目前窗口内元素的总和sum。当发现sum刚大于等于target时,立刻让j停止移动,而此时窗口不一定调整到了最小,可能去掉窗口前面几个元素仍然满足sum>=target,所以此时需要调整i的值,直到sum再次刚好大于等于target,此时窗口是j移动到该位置所能得到的最小窗口,和目前最小窗口比较,若比目前最小窗口还小,则将目前最小窗口赋值为该窗口。接下来继续移动j,按照同样的方案再次求得j所在位置能得到的最小窗口,直到j移动到末尾。
代码
int minSubArrayLen(int target, vector<int>& nums) {
int i=0;
int sum=0;
int minLen=INT_MAX;
for(int j=0; j<nums.size(); j++){
sum+=nums[j];
if(sum>=target){
while(sum>=target){
sum-=nums[i++];
}
//上面while语句导致i多加了1,sum多减了nums[i],需要恢复一下
i--;
sum+=nums[i]; //多减了
if(j-i+1<minLen){
minLen=j-i+1;
}
}
}
if(minLen==INT_MAX) return 0;
return minLen;
}
59 螺旋矩阵
题目
思路
看了代码随想录的解析,利用了这个矩阵的长和宽都相等,但后面我也看到了长和宽不相等的螺旋矩阵题目,所以我觉得代码随想录提供的解法不具有普适性,我自己做题时想到了一个解法,如下:
螺旋矩阵的特点是路径一直在旋转,依次是:上、右、下、左。什么时候判断是否该转弯呢?那就需要我们在上、右、下、左侧各设置一个挡板up、right、down、left,遇到挡板就转到下一个方向继续遍历,此时碰撞到的挡板也朝里移动一格。遍历到什么时候结束呢?那就看我们具体怎么设置挡板,比如说我设置初始挡板为:up=0, right=n-1, down=n-1, left=0。那么终止条件就是up>down||left>right。
代码
vector<vector<int>> generateMatrix(int n) {
int up=0, right=n-1, down=n-1, left=0;
int i, j;
int count=1;
vector<vector<int>> result(n, vector<int>(n, 0));
while(up<=down&&left<=right){
//上
i=up;
j=left;
while(j<=right){
result[i][j]=count++;
j++;
}
j--;
up++;
//右
while(i<down){
i++;
result[i][j]=count++;
}
right--;
//下
while(j>left){
j--;
result[i][j]=count++;
}
down--;
//左
while(i>up){
i--;
result[i][j]=count++;
}
left++;
}
return result;
}