LeetCode-Python-1395. 统计作战单位数（数组）

最新推荐文章于 2022-04-23 09:35:50 发布

暴躁老哥在线刷题

最新推荐文章于 2022-04-23 09:35:50 发布

阅读量820

点赞数

分类专栏： Leetcode 文章标签： leetcode 算法 python

本文链接：https://blog.csdn.net/qq_32424059/article/details/105194236

版权

Leetcode 专栏收录该内容

813 篇文章 121 订阅

订阅专栏

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/count-number-of-teams
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

第一种思路：

显然正反各来一次三重循环理论上可行，比较麻瓜没有实现。

时间复杂度：O（N^3)

空间复杂度：O（1）

第二种思路：

如果我们提前用双重循环找到每个数组元素有多少个在它右边的数比它大，

那么再来一次双重循环就可以知道所有可行的作战单位。

例如：在第二次双重循环里找到 A < B，那么结果直接加上 B右边比它大的数的个数即可。

时间复杂度：O（N^2)

空间复杂度：O（N）

class Solution(object):
    def numTeams(self, rating):
        """
        :type rating: List[int]
        :rtype: int
        """
        from collections import defaultdict
        
        
        def helper(rating):
            dic = defaultdict(int)
            res = 0
            
            for i in range(len(rating)):
                for j in range(i + 1, len(rating)):
                    if rating[j] > rating[i]:
                        dic[i] += 1


            for i in range(len(rating)):
                for j in range(i + 1, len(rating)):
                    if rating[j] > rating[i]:
                        res += dic[j]
            return res
        
        return helper(rating) + helper(rating[::-1])