LeetCode 1395. 统计作战单位数

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;
    }
}

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

posted @ 2020-05-19 22:04  ZJPang  阅读(132)  评论(0编辑  收藏  举报