参考:https://blog.csdn.net/qq_48081868/article/details/114699362
问题描述
小明这些天一直在思考这样一个奇怪而有趣的问题:
在1~N的某个全排列中有多少个连号区间呢?这里所说的连号区间的定义是:
如果区间[L, R] 里的所有元素(即此排列的第L个到第R个元素)递增排序后能得到一个长度为R-L+1的“连续”数列,则称这个区间连号区间。
当N很小的时候,小明可以很快地算出答案,但是当N变大的时候,问题就不是那么简单了,现在小明需要你的帮助。
输入格式
第一行是一个正整数N (1 <= N <= 50000), 表示全排列的规模。
第二行是N个不同的数字Pi(1 <= Pi <= N), 表示这N个数字的某一全排列。
输出格式
输出一个整数,表示不同连号区间的数目。
样例输入1
4
3 2 4 1
样例输出1
7
样例输入2
5
3 4 2 5 1
样例输出2
9
解决:
思路:记录区间最大最小值,判断最大值与最小值之间的距离是否与区间长度相同。同时应用剪枝思想,若不剪枝则只能得到80分,加入剪枝可得到满分。剪枝方法为,若当前区间最大最小值之间的差距已经大于区间开始到整个序列末尾的长度,则无需继续扩展区间。
N = int(input())
l = [int(i) for i in input().split()]
total = len(l)
for start in range(len(l)-1):
min_s = max_s = l[start]
for end in range(start+1, len(l)):
if l[end] < min_s:
min_s = l[end]
elif l[end] > max_s:
max_s = l[end]
if max_s - min_s == end - start:
total += 1
if max_s - min_s + 1 > end-start+1+len(l)-1-end: # 剪枝
break
print(total)