|
发表于 2023-11-18 23:51:41
|
显示全部楼层
- # 统计两个相等元素之间的元素个数
- def count_equal_pairs(arr):
- count = 0
- for i in range(len(arr)):
- for j in range(i+1, len(arr)):
- if arr[ i] == arr[j]:
- count += (j - i - 1)
- return count
- # 统计三个相等元素之间的元素个数
- def count_equal_triplets(arr):
- count = 0
- for i in range(len(arr)):
- for j in range(i+1, len(arr)):
- if arr[ i] == arr[j]:
- for k in range(j+1, len(arr)):
- if arr[j] == arr[k]:
- count += (k - j - 1)
- return count
- arr = [1, 2, 1, 3, 2]
- print(count_equal_pairs(arr)) # 输出:2
- print(count_equal_triplets(arr)) # 输出:0
复制代码
这段代码中,首先定义了两个函数 count_equal_pairs 和 count_equal_triplets ,分别用于计算两个相等元素之间的元素个数和三个相等元素之间的元素个数。然后在主程序中给出了一个示例数组 arr ,并分别调用这两个函数进行计算并输出结果。 count_equal_pairs 函数使用两层循环遍历数组,当找到相等的元素时,计算两个相等元素之间的元素个数并累加到 count 中。 count_equal_triplets 函数同样使用三层循环来计算三个相等元素之间的元素个数。 |
|