315. Count of Smaller Numbers After Self

数组中每个元素所在位置右边比它小的元素

int[] nums = { 5, 2, 6, 1 };
var rslt = CountSmaller(nums);
foreach (var item in rslt)
{
	Console.Write(item + "  ");
}

private int[] CountSmaller(int[] nums)
{
	if (nums == null || nums.Length == 0)
	{
		return new int[] { };
	}
	var tree = new SegmentCountTree();
	var numsMap = new Dictionary<int, int>();
	var numsArray = new List<int>();
	var rslt = new int[nums.Length];
	for (int i = 0; i < nums.Length; i++)
	{
		numsMap[nums[i]] = nums[i];
	}
	foreach (var item in numsMap)
	{
		numsArray.Add(item.Value);
	}
	// 若不排序,线段树中的区间会有不合法的
	numsArray.Sort();
	var minNum = numsArray[0];
	tree.Init(numsArray.ToArray(), OperZero);
	for (int i = nums.Length - 1; i >= 0; i--)
	{
		if (nums[i] == minNum)  // 未找到
		{
			rslt[i] = 0;
			tree.UpdateCount(nums[i]);
			continue;
		}
		tree.UpdateCount(nums[i]);  // 
		rslt[i] = tree.Query(minNum, nums[i] - 1);
	}
	return rslt;
}

private int OperZero(int arg1, int arg2)
{
	return 0;
}
posted @   wesson2019  阅读(25)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 10年+ .NET Coder 心语 ── 封装的思维:从隐藏、稳定开始理解其本质意义
· 地球OL攻略 —— 某应届生求职总结
· 提示词工程——AI应用必不可少的技术
· Open-Sora 2.0 重磅开源!
· 周边上新:园子的第一款马克杯温暖上架
点击右上角即可分享
微信分享提示