2024.1华为od机试C卷【结队编程】Python

目录

题目

代码

---

题目

某部门计划通过结队编程来进行项目开发，已知该部门有 N 名员工，每个员工有独一无二的职级，每三个员工形成一个小组进行结队编程，结队分组规则如下:
从部门中选出序号分别为 i、j、k 的3名员工，他们的职级分别为 leve[i]，level[j]，level[k],结队小组满足 level[i] < level[j] < level[k] 或者 level[i] > level[j] > level[k],其中0<i<j<k<n。
请你按上述条件计算可能组合的小组数量。同一员工可以参加多个小组.

输入描述
第一行输入: 员工总数 n
第二行输入: 按序号依次排列的员工的职级 level，中间用空格隔开
备注:
1≤n≤6000
1 ≤ level[i] ≤10^5

输出描述
可能结队的小组数量

示例1：
输入

4
1 2 3 4
输出

4
说明
可能结队成的组合(1,2,3)、(1,2,4)、(1,3,4)、(2,3,4)

示例2：
输入
3
5 4 7
输出

0
说明
根据结队条件，我们无法为该部门组建小组

代码

n = int(input())
level_lst = list(map(int, input().split(' ')))
length = len(level_lst)
res_num = 0
for i in range(length - 2):
    for j in range(i + 1, length - 1):
        for k in range(j + 1, length):
            group = [level_lst[i], level_lst[j], level_lst[k]]
            if group[0] > group[1] > group[2] or group[0] < group[1] < group[2]:
                res_num += 1
print(res_num)