数组part2 主要是为了理解数组中双指针的操作和实现方法。
一、977 有序数组的平方
题目
问题要点:使用暴力解法,将数组中的所有元素平方后,使用C++ Vector自带的sort()排序,复杂度为 O(nlogn)。而本题目的最优策略是使用双指针,时间复杂度O(n)。由于该数组已经是预先排序好的数组,因此对于平方值,最大的值只会出现在原数组的最左(绝对值最大的负数)和最右(最大的正数),因此用start 和 end 两个指针分别指向头尾,依次比较他们的最大值放入新的数组中即可。
class Solution {
public:
int power(int a) {
return a * a;
}
vector<int> sortedSquares(vector<int>& nums) {
int start = 0;
int end = nums.size() - 1;
vector<int> ans;
// while循环的边界条件很重要
// start == end时,该元素也要放进数组
// 当 start = end + 1时,说明已经不合法了
while(start < end + 1) {
if(power(nums[start]) > power(nums[end])) {
ans.insert(ans.begin(), power(nums[start++]));
} else {
ans.insert(ans.begin(), power(nums[end--]));
}
}
return ans;
}
};
二、 209 长度最小的子数组
题目
思路:本题采用滑动窗口思想,但实现的方式其实也是通过双指针。需要注意的是为了保证时间复杂度为On,外圈的判断条件 因该是 滑动窗口的尾节点End,在每次循环内处理时,处理挪动滑动窗口的头节点Start
class Solution {
public:
int minSubArrayLen(int target, vector<int>& nums) {
int start = 0;
int end = 0;
int sum = 0;
int ans = INT32_MAX;
for(; end < nums.size(); end++) {
sum += nums[end];
while(sum >= target) {
ans = min(ans, (end - start + 1));
sum -= nums[start++];
}
//printf("%d %d %d %d\n", start, end, sum, ans);
}
return ans == INT32_MAX ? 0:ans;
}
};
滑动窗口扩展题目1:904. 水果成篮
题目
思路:本题也是采用了滑动窗口的思想,不过根据题目中的判断条件,需要用Hashmap来存键值对,代表两个篮子中装有的水果数量。
注意:熟悉C++ 的unordered_map的相关操作和实现细节。
class Solution {
public:
int totalFruit(vector<int>& fruits) {
// 使用滑动窗口思想
int start = 0;
int ans = 0;
// 用哈希表存 代表两个水果篮子
// key: 该种水果类型
// value: 该种水果出现的次数
unordered_map <int, int> basket;
// tips:
// 要熟悉下哈希表中对元素的操作如何编写
// 迭代器iterator的second为该key的value
for(int end = 0; end < fruits.size(); end++) {
basket[fruits[end]]++;
while(basket.size() > 2) {
auto it = basket.find(fruits[start]);
it->second--;
if(it->second == 0) {
basket.erase(it);
}
start++;
}
ans = max(ans, end - start + 1);
}
return ans;
}
};
三、59 螺旋矩阵 2
题目
思路:主要就是用代码实现题目中的操作逻辑,将四个位移方向定义为4个 {x, y} 的数组即可轻松实现。
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
vector<vector<int>> m(n, vector<int>(n, 0));
int length = n*n;
int x = 0, y = 0;
int startDir = 0;
int nextX, nextY;
// 定义四个方向,代表 右,下,左,上
vector<vector<int>> dirs = {{0,1}, {1, 0}, {0, -1}, {-1, 0}};
for(int i = 1; i <= length; i++) {
m[x][y] = i;
nextX = x + dirs[startDir][0];
nextY = y + dirs[startDir][1];
if(nextX >= n || nextX < 0 || nextY >= n || nextY < 0 || m[nextX][nextY] != 0) {
startDir = (startDir + 1) % 4;
}
x += dirs[startDir][0];
y += dirs[startDir][1];
}
return m;
}
};
总结:数组part2的前两道题,主要为了让我熟悉双指针的方法。双指针的时间复杂度一般都为O(n),注意滑动窗口的循环条件是要以窗口的尾节点为准。