【zzuli-oj】-1917-定义序列价值（二分 & STL，好）

1917: E

Time Limit: 1 Sec   Memory Limit: 128 MB
Submit: 135   Solved: 28

Submit Status Web 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

题解：首先理解：定义一个序列的价值为序列中所有元素的最小值。

看例子：5 4 1 3 2

以 1 为最小值的序列价值和为：1*（L+1）*（R+1）=9 L：1左边大于1的个数 2。R：1右边大于1的个数 2

以 2 为最小值的序列价值和为：2*2（2,23） L:   2左边大于2的个数1（不能包含1，所有只有3）。R :   2右边大于2的个数 0

以 3 为最小值的序列价值和为：1 （只有3）L:   3左边大于3的个数0。R :   3右边大于3的个数 0

以 4 为最小值的序列价值和为：4*2（4和4 5） L:   4左边大于4的个数1。R :   4右边大于4的个数 0

以 5 为最小值的序列价值和为：5*1（只有5）  L:   5左边大于5的个数0。R :   5右边大于5的个数 0

先把数字编码，然后sort一下。依次取当前最小的数，往vector中插入（二分插入），它前后的下标是自身贡献的区间的区间边缘，就是说，到那些数的时候，它的贡献停止。

为了方便计算，我们先把vector中加入一个0一个n+1。

#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
#define CLR(a,b)  memset(a,b,sizeof(a))
#define M 50000+10
#define LL long long
struct node
{
	int num,k;
}a[M];
bool cmp(node x,node y)
{
	return x.num<y.num;	
} 
vector<int> G;			//存放数值对应的序号 
int main()
{
	int u;
	scanf("%d",&u);
	while(u--)
	{
		int n;
		scanf("%d",&n);
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i].num);
			a[i].k=i;				//数值对应的序号 
		}
		sort(a+1,a+1+n,cmp);			//数值排序 
		G.clear();
		LL ans=0;
		G.push_back(0);
		G.push_back(n+1);			//保险起见，先加入 0,n+1 
		for(int i=1;i<=n;i++)
		{
			int pos=upper_bound(G.begin(),G.end(),a[i].k)-G.begin();	//pos是 vectorG[]里第一个大于a[i].k的数的位置 
			ans+=a[i].num*(a[i].k-G[pos-1])*(G[pos]-a[i].k);
			G.insert(G.begin()+pos,a[i].k);				//在正好大于a[i].k的位置插入序号a[i].k 
		}		
		printf("%lld\n",ans);
	}
	return 0;
}