所有子序列和STL

1917: E

Time Limit: 1 Sec   Memory Limit: 128 MB
Submit: 110   Solved: 15

Submit Status Web Board

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

用两次单调栈分别求出ai左边最近且小于ai的位置L,和右边最近且小于ai的位置R，然后起点位于L+1到i，终点位于i到R-1的区间价值都为ai。
这样就可以求出所有子序列的价值和。

看了别人的代码还是不懂，学的东西太少，有些都不知道什么用法，还是先多掌握点算法再来看这道题。。。

#include<cstdio>  
#include<cstring>  
#include<algorithm>  
#include<vector>  
using namespace std;  
#define LL long long  
#define MA 60000  
struct node{  
    int hao,zhi;  
}dian[MA];  
bool cmp(node xx,node yy)  
{  
    return xx.zhi<yy.zhi;  
}  
int n;   
int main()  
{  
    int t;scanf("%d",&t);  
    while (t--)  
    scanf("%d",&n);  
    for (int i=0;i<n;i++)  
    {  
        scanf("%d",&dian[i].zhi);  
        dian[i].hao=i+2;  
    }  
    LL ans=0,lp;  
    int zuo,you;  
    sort(dian,dian+n,cmp);  //使用upper_bound之前必须保证为有序数列 
    vector <int > ve;//虚拟存储--一维  
    ve.push_back(1);  
    ve.push_back(n+2);  
    for (int i=0;i<n;i++)  
    {  
        int k=dian[i].hao;  
        int kp=upper_bound(ve.begin(),ve.end(),k)-ve.begin();  
        lp=(ve[kp]-k)*(k-ve[kp-1]);  
        ans+=lp*dian[i].zhi;  
        ve.insert(ve.begin()+kp,k);  
    }  
    printf("%lld\n",ans);  
    return 0;  
}