蓝桥杯练习-146三元组中心问题

题目描述

在数列 a1​,a2​,⋯,an​ 中，如果对于下标 i, j, k 满足 0 < i < j < k < n+1 且 ai​<aj​<ak​，则称 ai​,aj​,ak​ 为一组递增三元组，aj​ 为递增三元组的中心。

给定一个数列，请问数列中有多少个元素可能是递增三元组的中心。

输入描述

输入的第一行包含一个整数 n。

第二行包含 n 个整数 1,2,⋯,a1​,a2​,⋯,an​，相邻的整数间用空格分隔，表示给定的数列。

其中，2≤n≤1000，0≤数列中的数≤10000。

输出描述

输出一行包含一个整数，表示答案。

样例输入

5

1 2 5 3 5

样例输出

2

样例说明

a2​ 和 a4​ 可能是三元组的中心。

运行限制

最大运行时间：1s

最大运行内存: 256M

解决思路

如何判定一个数是否是三元组的中心？其实只要在它的左边有小于它的数，它的右边有大于它的数就可以了。

那么我们需要遍历两次nums数组，第一次遍历从左往右，记录下nums[0...i-1]中的最小值min_num；第二次遍历从右往左，记录下nums[i+1...n-1]中的最大值max_num。如果nums[i]大于min_num，小于max_num，那么nums[i]就是三元组的中心。

思路很简单，来看代码：

n = int(input())
nums = list(map(int,input().split()))

# dp[i]是dp[i+1...]中的最大值 用一维数组记录一下
dp = [nums[n-1]]*n

for i in range(n-2,-1,-1):
    dp[i] = max(nums[i+1],dp[i+1])

# min_num是dp[0...i-1]中的最小值
min_num = nums[0]
c = 0
for i in range(1,n):
    if min_num<nums[i] and nums[i]<dp[i]: # 边遍历边比较
        c += 1
    min_num = min(min_num,nums[i])

print(c)