01.12 力扣题目(统计作战单位数)

题目

n 名士兵站成一排。每个士兵都有一个 独一无二 的评分 rating 。
每 3 个士兵可以组成一个作战单位，分组规则如下：
从队伍中选出下标分别为 i、j、k 的 3 名士兵，他们的评分分别为 rating[i]、rating[j]、rating[k]
作战单位需满足： rating[i] < rating[j] < rating[k] 或者 rating[i] > rating[j] > rating[k] ，其中 0 <= i < j < k < n
请你返回按上述条件可以组建的作战单位数量。每个士兵都可以是多个作战单位的一部分

思路

可以写三层for循环遍历数组，求出每个元素，因为0 <= i < j < k ，所以j可以=i+0，k=j+0；然后按照条件进行依次比较，符合条件的count++，返回结果。

问题

上传答案时出现超出时间限制，在指导下做了点稍微简单点的优化，首先因为他是三名士兵，我们可以把他中间的那个士兵找出来，也就是j，然后写一个循环，j的下标应该是从1开始，长度应该是数组的长度减1，然后求出比他小的数有几个，比他大的数有几个，依次循环，求出每个元素的比他小或大的数的个数，此时的次数应该就是左边比他小的数的个数乘右边比他大的数的个数加上左边比他大的数的个数乘右边比他小的数的个数。代码如下

public static int numTeams2(int[] rating) {
        int count = 0;
        for (int j = 1; j < rating.length - 1; j++) {
            int leftMin = 0;
            int leftMax = 0;
            int rightMin = 0;
            int rightMax = 0;
            for (int i = j - 1; i >= 0; i--) {
                if (rating[i] < rating[j]) {
                    leftMin++;
                } else {
                    leftMax++;
                }
            }
            for (int k = j + 1; k < rating.length; k++) {
                if (rating[k] < rating[j]) {
                    rightMin++;
                } else {
                    rightMax++;
                }
            }
            count = count + (leftMin * rightMax + leftMax * rightMin);
        }
        return count;
    }

做完优化后也有一点启发，像以后工作中遇到三层四层更多的循环的时候应该想一下怎么把他优化成两次循环，一次循环，减少一次，消耗的时间和内存也会减少很多。

代码

public static int numTeams(int[] rating) {

        int count = 0;
        for (int i = 0; i < rating.length; i++) {
            for (int j = i + 0; j < rating.length; j++) {
                for (int k = j + 0; k < rating.length; k++) {
                    if ((rating[i] < rating[j] && rating[j] < rating[k]) || (rating[i] > rating[j] && rating[j] > rating[k])) {
                        count++;
                    }
                }
            }
        }

        return count;
    }