之前做过类似的,一眼就看出来rmq+二分,可惜代码能力太菜了,写了快两个小时才写出来。
思想:肯定是一层for()循环遍历每一个下标,对于每个下标i向左查找到第一个比自身值小的右边那个下标left,同时向右查找到第一个比自身值小的左边那个下标right。比如然后(right-i+1)(i-left+1)a[i],即为所求值。比如1 5 4 6 2对应4这个数,其下标为3,那么左边第一个比自身值小的右边那个下标left为2,向右第一个比自身值小的左边那个下标right为4,其所能组成的所有子序列为{5,4},{4},{4,6},{5,4,6},四个,所以一共为(right-i+1)*(i-left+1)*a[i]。
题目:
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
using namespace std;
int a[50010][20],n;
void rmq() //rmq()
{
int i,j;
for(i=1; i<19; i++)
for(j=1; j<=n; j++)
if(j+(1<<i)-1<=n)
a[j][i]=min(a[j][i-1],a[j+(1<<(i-1))][i-1]);
}
int main()
{
int nn;
scanf("%d",&nn);
while(nn--)
{
int i,left,right;
long long sum=0;
scanf("%d",&n);
for(i=1; i<=n; i++)
scanf("%d",&a[i][0]);
rmq();
for(i=1; i<=n; i++)
{
int l,r=i,mid,mini;
l=1;
while(l<=r) //二分查找左边mini
{
mid=(l+r)>>1;
int s=(int)(log (i-mid+1)/log(2));
mini=min(a[mid][s],a[i-(1<<s)+1][s]);
if(mini<a[i][0])
l=mid+1;
else
r=mid-1;
}
left=l;
l=i,r=n;
while(l<=r)//二分查找右边mini
{
mid=(l+r)>>1;
int s=(int)(log(mid-i+1)/log(2));
mini=min(a[i][s],a[mid-(1<<s)+1][s]);
if(mini>=a[i][0])
l=mid+1;
else
r=mid-1;
}
right=l-1;
sum+=(right-i+1)*(i-left+1)*a[i][0];
}
printf("%lld\n",sum);
}
}