【Java】LeetCode 1395. 统计作战单位数

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
请你返回按上述条件可以组建的作战单位数量。每个士兵都可以是多个作战单位的一部分。
来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/count-number-of-teams
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

简单来说，就是给你一个数组，找出所有i<j<k使得数组对应值严格单增/单减

第一想到的当然是暴力破解了，直接遍历所有ijk，三层循环很容易实现

class Solution {
    public int numTeams(int[] rating) {
    	//记录满足条件的ijk个数
        int count=0;
        for(int i=0;i<rating.length;i++){
            for(int j=i+1;j<rating.length;j++){
                for(int k=j+1;k<rating.length;k++){
                    if((rating[i]>rating[j]&&rating[j]>rating[k])
                    	||(rating[i]<rating[j]&&rating[j]<rating[k])){
                    	//单增或者单减计数
                        count++;
                    }
                }
            }
        }
        return count;
    }
}

性能似乎已经很不错了

再想想：对于正中间的j来说，要遍历所有的ik情况大概是要（N/2）² ，是否可以节省一点？

对于j来说，只要Num（i）>Num(j)且Num（i）<Num(j)或者Num（i）<Num(j)且Num（i）<Num(j)即可，那么，我们把ik都分成两堆：大于Num（j）或者小于Num（j）（个数分别记为Si Sk Bi Bk）

然后Si*Bk+Bi*Sk就是固定i所对应满足条件的个数

这样的话，只需要遍历一次数组就可以了（N次访问）

class Solution {
    public int numTeams(int[] rating) {
        int count=0;
        int Bi,Bk,Si,Sk;
        for(int j=0;j<rating.length;j++){
            Bi=0;Bk=0;Si=0;Sk=0;
            for(int i=0;i<j;i++){
                //记录各个i和j的大小关系
                //注意不能用简单的..?..:..等于的情况不能计数
                if(rating[i]>rating[j]){
                    Bi++;
                }else if(rating[i]<rating[j]){
                    Si++;
                }
            }
            for(int k=j+1;k<rating.length;k++){
                //记录k和j的大小关系
                if(rating[k]>rating[j]){
                    Bk++;
                }else if(rating[k]<rating[j]){
                    Sk++;
                }
            }
            count+=(Bk*Si+Sk*Bi);
        }
        return count;
    }
}

由于对于一个j来说，所需要遍历的i和k从N² 数量级一下降低为N，总的次数也从N³ 降为N² ，因此效率提升了很多