根据给定数字划分数组

/*
 * 题目思路:
 * 1. 使用Java Stream来处理数组，将元素分为小于、等于和大于pivot的三个部分。
 * 2. 将这三个部分合并成一个最终结果。
 */

import java.util.Arrays;
import java.util.stream.Collectors;

public class Solution {
    public int[] pivotArray(int[] nums, int pivot) {
        // 使用stream过滤和收集不同条件的元素
        int[] less = Arrays.stream(nums).filter(n -> n < pivot).toArray();
        int[] equal = Arrays.stream(nums).filter(n -> n == pivot).toArray();
        int[] greater = Arrays.stream(nums).filter(n -> n > pivot).toArray();

        // 合并结果
        return Arrays.stream(new int[][]{less, equal, greater})
                .flatMapToInt(Arrays::stream)
                .toArray();
    }
}