Minimum Sum(二分+RMQ查找下标) AtCoder Grand Contest 005

Minimum Sum

题目描述

One day, Snuke was given a permutation of length N, a1,a2,…,aN, from his friend.

Find the following:

Constraints
1≤N≤200,000
(a1,a2,…,aN) is a permutation of (1,2,…,N).

输入

The input is given from Standard Input in the following format:
N
a1 a2 … aN

输出

Print the answer.
Note that the answer may not fit into a 32-bit integer.

样例输入

3
2 1 3

样例输出

9

题意:给出n个数,找出所有L(1~n)----R(L~n)的最小值的和。

按一般的想法来做,要用3层for循环这样肯定超时。

先找出从1-n之间的最小值,记住下标i,那么这一个最小值就会被用到(i-L+1)*(R-i+1)次,之后把这个数组从最小值哪里分成两个新的数组,之后再重复之前的过程。

这个过程和快速排序的过程类似。

然后每一次找最小值下标时如果直接找也会超时,这时候就要用到RMQ算法了。


#include<iostream>
#include<stdlib.h>
#include<stdio.h>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
ll a[200105];
ll sum=0;
int d[200105][35],n;
 
void init()
{
    int i,j;
    for(i=0;i<n;i++)
        d[i][0]=i;
    for(j=1;(1<<j)<=n;j++)
    for(i=0;i+(1<<j)-1<n;i++)
        d[i][j]=a[d[i][j-1]] < a[d[i+(1<<(j-1))][j-1]]? d[i][j-1]:d[i+(1<<(j-1))][j-1];
}
int RMQ_indx_min(int s,int v)
{
    int k=(int)(log((v-s+1)*1.0)/log(2.0));
    return a[d[s][k]]<a[d[v-(1<<k)+1][k]]? d[s][k]:d[v-(1<<k)+1][k];
}


void dfs(int l,int r)
{
    if(l==r){
        sum+=(ll)a[l];
        return ;
    }
    int minn=RMQ_indx_min(l,r);
    sum=sum+(ll)(minn-l+1)*(r-minn+1)*a[minn];
    if(l<=minn-1)
        dfs(l,minn-1);
    if(r>=minn+1)
        dfs(minn+1,r);
}
int main (){
    scanf("%d",&n);
    for(int i=0;i<n;i++)
        scanf("%lld",&a[i]);
    init();
    dfs(0,n-1);
    printf("%lld\n",sum);
    return 0;
}




阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/sxh759151483/article/details/79951449
个人分类: RMQ
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

不良信息举报

Minimum Sum(二分+RMQ查找下标) AtCoder Grand Contest 005

最多只允许输入30个字

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭