所有子序列和STL

1917: E

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

SubmitStatusWeb Board

Description

晴天有非常严重的选择恐惧症，每次吃饭前他都在纠结到底吃什么。。今天又到了吃饭的时候了。

重光：我给你一个包含n个不同整数的序列a，如果它所有连续子序列的价值和是素数咱们就吃米，不然就吃面。

定义一个序列的价值为序列中所有元素的最小值。

晴天：这不是分分钟给你算出来。

嗯...十分钟过去了，晴天选择死亡。

这个任务就交给你啦。

算出所有连续子序列的价值和。

Input

第一行输入一个整数t，代表有t组测试数据。
每组数据第一行包含一个整数n，表示序列a的元素个数。
接下来一行包含n个整数，表示序列a。
0<=n<=50000，1<=ai<=50000。

Output

对于每组数据输出一个整数，表示序列a的所有连续子序列的价值和。

Sample Input

131 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;  }