题目描述
给你一个下标从 1 开始、长度为 n 的整数数组 nums 。
现定义函数 greaterCount ,使得 greaterCount(arr, val) 返回数组 arr 中 严格大于 val 的元素数量。
你需要使用 n 次操作,将 nums 的所有元素分配到两个数组 arr1 和 arr2 中。在第一次操作中,将 nums[1] 追加到 arr1 。在第二次操作中,将 nums[2] 追加到 arr2 。之后,在第 i 次操作中:
如果 greaterCount(arr1, nums[i]) > greaterCount(arr2, nums[i]) ,将 nums[i] 追加到 arr1 。
如果 greaterCount(arr1, nums[i]) < greaterCount(arr2, nums[i]) ,将 nums[i] 追加到 arr2 。
如果 greaterCount(arr1, nums[i]) == greaterCount(arr2, nums[i]) ,将 nums[i] 追加到元素数量较少的数组中。
如果仍然相等,那么将 nums[i] 追加到 arr1 。
连接数组 arr1 和 arr2 形成数组 result 。例如,如果 arr1 = [1,2,3] 且 arr2 = [4,5,6] ,那么 result = [1,2,3,4,5,6] 。
返回整数数组 result 。
示例
示例1:
输入:nums = [2,1,3,3]
输出:[2,3,1,3]
解释:在前两次操作后,arr1 = [2] ,arr2 = [1] 。
在第 3 次操作中,两个数组中大于 3 的元素数量都是零,并且长度相等,因此,将 nums[3] 追加到 arr1 。
在第 4次操作中,两个数组中大于 3 的元素数量都是零,但 arr2 的长度较小,因此,将 nums[4] 追加到 arr2 。
在 4次操作后,arr1 = [2,3] ,arr2 = [1,3] 。
因此,连接形成的数组 result 是 [2,3,1,3] 。
示例2:
输入:nums = [5,14,3,1,2]
输出:[5,3,1,2,14]
解释:在前两次操作后,arr1 = [5] ,arr2 = [14] 。
在第 3 次操作中,两个数组中大于 3 的元素数量都是一,并且长度相等,因此,将 nums[3] 追加到 arr1 。
在第 4 次操作中,arr1 中大于 1 的元素数量大于 arr2 中的数量(2 > 1),因此,将 nums[4] 追加到 arr1 。
在第 5 次操作中,arr1 中大于 2 的元素数量大于 arr2 中的数量(2 > 1),因此,将 nums[5] 追加到 arr1 。
在 5 次操作后,arr1 = [5,3,1,2] ,arr2 = [14] 。
因此,连接形成的数组 result 是 [5,3,1,2,14] 。
示例3:
输入:nums = [3,3,3,3]
输出:[3,3,3,3]
解释:在 4 次操作后,arr1 = [3,3] ,arr2 = [3,3] 。
因此,连接形成的数组 result 是 [3,3,3,3] 。
思路
数组元素数量比较,查找时使用二分查找,否则会超时!
代码
public static int[] resultArray(int[] nums) {
List<Integer> arr1 = new ArrayList<>();
List<Integer> arr11 = new ArrayList<>();
List<Integer> arr2 = new ArrayList<>();
List<Integer> arr22 = new ArrayList<>();
for (int e : nums) {
if (arr1.isEmpty()) {
arr1.add(e);
arr11.add(e);
continue;
}
if (arr2.isEmpty()) {
arr2.add(e);
arr22.add(e);
continue;
}
int idx1 = binarySearch(arr1, e + 1);
int idx2 = binarySearch(arr2, e + 1);
int cnt1 = arr1.size() - idx1;
int cnt2 = arr2.size() - idx2;
if (cnt1 > cnt2) {
arr1.add(idx1, e);
arr11.add(e);
continue;
}
if (cnt1 < cnt2) {
arr2.add(idx2, e);
arr22.add(e);
continue;
}
if (cnt1 == cnt2) {
if (arr1.size() <= arr2.size()) {
arr1.add(idx1, e);
arr11.add(e);
} else {
arr2.add(idx2, e);
arr22.add(e);
}
}
}
List<Integer> result = new ArrayList<>(arr11);
result.addAll(arr22);
// 将结果列表转换为数组
int[] resultArray = new int[result.size()];
for (int i = 0; i < result.size(); i++) {
resultArray[i] = result.get(i);
}
return resultArray;
}
public static int binarySearch(List<Integer> arr, int target) {
int low = 0, high = arr.size();
while (low < high) {
int mid = (low + high) / 2;
if (arr.get(mid) < target) {
low = mid + 1;
} else {
high = mid;
}
}
return low;
}