LeetCode 1395. 统计作战单位数

Posted 2021-02-18 菜鸡的世界

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

这个题目虽然是中等题，但是给的测试用例很少，最大才200个，所以直接暴力三循环也可以过，时间复杂度O(n^3)，代码就不贴了。

后来看了花花酱的题解，学到个更加牛逼的技巧？

我们固定一个i,然后寻找左边有多少个比rating[i]小的记作l，右边有多少个比其大的记作r. 然后区间长度减去l或者r就能算出左边有多少个比rating[i]大的，右边有多少个比其小的 相加就是答案

class Solution {
    public int numTeams(int[] rating) {
        int ans = 0;
        for(int j = 0; j < rating.length; j++){
            int left = 0;
            int right = 0;
            for(int i = 0; i < j; i++){
                if(rating[i] < rating[j]){
                    left++;
                }
            }
            for(int k = j+1; k < rating.length; k++){
                if(rating[k] > rating[j]){
                    right++;
                }
            }
            ans += (left * right) + ((j-left) * (rating.length - j - right - 1));
        }
        return ans;
    }
}

其实这个就是利用到了数学特性，算法还是靠数学呀