校园自行车分配

/*
 * 思路：
 * 1. 使用流计算每对学生和自行车的距离，并存储到一个列表中。
 * 2. 对列表进行排序。
 * 3. 使用流从最短距离开始分配自行车给学生。
 */

import java.util.*;
import java.util.stream.*;

public class CampusBikeAssignmentStream {
    public int[] assignBikes(int[][] students, int[][] bikes) {
        int n = students.length;
        int m = bikes.length;
        int[] result = new int[n];
        boolean[] bikeAssigned = new boolean[m];
        Arrays.fill(result, -1);

        List<int[]> distances = IntStream.range(0, n)
            .boxed()
            .flatMap(i -> IntStream.range(0, m)
                .mapToObj(j -> new int[]{Math.abs(students[i][0] - bikes[j][0]) + Math.abs(students[i][1] - bikes[j][1]), i, j}))
            .sorted(Comparator.comparingInt(a -> a[0]))
            .collect(Collectors.toList());

        distances.stream()
            .filter(distance -> result[distance[1]] == -1 && !bikeAssigned[distance[2]])
            .forEach(distance -> {
                result[distance[1]] = distance[2];
                bikeAssigned[distance[2]] = true;
            });

        return result;
    }
}