如何用Java BiFunction优化迭代式活动排名赋值代码？

你的需求是给已排序的Activity列表分配排名，规则是相邻元素qualityAverage相同时共享同一排名，不同则排名递增。下面用函数式编程（结合Stream.reduce和状态追踪）实现更简洁的版本：

实现方案 #

首先定义一个轻量状态类，用来跟踪遍历过程中的当前排名和前一个元素的qualityAverage：

private static class RankState {
    int currentRank;
    Float previousAverage;

    RankState(int currentRank, Float previousAverage) {
        this.currentRank = currentRank;
        this.previousAverage = previousAverage;
    }
}

然后通过Stream.reduce来完成遍历和排名赋值：

if (!activities.isEmpty()) {
    // 初始化第一个元素的排名与状态
    RankState initialState = new RankState(1, activities.get(0).getQualityAverage());
    activities.get(0).setRank(initialState.currentRank);

    activities.stream().skip(1).reduce(initialState, (state, currentActivity) -> {
        float currentAverage = currentActivity.getQualityAverage();
        // 判断是否需要递增排名
        int newRank = Float.compare(currentAverage, state.previousAverage) == 0 
            ? state.currentRank 
            : state.currentRank + 1;
        // 为当前元素设置排名
        currentActivity.setRank(newRank);
        // 返回更新后的状态，供下一个元素使用
        return new RankState(newRank, currentAverage);
    }, (state1, state2) -> state1); // 并行流合并逻辑，此处无需实际合并，返回任意状态即可
}

代码说明 #

1. 状态类RankState：替代原代码中的外部可变变量rank，把遍历需要的状态（当前排名、前一个平均值）封装起来，符合函数式编程的无副作用原则。
2. 初始化处理：手动处理第一个元素，设置初始排名并初始化状态对象。
3. Stream遍历逻辑：用skip(1)跳过已处理的第一个元素，通过reduce的累积器（BiFunction类型参数）完成核心逻辑——比较当前元素与前一个的平均值，决定排名是否递增，更新当前元素的排名后返回新状态。

紧凑变体（无需自定义类） #

如果不想额外定义状态类，也可以用AtomicReference封装状态数组，代码更紧凑：

if (!activities.isEmpty()) {
    AtomicReference<int[]> state = new AtomicReference<>(new int[]{1, Float.floatToIntBits(activities.get(0).getQualityAverage())});
    activities.get(0).setRank(1);

    activities.stream().skip(1).forEach(activity -> {
        int[] currentState = state.get();
        int currentRank = currentState[0];
        float prevAvg = Float.intBitsToFloat(currentState[1]);
        float currAvg = activity.getQualityAverage();

        int newRank = Float.compare(currAvg, prevAvg) == 0 ? currentRank : currentRank + 1;
        activity.setRank(newRank);
        state.set(new int[]{newRank, Float.floatToIntBits(currAvg)});
    });
}

这里用int[]存储排名和前一个平均值的位表示（避免浮点数直接==比较的精度问题），通过AtomicReference实现状态的安全更新。

注意事项 #

• 因为列表是已排序的，必须保证顺序遍历，禁止使用并行流（会破坏排名逻辑）。
• 浮点数比较建议用Float.compare替代直接==，避免精度误差导致的错误判断。

内容的提问来源于stack exchange，提问作者Alain Carmona