今日头条2018校招算法方向（第一批）编程题2（枚举）

时间限制：3秒

空间限制：131072K

给定一个数组序列, 需要求选出一个区间, 使得该区间是所有区间中经过如下计算的值最大的一个：

区间中的最小数 * 区间所有数的和最后程序输出经过计算后的最大值即可，不需要输出具体的区间。如给定序列  [6 2 1]则根据上述公式, 可得到所有可以选定各个区间的计算值:

 

[6] = 6 * 6 = 36;

[2] = 2 * 2 = 4;

[1] = 1 * 1 = 1;

[6,2] = 2 * 8 = 16;

[2,1] = 1 * 3 = 3;

[6, 2, 1] = 1 * 9 = 9;

 

从上述计算可见选定区间 [6] ，计算值为 36， 则程序输出为 36。

区间内的所有数字都在[0, 100]的范围内;

输入描述:

第一行输入数组序列长度n，第二行输入数组序列。
对于 50%的数据,  1 <= n <= 10000;
对于 100%的数据, 1 <= n <= 500000;

输出描述:

输出数组经过计算后的最大值。

输入例子1:

3
6 2 1

输出例子1:

36

#include <bits/stdc++.h>
using namespace std;
int a[500005], pre[500005], l[500005], r[500005];
int main(){
	int n;
	scanf("%d", &n);
	for(int i = 1; i <= n; ++i){
		scanf("%d", &a[i]);
	}
	pre[0] = 0;
	for(int i = 1; i <= n; ++i){
		pre[i] = pre[i - 1] + a[i];
	}
	int L, R;
	for(int i = 1; i <= 100; ++i){
		L = 1;
		for(int j = 1; j <= n; ++j){
			if(a[j] < i){
				L = j + 1;
			}
			if(a[j] == i){
				l[j] = L;
			}
		}
	}
	for(int i = 1; i <= 100; ++i){
		R = n;
		for(int j = n; j >= 1; --j){
			if(a[j] < i){
				R = j - 1;
			}
			if(a[j] == i){
				r[j] = R;
			}
		}
	}
	int ans = 0;
	for(int i = 1; i <= n; ++i){
		ans = max(ans, (pre[r[i]] - pre[l[i] - 1]) * a[i]);
	}
	cout << ans << endl;
	
}

/*
题意：
5e5个数，求所有连续区间中区间和*区间最小值的最大值。

思路：
由于每个数的范围很小，我们可以枚举每一个数字，假设它是所在区间中的最小数，然后看它最左最右的范围。
预处理一下区间和，然后搞一下区间范围，然后枚举答案。。。
*/