算法
来源:力扣(LeetCode)315. 计算右侧小于当前元素的个数
给定一个整数数组 nums,按要求返回一个新数组 counts。数组 counts 有该性质: counts[i] 的值是 nums[i] 右侧小于 nums[i] 的元素的数量。
示例:
输入:nums = [5,2,6,1]
输出:[2,1,1,0]
解释:
5 的右侧有 2 个更小的元素 (2 和 1)
2 的右侧仅有 1 个更小的元素 (1)
6 的右侧有 1 个更小的元素 (1)
1 的右侧有 0 个更小的元素
提示:
0 <= nums.length <= 10^5
-10^4 <= nums[i] <= 10^4
解题思路
这题需要现在逆序的基础上讲解,假设现在是要我们统计逆序对有多少个,[5,2,6,1]得到的逆序对一共有4对,分别是[5,2],[5,1],[2,1],[6,1] 在5这个上贡献了两个,2贡献一个,6贡献一个。现在假设我们有两个已排序的序列等待合并,分别是 L = { 8, 12 } 和 R={9,26,55},我们让指针pl指向8,pr指向9,比较大小,小的进即8入集合M={8},由于pr没有移动指的还是头这个位置,所以8当前合并没有贡献逆序对,然后pl右移到一位到12与pr比较大小,这时9进入M={8,9},pr右移一位到26,pl与pr比较大小,12进入集合M{8,9,12} 由于pr向右移动一位,所以12当前合并贡献一对逆序对,然后把剩下的R集合依次放入,它们当前合并贡献的贡献都是0。
解决记录数字的贡献值后,可是排完序的数组和之前的数组顺序变了,所以我们需要记录数组原先的位置。利用pair构成一对一。
代码
class Solution {
public:
vector<int> countSmaller(vector<int>& nums) {
vector<pair<int,int>>vec;
vector<int> count;
for(int i=0;i<nums.size();i++){
vec.push_back(make_pair(nums[i],i));
count.push_back(0);
}
merge_sort(vec,count);
return count;
}
private:
void merge_sort_two_vec(vector<pair<int,int>> &vec1,
vector<pair<int,int>> &vec2,vector<pair<int,int>> &vec,
vector<int> &count){
int i=0;
int j=0;
while(i<vec1.size()&&j<vec2.size()){
if(vec1[i].first<=vec2[j].first){
count[vec1[i].second]+=j;
vec.push_back(vec1[i]);
i++;
}else{
vec.push_back(vec2[j]);
j++;
}
}
for(;i<vec1.size();i++){
count[vec1[i].second]+=j;
vec.push_back(vec1[i]);
}
for(;j<vec2.size();j++){
vec.push_back(vec2[j]);
}
}
void merge_sort(vector<pair<int,int>> &vec,vector<int> &count){
if(vec.size()<2){
return;
}
int mid=vec.size()/2;
vector<pair<int,int>> vec1;
vector<pair<int,int>> vec2;
for(int i=0;i<mid;i++){
vec1.push_back(vec[i]);
}
for(int i=mid;i<vec.size();i++){
vec2.push_back(vec[i]);
}
merge_sort(vec1,count);
merge_sort(vec2,count);
vec.clear();
merge_sort_two_vec(vec1,vec2,vec,count);
}
};
来源:力扣(LeetCode)113. 路径总和 II
给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。
说明: 叶子节点是指没有子节点的节点。
示例:
给定如下二叉树,以及目标和 sum = 22,
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
返回:
[
[5,4,11,2],
[5,8,4,5]
]
解题思路
由于题目要求一定要根节点到叶子节点,同时也不确定是否有负数,所以我觉得无法剪枝,也就是嘚老老实实深度遍历一遍二叉树,遍历二叉树并不难,同时拿多一个容器来存储节点,最后走完这一条路径,记得一定要依次把节点pop移出去即可。
代码
/**
* 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 *node,vector<vector<int>>&result,vector<int> &path,int targetSum){
if(!node){
return;
}
path.push_back(node->val);
if(accumulate(path.begin(),path.end(),0)==targetSum&&!node->left&&!node->right){
result.push_back(path);
}
traversal(node->left,result,path,targetSum);
traversal(node->right,result,path,targetSum);
path.pop_back();
}
vector<vector<int>> pathSum(TreeNode* root, int targetSum) {
vector<vector<int>>result;
vector<int> path;
traversal(root,result,path,targetSum);
return result;
}
};