统计作战单位数

/*
 * 题目思路：
 * 1. 使用 Java Stream API 实现类似的功能。
 * 2. 我们可以使用 IntStream 的双重流来遍历可能的组合，并进行条件检查。
 */

import java.util.stream.IntStream;

public class Solution {
    public int numTeams(int[] rating) {
        int n = rating.length;
        return (int) IntStream.range(0, n - 2)
            .boxed()
            .flatMap(i -> IntStream.range(i + 1, n - 1)
                .boxed()
                .flatMap(j -> IntStream.range(j + 1, n)
                    .filter(k -> (rating[i] < rating[j] && rating[j] < rating[k]) ||
                                  (rating[i] > rating[j] && rating[j] > rating[k]))
                    .mapToObj(k -> new int[]{i, j, k})))
            .count();
    }
}