leetcode Count of Smaller Numbers After Self

题目连接

https://leetcode.com/problems/count-of-smaller-numbers-after-self/

Count of Smaller Numbers After Self

Description

You are given an integer array nums and you have to return a new counts array. The counts array has the property where $counts[i]$ is the number of smaller elements to the right of $nums[i]$.

Example:

Given $nums = [5,2,6,1]$
To the right of 5 there are 2 smaller elements (2 and 1).
To the right of 2 there is only 1 smaller element (1).
To the right of 6 there is 1 smaller element (1).
To the right of 1 there is 0 smaller element.
Return the array $[2, 1, 1, 0]$.

线段树求逆序数,先离散化,再用线段树统计。。

class Solution {
private:
	typedef vector<int> vec;
public:
	vec countSmaller(vec& nums) {
		vec ans;
		int n = 0;
		if (!(n = nums.size())) return ans;
		ans.resize(n, 0);
		ret = new P[n + 10];
		for (int i = 1; i <= n; i++) ret[i] = P(nums[i - 1], i);
		sort(ret + 1, ret + n + 1);
		seg = new int[(n + 10) << 2];
		memset(seg, 0, sizeof(int)* ((n + 10) << 2));
		for (int i = 1; i <= n; i++) {
			int v = query(1, 1, n, ret[i].id, n);
			insert(1, 1, n, ret[i].id);
			ans[ret[i].id - 1] = v;
		}
		delete[]seg; delete[]ret;
		return ans;
	}
private:
	struct P {
		int v, id;
		P(int _i_ = 0, int _j_ = 0) :v(_i_), id(_j_) {}
		friend bool operator<(const P &a, const P &b) {
			return a.v == b.v ? a.id < b.id : a.v < b.v;
		}
	}*ret;
	int *seg;
	inline void insert(int root, int l, int r, int p) {
		if (p > r || p < l) return;
		if (p <= l && p >= r) { seg[root]++; return; }
		int mid = (l + r) >> 1;
		insert(root << 1, l, mid, p);
		insert(root << 1 | 1, mid + 1, r, p);
		seg[root] = seg[root << 1] + seg[root << 1 | 1];
	}
	inline int query(int root, int l, int r, int x, int y) {
		if (x > r || y < l) return 0;
		if (x <= l && y >= r) return seg[root];
		int ret = 0;
		int mid = (l + r) >> 1;
		ret += query(root<<1, l, mid, x, y);
		ret += query(root << 1 | 1, mid + 1, r, x, y);
		return ret;
	}
};
时间: 2024-10-13 16:21:54

leetcode Count of Smaller Numbers After Self的相关文章

[LeetCode] Count of Smaller Numbers After Self 计算后面较小数字的个数

You are given an integer array nums and you have to return a new counts array. The counts array has the property where counts[i] is the number of smaller elements to the right of nums[i]. Example: Given nums = [5, 2, 6, 1] To the right of 5 there are

[LeetCode][JavaScript]Count of Smaller Numbers After Self

Count of Smaller Numbers After Self You are given an integer array nums and you have to return a new counts array. The counts array has the property where counts[i] is the number of smaller elements to the right of nums[i]. Example: Given nums = [5,

剑指Offer 面试题36:数组中的逆序对及其变形(Leetcode 315. Count of Smaller Numbers After Self)题解

剑指Offer 面试题36:数组中的逆序对 题目:在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对.输入一个数组,求出这个数组中的逆序对的总数. 例如, 在数组{7,5,6,4}中,一共存在5个逆序对,分别是(7,6),(7,5),(7,4),(6,4)和(5,4),输出5. 提交网址: http://www.nowcoder.com/practice/96bd6684e04a44eb80e6a68efc0ec6c5?tpId=13&tqId=11188 或 htt

Count of Smaller Numbers After Self -- LeetCode

You are given an integer array nums and you have to return a new counts array. The counts array has the property where counts[i] is the number of smaller elements to the right of nums[i]. Example: Given nums = [5, 2, 6, 1] To the right of 5 there are

【Leetcode】Count of Smaller Numbers After Self

题目链接:https://leetcode.com/problems/count-of-smaller-numbers-after-self/ 题目: You are given an integer array nums and you have to return a new counts array. The counts array has the property where counts[i] is the number of smaller elements to the righ

leetcode 315. Count of Smaller Numbers After Self

You are given an integer array nums and you have to return a new counts array. The counts array has the property where counts[i] is the number of smaller elements to the right of nums[i]. Example: Given nums = [5, 2, 6, 1] To the right of 5 there are

第十四周 Leetcode 315. Count of Smaller Numbers After Self(HARD) 主席树

Leetcode315 题意很简单,给定一个序列,求每一个数的右边有多少小于它的树. O(n^2)的算法是显而易见的. 用普通的线段树可以优化到O(nlogn) 我们可以直接套用主席树的模板. 主席树的功能是什么呢? 其实就是一句话. 原序列a的子序列a[l,r]在a排序后的序列b的子序列[L,R]中的个数. 显然本题只用到了主席树的一小部分功能. const int N = 100000 + 5; int a[N], b[N], rt[N * 20], ls[N * 20], rs[N * 2

[email&#160;protected] [315/215] Count of Smaller Numbers After Self / Kth Largest Element in an Array (BST)

https://leetcode.com/problems/count-of-smaller-numbers-after-self/ You are given an integer array nums and you have to return a new counts array. The counts array has the property where counts[i] is the number of smaller elements to the right of nums

Count of Smaller Numbers After Self

题目: You are given an integer array nums and you have to return a new counts array. The counts array has the property where counts[i] is the number of smaller elements to the right of nums[i]. Example: Given nums = [5, 2, 6, 1] To the right of 5 there