2765. 最长交替子数组
给你一个下标从 0 开始的整数数组 nums
。如果 nums
中长度为 m
的子数组 s
满足以下条件,我们称它是一个 交替子数组 :
m
大于 1
。s1 = s0 + 1
。s
与数组 [s0, s1, s0, s1,...,s(m-1) % 2]
一样。也就是说,s1 - s0 = 1
,s2 - s1 = -1
,s3 - s2 = 1
,s4 - s3 = -1
,以此类推,直到 s[m - 1] - s[m - 2] = (-1)m
。请你返回 nums
中所有 交替 子数组中,最长的长度,如果不存在交替子数组,请你返回 -1
。
子数组是一个数组中一段连续 非空 的元素序列。
示例 1:
输入:nums = [2,3,4,3,4] 输出:4 解释:交替子数组有 [3,4] ,[3,4,3] 和 [3,4,3,4] 。最长的子数组为 [3,4,3,4] ,长度为4 。
示例 2:
输入:nums = [4,5,6] 输出:2 解释:[4,5] 和 [5,6] 是仅有的两个交替子数组。它们长度都为 2 。
提示:
2 <= nums.length <= 100
1 <= nums[i] <= 104
上来很容易想到O(n^2)的做法,就是两层for循环,其实这个就是用nums[j] - nums[i] == (length - 1) % 2来算的,如果相邻就是大1,如果不相邻就是相等,用这个递推公式来判断,但是这种会重复计算,会重复判断,有没有更好的方法呢?
class Solution {
public:
int alternatingSubarray(vector& nums) {
int length = 0, res = -1;
for(int i = 0;i < nums.size();i++){
for(int j = i + 1;j < nums.size();j++){
length = j - i + 1;
if(nums[j] - nums[i] == (length - 1) % 2){
res = max(res, length);
} else break;
}
}
return res;
}
};
下面是O(n)的做法,为什么是O(n),其实第一层循环是找到符合i + 1个数 - i个数满足1的,也就是找到满足的开头,第二层for循环相当于针对找到的开头来去寻找到最长的子串,那么他们并没有一种嵌套的形式,而是从前往后递进的,因此也就是O(n).
class Solution {
public:
int alternatingSubarray(vector& nums) {
int n = nums.size(), start = 0, res = -1, i = 0;
while(i < n - 1){
if(nums[i + 1] - nums[i] != 1){
i++;
continue;
}
start = i;
i += 2;
while(i < n && nums[i] - nums[start] == (i - start) % 2 ){
i++;
}
res = max(res, i - start);
i--;
}
return res;
}
};
接下来更新一下二叉树的遍历,首先是前中后序遍历,这里的话只挑中序遍历来写,两种方法,递归和迭代。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
void Traversal(TreeNode* cur,vector &vec){
//中序遍历
if(cur == NULL) return;
Traversal(cur->left,vec);
vec.push_back(cur->val);
Traversal(cur->right,vec);
}
vector inorderTraversal(TreeNode* root) {
vector result;
Traversal(root,result);
return result;
}
};
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
vector inorderTraversal(TreeNode* root) {
stack st;
TreeNode* cur = root;
vector res;
while(cur != NULL || !st.empty()){
if(cur != NULL){
st.push(cur);
cur = cur->left;
} else{
cur = st.top();
st.pop();
res.push_back(cur->val);
cur = cur->right;
}
}
return res;
}
};
二叉树的层序遍历
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
vector> levelOrder(TreeNode* root) {
queue que;
if(root) que.push(root);
vector> result;
while(!que.empty()){
int size = que.size();
vector vec;
for(int i = 0;i < size;i++){
TreeNode* cur = que.front();
que.pop();
vec.push_back(cur->val);
if(cur->left) que.push(cur->left);
if(cur->right) que.push(cur->right);
}
result.push_back(vec);
}
return result;
}
};
226. 翻转二叉树
给你一棵二叉树的根节点 root
,翻转这棵二叉树,并返回其根节点。
示例 1:
输入:root = [4,2,7,1,3,6,9] 输出:[4,7,2,9,6,3,1]
示例 2:
输入:root = [2,1,3] 输出:[2,3,1]
示例 3:
输入:root = [] 输出:[]
提示:
[0, 100]
内-100 <= Node.val <= 100
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
TreeNode* invertTree(TreeNode* root) {
if(root == NULL) return root;
queue que;
que.push(root);
while(!que.empty()){
int size_ = que.size();
for(int i = 0;i < size_;i++){
TreeNode* cur = que.front();
que.pop();
swap(cur->left,cur->right);
if(cur->left) que.push(cur->left);
if(cur->right) que.push(cur->right);
}
}
return root;
}
};