Description
晴天有非常严重的选择恐惧症,每次吃饭前他都在纠结到底吃什么。。今天又到了吃饭的时候了。
重光:我给你一个包含n个不同整数的序列a,如果它所有连续子序列的价值和是素数咱们就吃米,不然就吃面。
定义一个序列的价值为序列中所有元素的最小值。
晴天:这不是分分钟给你算出来。
嗯…十分钟过去了,晴天选择死亡。
这个任务就交给你啦。
算出所有连续子序列的价值和。
Input
第一行输入一个整数t,代表有t组测试数据。
每组数据第一行包含一个整数n,表示序列a的元素个数。
接下来一行包含n个整数,表示序列a。
0<=n<=50000,1<=ai<=50000。
Output
对于每组数据输出一个整数,表示序列a的所有连续子序列的价值和。
Sample Input
1
3
1 2 3
Sample Output
10
今天我也做了一个rmq的题,第一次做,是参考我的同学的代码,我真的要谢谢他了,又让我见到我未见到过的知识
不多说了,看看代码就知道了
#include<stdio.h>
#include<iostream>
#include<math.h>
#include<algorithm>
#include<string.h>
using namespace std;
int a[50010][20],n;
void 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();
int l,mid,mini;
for(i=1; i<=n; i++)
{
int r=i,
l=1;
while(l<=r)
{
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)
{
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);
}
return 0;
}