当前位置: 代码迷 >> 综合 >> leetcode-1395. 统计作战单位数
  详细解决方案

leetcode-1395. 统计作战单位数

热度:53   发布时间:2023-12-28 22:11:56.0
  1. 统计作战单位数
    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 Listclass Solution:def numTeams(self, rating: List[int]) -> int:n = len(rating)ans = 0for i in range(0, n):i_l = 0i_r = 0for j in range(0, i):if rating[j] < rating[i]:i_l += 1for k in range(i, n):if rating[k] > rating[i]:i_r += 1ans += (i_r * i_l) + (i - i_l) * (n - 1 - i - i_r)return ans