题目地址:http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=1985
思路:枚举起始点O(n^3)肯定超时,O(n^2)都超时了,参考别人博客写的,用left和right数组保存比当前矩形高度高的最小下标和最大下标
待测代码:
<pre code_snippet_id="1647064" snippet_file_name="blog_20160414_1_9584762" name="code" class="cpp">#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <queue>
#include <stack>
#include <map>
#include <cstring>
#include <climits>
#include <cmath>
#include <cctype>
const int inf = 0x3f3f3f3f;//1061109567
typedef long long LL;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;
int a[100010],left1[100010],right1[100010];
int main()
{
int n;
while(scanf("%d",&n) && n)
{
/*for(int i=1; i<=n; i++)
{
scanf("%d",&a[i]);
left1[i] = i;
right1[i] = i;
}*/
for (int i = 1; i <= n; i++)
{
scanf("%d", &a[i]);
left1[i] = i;
right1[i] = i;
}
/*for(int i=1; i<=n; i++)
{
int temp = i;
while(temp-1>=1 && a[temp-1] >= a[i])
temp = left1[temp-1];
left1[i] = temp;
}
for(int i=n; i>=1; i--)
{
int temp = i;
while(temp + 1 <= n && a[temp+1] >= a[i])
temp = right1[temp+1];
right1[i] = temp;
}*/
int i,temp;
for (i = 1; i <= n; i++)
{
temp = i;
while (temp - 1 >= 1 && a[temp - 1] >= a[i])
temp = left1[temp - 1];
left1[i] = temp;
}
for (i = n; i >= 1; i--)
{
temp = i;
while (temp + 1 <= n && a[temp + 1] >= a[i])
temp = right1[temp + 1];
right1[i] = temp;
}
/*LL max1 = 0;
for(int i=1; i<=n; i++)
{
LL sum = (LL)a[i] * (LL)(right1[i]-left1[i]+1);
if(sum > max1)
max1 = sum;
}
printf("%I64d\n",max1);*/
long long max = -1, res;
for (i = 1; i <= n; i++)
{
res = (long long) a[i] * (long long) (right1[i] - left1[i] + 1);
if (res > max)
max = res;
}
printf("%I64d\n",max);
}
return 0;
}
超时代码:
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <queue>
#include <stack>
#include <map>
#include <cstring>
#include <climits>
#include <cmath>
#include <cctype>
const int inf = 0x3f3f3f3f;//1061109567
typedef long long LL;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;
int a[100010];
int main()
{
int n;
while(scanf("%d",&n) && n)
{
int max1 = 0;
for(int i=1; i<=n; i++)
{
scanf("%d",&a[i]);
int min1 = a[i];
for(int j=i; j>=1; j--)
{
min1 = min(min1,a[j]);
int sum = min1 * (i - j + 1);
if(sum > max1)
max1 = sum;
}
}
printf("%d\n",max1);
}
return 0;
}