小技巧 ----- 计数数组中小于 target 的元素个数

二分

int[] arr = new int[N];
int cnt = 0;

void run() {
	Arrays.sort(arr, 0, n);
	
	if (arr[n - 1] < target) cnt = n;
	else if (arr[0] >= target) cnt = 0;
	else {
		int l = 0, r = arr.length - 1;
		while (l < r) {
			int mid = (l + r) / 2;
			if (arr[mid] <= target) {
				l = mid + 1;
			} else {
				r = mid;
			}
		}
		cnt = l;
	}
}

flag + 前缀和

int[] flag = new int[N];
int[] s = new int[N];
int[] arr = new int[N];
int cnt = 0;

void run() {
	// 初始化flag
	for (int i = 0; i < n; i++) {
		flag[arr[i]]++;
	}
	
	// 对flag数组求前缀和:flag[i] 表示 arr数组中所有取值在 [0, i] 的元素的个数
	s[0] = flag[0];
	for (int i = 1; i < n; i++) {
		s[i] = s[i - 1] + flag[i];
	}
	
	// 使用flag数组得出arr数组中小于target的数的数量
	cnt = target - 1 == 0 ? 0 : flag[target - 1];
}
©️2020 CSDN 皮肤主题: 游动-白 设计师:上身试试 返回首页