结对编程(200)
- 某部门计划通过结队编程来进行项目开发,部门有N名员工,每个员工有唯一的职级,三个员工为一组,结队分组规则如下:
- 选出序号为i,j,k的员工,职级分别为level[i], level[j], level[k],满足level[i]<level[j]<level[k] 或者 level[i] > level[j] > level[k] 其中0<i<j<k<n
- 计算可能组合的小组数量,同一员工可以参加多个小组;
输入描述:
第一行:员工总数n,【1,6000】
第二行:按序号依次排列的员工的职级,level[i] 在【1,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
思路:
- 找到满足两个条件的小组;
- 条件1,0<i<j<k<n;
- 条件2,level[i]<level[j]<level[k] 或者 level[i] > level[j] > level[k]
- 两组对应相乘
n = int(input())
nums = [int(x) for x in input().split(" ")]
compare_count = {}
for i in range(n):
compare_count[i]= [0,0,0,0]
result = 0
k=1
while(True):
if(k>=n):
break
else :
for j in range(k):
if (nums[j] < nums[k]) :
compare_count[k][0] +=1
elif (nums[j] > nums[k]) :
compare_count[k][1] +=1
k+=1
for i in range(n-2, 0,-1):
for j in range(n-1,i,-1):
if (nums[j] < nums[i]) :
compare_count[i][2] +=1
elif (nums[j] > nums[i]) :
compare_count[i][3] +=1
for i in range(n):
temp = compare_count[i]
result += temp[0] * temp[3] + temp[1] * temp[2]
print(result)