Leetcode: NO.1395 统计作战单位数

题目

链接：https://leetcode-cn.com/problems/count-number-of-teams

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
请你返回按上述条件可以组建的作战单位数量。每个士兵都可以是多个作战单位的一部分。

示例 1：

输入：rating = [2,5,3,4,1]
输出：3
解释：我们可以组建三个作战单位 (2,3,4)、(5,4,1)、(5,3,1) 。
示例 2：

输入：rating = [2,1,3]
输出：0
解释：根据题目条件，我们无法组建作战单位。
示例 3：

输入：rating = [1,2,3,4]
输出：4
 

提示：

n == rating.length
1 <= n <= 200
1 <= rating[i] <= 10^5

解题记录

列表中找到3个数值递增或递减即可，可以遍历列表确定为中间值，然后通过查看该值左右是否有满足条件数值使三个数递增或递减即可：

class Solution {
    public int numTeams(int[] rating) {
        int count = 0;
        for (int i=0; i<rating.length; i++) {
            for (int j=0; j<i;j++){
                if (rating[j]<rating[i]){
                    for (int k=i+1; k<rating.length; k++){
                        if (rating[i] < rating[k]) count++;
                    }
                }
            }

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

进阶

中间值不可能在首尾因此循环从1到-2位，通过统计中间值左右两边大于，小于中间值的数值个数，最终相乘得到该中间值满足的情况：

class Solution {
    public int numTeams(int[] rating) {
        int count = 0;
        for (int i=1; i<rating.length-1; i++) {
            int left_more = 0 , left_less = 0, right_more=0, right_less=0;
            for (int j=0; j<i; j++){
                if (rating[j]<rating[i]) left_less++;
                else if (rating[j]>rating[i]) left_more++;
            }
            for (int k=i+1; k<rating.length; k++){
                if (rating[i] < rating[k]) right_more++;
                else if (rating[i] > rating[k]) right_less++;
            }
            count += left_less*right_more + left_more*right_less;
        }
        return count;
    }
}

第一次时间内存都比较靠前：

结果

37,4 -> 100% 那之前的 37.3 岂不是的 101% ，感觉这个统计好诡异，手动滑稽

来源：oschina

链接：https://my.oschina.net/u/4402583/blog/3219139