Range Sorting (Hard Version)
题意:
给出一个数组,对所有子数组,找到其中的非升序区间r,l。其花费为r - l。求数组a的总花费。
hard范围变成了3e5,不能像easy题那样暴力了。
思路:
先初始化答案,即对于每个i,ans += (i - 1) * ((i - 1) + 1) / 2。
即等差数列求和,把所有合法不合法的花费都加起来,后面再把合法的减去。合法即升序,不合法为非升序。
随后对于每个位置i,找到它前面的合法区间和后面的合法区间,ans减去前面的合法区间*后面的合法区间。
这题思路还是一样,但是找合法区间得换一种方法,因为暴力会超时。
至于每个位置后面的合法区间,可以用r数组代替,可以用单调栈的方法先预处理好r数组。
对于前面的合法区间可以用单调栈先找到区间的右端点,然后再用二进制倍增的思想找到左断点。
其中tmi[k][i]表示以i为右端点,长度为2^k的区间的最大值。
具体操作看代码。
代码:
/*************************************************************************
> File Name: d2.cpp
> Author: Beans
> Mail: 3112748286@qq.com
> Created Time: 2023/5/15 11:38:39
************************************************************************/
#include <iostream>
#include <algorithm>
#include <stack>
#include <vector>
#define int long long
#define endl '\n'
using namespace std;
const int maxn = 3e5 + 7;
const int ML = 19;
int t, n, a[maxn];
void solve(){
cin >> n; a[n + 1] = 0;
int ans = 0;
vector<vector<int>> tmi(ML, vector<int>(n + 1));
for(int i = 1; i <= n; i ++ ) cin >> a[i], ans += (i - 1) * ((i - 1) + 1) / 2, tmi[0][i] = a[i];
vector<int> s;
s.push_back(n + 1);
vector<int> r(n + 1);
for(int i = n; i >= 1; i -- ){
while(s.size() && a[s.back()] > a[i])
s.pop_back();
r[i] = s.back();
s.push_back(i);
}
for(int k = 1; k < ML; k ++ )
for(int i = 1 << k; i <= n; i ++ )
tmi[k][i] = max(tmi[k - 1][i], tmi[k - 1][i - (1 << (k - 1))]);
s.clear();
s.push_back(0);
for(int i = 1; i <= n; i ++ ){
while(s.size() && a[s.back()] > a[i])
s.pop_back();
int j = s.back();
for(int k = ML - 1; k >= 0; k -- )
if(j >= (1 << k) && tmi[k][j] < a[i])
j -= (1 << k);
ans -= (r[i] - i) * (s.back() - j);
s.push_back(i);
}
cout << ans << endl;
}
signed main(){
ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
cin >> t;
while(t -- )
solve();
}