leetcode-1395. 统计作战单位数

最新推荐文章于 2022-01-20 20:32:58 发布

AI_Younger_Man

最新推荐文章于 2022-01-20 20:32:58 发布

阅读量158

点赞数

分类专栏： # Leetcode

本文链接：https://blog.csdn.net/qq_38888209/article/details/108819922

版权

Leetcode 专栏收录该内容

84 篇文章 0 订阅

订阅专栏

统计作战单位数
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

思路：
因为子数组可以是递增，也可以是递减，又因为本题目的数据估摸不大，所以可以使用暴力搜索。
思路2：
组合数学的思想：扫面一遍，计数J的左边比我小的，有L个，计数J的右边比我大的有R个。递增情况下，总共有L * R个
递减情况下，左边有 J - L 个比我大的。右边有 N-1 - J - R 个比我小的（N是个数所以要先-1）
苏有情况 = 递增 + 递减
在这里插入图片描述

from typing import List


class Solution:
    def numTeams(self, rating: List[int]) -> int:
        n = len(rating)
        ans = 0
        for i in range(0, n):
            i_l = 0
            i_r = 0
            for j in range(0, i):
                if rating[j] < rating[i]:
                    i_l += 1
            for k in range(i, n):
                if rating[k] > rating[i]:
                    i_r += 1
            ans += (i_r * i_l) + (i - i_l) * (n - 1 - i - i_r)
        return ans