小美的仓库整理（美团2020秋招笔试题）

小美是美团仓库的管理员，她会根据单据的要求按顺序取出仓库中的货物，每取出一件货物后会把剩余货物重新堆放，使得自己方便查找。已知货物入库的时候是按顺序堆放在一起的。如果小美取出其中一件货物，则会把货物所在的一堆物品以取出的货物为界分成两堆，这样可以保证货物局部的顺序不变。
已知货物最初是按 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]);
    }
}