[HDU 5696] 區間的價值 (單調棧+RMQ)
阿新 • • 發佈:2019-01-02
連結
題意
給出一個長度為n(1 <= n <= 100000)的序列,區間權值為區間最大值乘以區間最小值,輸出長度為1到n的區間的最大權值。
題解
先利用單調棧算出以a[i]為最小值的最大區間延伸,設該區間寬度為L,則更新ans[L](這裡用到st表查詢該區間最大值),最後從寬度較大的區間重新整理寬度較小的區間的長度,即為答案。
為什麼是對的,可以證明一下,從結果考慮,寬度為k的區間的最優值一定可以被這種方式取到。
程式碼
#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
#include <iostream>
using namespace std;
typedef long long lint;
#define maxn (100010)
#define left lleft
#define right rright
#define log2 Log2
int a[maxn], left[maxn], right[maxn];
lint ans[maxn];
int log2[maxn], st[maxn][32];
void st_prepare(int arr[], int n)
{
log2[1 ] = 0;
for(int i = 2; i <= n; i++)
{
log2[i] = log2[i-1];
if((1 << log2[i] + 1) == i)
log2[i]++;
}
for(int i = n-1; i >= 0; i--)
{
st[i][0] = arr[i];
for(int j = 1; i + (1 << j) - 1 < n; j++)
{
st[i][j] = max(st[i][j-1 ], st[i+(1<<j-1)][j-1]);
}
}
}
int st_query(int l, int r)
{
int len = log2[r - l + 1];
return max(st[l][len], st[r - (1 << len) + 1][len]);
}
#define stack Stack
pair<int, int> stack[maxn];
int top;
void push_stack(int id, int w, int val[], int mx)
{
while(top > 0)
{
if(stack[top].first < w)
break;
top--;
}
if(!top) val[id] = mx;
else val[id] = stack[top].second;
stack[++top] = make_pair(w, id);
}
int main()
{
int n;
while(cin >> n)
{
for(int i = 1; i <= n; i++)
scanf("%d", &a[i]);
st_prepare(a + 1, n);
top = 0;
for(int i = 1; i <= n; i++)
{
push_stack(i, a[i], left, 0);
}
top = 0;
for(int i = n; i >= 1; i--)
{
push_stack(i, a[i], right, n + 1);
}
for(int i = 0; i <= n; i++) ans[i] = -1;
for(int i = 1, l, r, k; i <= n; i++)
{
l = left[i];
r = right[i] - 2;
k = st_query(l, r);
ans[r - l + 1] = max(ans[r - l + 1], (lint)k * a[i]);
}
for(int i = n - 1; i >= 1; i--)
ans[i] = max(ans[i], ans[i + 1]);
for(int i = 1; i <= n; i++)
printf("%I64d\n", ans[i]);
}
return 0;
}