【STL+前缀和】hihocoder1604 枚举+优化套路 股票价格Ⅱ

传送门：

1604 股票价格Ⅱ

想法：

在最高价格之前每个股票都买，且总在价格最高时全部卖出手头上的股票，这样挣得最多。

算法：

每次扫描一下当前数组中存在的最大值及其位置，求最大值与在此之前的每个元素之差，求和。下次数组删去最大值位置之前的数组，重复此过程。

实现：

应用(ordered_)map和(ordered_)set组成的map<int,set<int> >

其中map的pair.first代表股票价格(value)，pair.second的set存放的是出现该值的位置(position)

根据map的有序特性，map.end()-1是键值最大的pair，也就是价格最高。如果最高的天数不止一天，只需要知道最后一次价格最高是哪天就好，因为上一次最高的时候买或者不买对于求差值是无影响的。根据set的有序特性，set.end()-1是值最大的，也就是最大值最后出现的位置。

但是，还有一个问题，每个元素对应求差值再求和的话，所需时间未免太长，因此，维护一个前缀和数组，用空间换时间。求某一段的和，只需要sum[r]-sum[l]即可。

代码：

#include <bits/stdc++.h>
using namespace std;
int sum[1000010];
int main(){
	int n,x;
	map<int,set<int> >m;
	scanf("%d",&n);
	for(int i = 1;i <= n;i++){
		scanf("%d",&x);
		m[x].insert(i);
		sum[i] = sum[i-1] + x;
	}
	int pos = 0,ans = 0;
	auto it = m.end();
	for(it--;it != m.begin();it--){
		auto iter = (it -> second).end();
		iter--;
		if(*iter > pos){
			ans += (*iter - pos) * (it -> first) - sum[*iter] + sum[pos];
			pos = *iter;
		}
	}
	printf("%d\n",ans);
	return 0;
}