1734:排列（New Online Judge）

1734:排列（New Online Judge）

题目描述

给你一个1-n的排列，每个<i,j>对的价值是j-i+1，计算所有满足以下条件<i,j>对的总价值：

• 1≤i<j≤n
• a[i]-a[j]之间的数字均小于min(a[i],a[j])
• a[i]-a[j]之间不存在其他数字则直接满足

输入

输入第一行包含正整数N(N≤300000)
第二行包含N个正整数，表示一个1-N的排列a

输出

输出一个正整数表示答案。

样例输入

7
4 3 1 2 5 6 7

样例输出

24

题解1（C++版本）

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 300010;

int n, a[N], st[N], top = -1;

int main(){
	LL ans = 0; //最终答案可能超出int类型的最大值 
	scanf("%d", &n);
	for(int i = 1; i <= n; i++){
		scanf("%d", &a[i]);
	}
	//从左到右构建一个递减的单调栈 
	for(int i = 1; i <= n; i++){
		while(top != -1 && a[st[top]] < a[i]){
			ans += (1LL*i - st[top] + 1);
			top--;
		}
		st[++top] = i;
	}
	top = -1;
	//从右到左构建一个递增的单调栈 
	for(int i = n; i > 0; i--){
		while(top != -1 && a[st[top]] < a[i]){
			ans += (1LL*st[top] - i + 1);
			top--;
		}
		st[++top] = i;
	}

	printf("%lld\n", ans);
	return 0;
}