小美是美团仓库的管理员,她会根据单据的要求按顺序取出仓库中的货物,每取出一件货物后会把剩余货物重新堆放,使得自己方便查找。已知货物入库的时候是按顺序堆放在一起的。如果小美取出其中一件货物,则会把货物所在的一堆物品以取出的货物为界分成两堆,这样可以保证货物局部的顺序不变。
已知货物最初是按 1~n 的顺序堆放的,每件货物的重量为 w[i] ,小美会根据单据依次不放回的取出货物。请问根据上述操作,小美每取出一件货物之后,重量和最大的一堆货物重量是多少?
格式:
输入:
- 输入第一行包含一个正整数 n ,表示货物的数量。
- 输入第二行包含 n 个正整数,表示 1~n 号货物的重量 w[i] 。
- 输入第三行有 n 个数,表示小美按顺序取出的货物的编号,也就是一个 1~n 的全排列。
输出:
- 输出包含 n 行,每行一个整数,表示每取出一件货物以后,对于重量和最大的一堆货物,其重量和为多少。
示例:
输入:
5
3 2 4 4 5
4 3 5 2 1
输出:
9
5
5
3
0
解释:
原本的状态是 {{3,2,4,4,5}} ,取出 4 号货物后,得到 {{3,2,4},{5}} ,第一堆货物的和是 9 ,然后取出 3 号货物得到 {{3,2}{5}} ,此时第一堆和第二堆的和都是 5 ,以此类推。
提示:
1 <= n,m <= 50000
1 <= w[i] <= 100
思路:我们可以倒着执行操作,此时删除操作变成添加操作,并维护两个数组分别表示一个区间的左右断点,我们知道一个区间的中间点事没必要管的,因此通过更新这两个数组我们就能在O(1) 的复杂度内完成区间的合并操作!
public class Solution {
public static void main(String[] args) {
int n;
Scanner in = new Scanner(System.in);
n = in.nextInt();
int[] l = new int[n];
int[] r = new int[n];
int[] arr = new int[n];
int[] opt = new int[n];
int[] ans = new int[n];
int[] sum = new int[n];
boolean[] used = new boolean[n];
for (int i = 0; i < n; i++) {
arr[i] = in.nextInt();
if (i > 0) sum[i] += sum[i - 1];
sum[i] += arr[i];
}
for (int i = 0; i < n; i++) {
opt[i] = in.nextInt();
l[i] = r[i] = i;
}
PriorityQueue<Integer> q = new PriorityQueue<>((a, b) -> b - a);
q.add(arr[opt[n - 1] - 1]);
used[opt[n - 1] - 1] = true;
for (int i = n - 2; i >= 0; i--) {
int index = opt[i] - 1, ls = l[index], rs = r[index];
if (index > 0 && used[index - 1])
ls = l[index - 1];
if (index + 1 < n && used[index + 1])
rs = r[index + 1];
used[index] = true;
r[ls] = rs;
l[rs] = ls;
int now = sum[rs];
if (ls > 0) now -= sum[ls - 1];
ans[i] = q.peek();
q.add(now);
}
for (int i = 0; i < n; i++)
System.out.println(ans[i]);
}
}