题目描述:
链接:登录—专业IT笔试面试备考平台_牛客网
来源:牛客网
示例1
输入
3 3 4 2 3 5 1 8 4 3 9 20 2 8 15 1 10 5 19 19 3 5 6 6 2 8 2 12 16 3 8 17
输出
5
57
2712
说明
对于一组测试数据的解释: 区间[1, 2]的贡献为:4 - 2 = 2 区间[1, 3]的贡献为:4 - 2 = 2 区间[2, 3]的贡献为:3 - 2 = 1 2 + 1 + 2 = 5.
备注:
T⩽20,n⩽10^5,0⩽ai⩽10^5
不保证数据随机生成!
思路:(来自大佬)
先对式子进行简化,可以变成求区间所有最大值之和减去区间所有最小值之和。那么关键就在于求出所有区间的最大值以及最小值。但是如果光暴力的话(枚举区间长度+区间平移),是会超时的,那么可以选择换一种策略,选择对数组的元素进行枚举,那么建立两个数组l、r,则可以表示出以它作为最大值的区间的下界l【i】、上界r【i】,那么对于每一个枚举出来的元素(假设为mas【i】,就可以求出以它为最大值的区间的最大值的总和(最大值,即该元素的值,乘以所有满足条件的区间的个数),应该分为两部分,一种情况:i是区间端点,比如2 3 3 4 ,4作为区间端点。那么它所对应的值就应该是mas【i】*(r【i】-l【i】),即最大值*区间长度。还有另一种情况,该元素在区间内部,所有满足条件的区间的个数就应该是其左边能取的元素的个数(作为下界)*其右边能取的元素的个数(作为上界),即:mas【i】*(r【i】-i)*(i-l【i】);对所有元素遍历,将其两个部分相加,就得到所有最大值的总和了。那么对于最小值,只要把所有元素取反,再进行同样操作就可以了(因为元素取反后,上下界也会变化并改变大小关系,那么两个负的相乘,结果还是正的);
还有几个细节要注意,因为mas【i】是有可能作为区间端点存在的,所以 在求l【i】、r【i】时,枚举的j应该定义为i,而不是i+1或i-1。然后是取等问题,象3 1 4 4 4这个数组,三个4是重复的,都可以作为同一个区间的最大值(最小值),显然,求上界、下界两种情况时,如果我们不考虑等号的话,情况是会有遗漏的,比如遗漏3 1 4 4 这个区间(如果我们选第一个4作为最大值)。但是等号也不能全加,否则,以第一个4作为最大值时,会包括3 1 4 4 4 这个区间,而以第二个4作为最大值时,也会把这个3 1 4 4 4区间再次包含,那么情况就重了。那么应该只加一个等号,那么加哪个呢?其实这个等号就是代表了一种方向,即在不加等号的那种枚举里(求上界/下界),求的是满足这个方向内满足严格小于该元素的区间的端点,所以其实等号加在那个都没事,只要加了就行。还是上面那个例子,如果求下界时可以取等,但求上界时要求严格小于,则:当我们取第一个4作为区间最大值时,只能枚举到3 1 4这个区间,而以第二个4作为最大值时,就可以包括3 1 4 4,而不会有3 1 4 这个区间,也不会有3 1 4 4 4这个区间,这样就确保了对于每个元素,其区间的唯一性。最后一个,是在求区间上界时的枚举顺序,应该是逆序。
r【i】返回的是对于第i个元素,下标最大的不大于它的元素的下标,刚进入循环时,j=n,退出,r【n】=n;然后j=n-1,如果mas【n-1】满足条件的话,j=r【n-1+1】=n;如果不满足,就这样依次向下取。逆序枚举正是为了让一开始的r数组有具体的数字,如果正着枚举的话,就会陷入死循环,因为r的任意元素都没有值。
代码:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
ll t,n,mas[100010],sum=0;
ll l[100010],r[100010];//l:区间下界,l[i]表示下标;r:区间上界
ll f(){
//ll l[100010],r[100010];//l:区间下界,l[i]表示下标;r:区间上界
for(int i=1;i<=n;++i){
int j=i;//i有可能作为区间端点,即l[i]==i;
while((j>1)&&(mas[j-1]<mas[i]))
j=l[j-1];//迭代向下;
l[i]=j;
}
for(int i=n;i>=1;--i){
int j=i;
while((j<n)&&(mas[j+1]<=mas[i]))
j=r[j+1];
r[i]=j;
}
ll ans=0;
for(int i=1;i<=n;++i){
ans+=mas[i]*(r[i]-l[i]);
ans+=mas[i]*(r[i]-i)*(i-l[i]);
}
return ans;
}
int main()
{
cin>>t;
while(t--){
//sum=0;
cin>>n;
for(int i=1;i<=n;++i) cin>>mas[i];
sum=f();
for(int i=1;i<=n;++i) mas[i]=-mas[i];
sum+=f();
cout<<sum<<endl;
}
return 0;
}