统计作战单位数

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

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

分析: 不仅要取出数组中的每个元素的下角标按从小到大排序, 而且每个元素的数值排序要按从小到大排,也可以从大到小排.

解法一 | 动态规划法

def numTeams(rating):
     def func(nums):  
         #定义动态数组记录存储每个符合条件的解            
         dp = [0] * len_                   
         res = 0
         for i in range(1, len_):
             idx = i - 1
             # 用于循环, idx不可能<0, idx用作下角标排序比较使用
             while idx >= 0:
                 if nums[i] > nums[idx]:
                     dp[i] += 1
                     if dp[idx] > 0:
                         res += dp[idx]
                 idx -= 1
         return res

     len_ = len(rating)
     return func(rating[::-1]) + func(rating)

 print(numTeams([2,5,3,4,1]))

解法二 |

 def numTeams(rating):
     if len(rating) < 3:
         return 0
     answer = 0
     for ind, i in enumerate(rating): 
         # rating[ind+1],保证下角标顺序. 3次循环分别筛选出3个下角标按序排列的元素
         for ind_j, j in enumerate(rating[ind + 1:], ind + 1):
             for ind_z, z in enumerate(rating[ind_j + 1:], ind_j + 1):
                 //判断元素的数值是否符合要求
                 if i < j and j < z and z > i:
                     answer += 1

                 elif i > j and j > z and i > z:
                     answer += 1

     return answer

解法三 | 引入itertools工具包

import itertools
def func(rating):
    res=0
    # combinations函数枚举从5个元素中选出3个的所有不同的排列组合, 3即选出元素的个数, 省去了数组长度判断的步骤
    for i,j,k in itertools.combinations(rating,3):
        if i<j<k or i>j>k:
            print(i,j,k)
            #当符合条件时,计数+1
            res+=1
    return res
print(func([2,5,3,4,1]))