题目描述
在数列 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)