【2023】字节跳动 10 日心动计划——第三关
目录
- 1. 最长有效括号
- 2. 有序数组的平方
1. 最长有效括号
🔗 原题链接:32. 最长有效括号
类似于有效的括号,考虑用栈来解决。
具体来讲,我们始终保持栈底元素为当前已经遍历过的元素中「最后一个没有被匹配的右括号的下标」,这样的做法主要是考虑了边界条件的处理,栈里其他元素维护左括号的下标。
从左往右遍历整个字符串,如果遇到 (
,则将其下标压入栈中;如果遇到 )
,则先弹出栈顶元素,然后再判断栈是否为空,如果栈为空,说明当前的右括号为没有被匹配的右括号,将其压入栈中,否则,更新答案。
注意,任何时刻,只有栈底元素是右括号的下标,其他元素都是左括号的下标!
class Solution {
public:int longestValidParentheses(string s) {stack<int> stk;int ans = 0;stk.push(-1);for (int i = 0; i < s.size(); i++) {if (s[i] == '(') stk.push(i);else {stk.pop();if (stk.empty()) stk.push(i);else ans = max(ans, i - stk.top());}}return ans;}
};
2. 有序数组的平方
🔗 原题链接:977. 有序数组的平方
这里介绍两种做法。
方法一:找到正负元素的分界线,然后对正、负数组进行二路归并。
class Solution {
public:vector<int> sortedSquares(vector<int>& nums) {int p = lower_bound(nums.begin(), nums.end(), 0) - nums.begin();int i = p, j = p - 1;vector<int> res;while (i < nums.size() && j >= 0) {int a = pow(nums[i], 2), b = pow(nums[j], 2);if (a <= b) res.push_back(a), i++;else res.push_back(b), j--;}while (i < nums.size()) {int a = pow(nums[i], 2);res.push_back(a);i++;}while (j >= 0) {int b = pow(nums[j], 2);res.push_back(b);j--;}return res;}
};
方法二:同样使用双指针。之前我们是让两个指针从中间往两边移动,这次我们让两个指针从两边往中间移动,所以填答案的时候需要倒着填。
class Solution {
public:vector<int> sortedSquares(vector<int>& nums) {int n = nums.size();vector<int> res(n);int i = 0, j = n - 1, k = n - 1;while (i <= j) {int a = nums[i] * nums[i];int b = nums[j] * nums[j];if (a >= b) res[k] = a, i++;else res[k] = b, j--;k--;}return res;}
};