codeforces 985D(二分+前缀和)

传送门

题意：Alice每天可以制造Ai体积的雪，但是由于每天温度为Ti，一堆雪在一天会融化Ti体积，直到这对雪全部融化后才会停止

思路：首先将所有的体积进行前缀和加起来得到一个温度的前缀和数组，然后对于每一个Ai进行二分查找，找到Ai最后融化为0的地方，并用一个差分数组记录在哪些地方是可以融化Ti(每一个Ai对应的Ti)，用一个sum数组记录所有在i处融化但是融化体积不是Ti的和,最后将差分数组进行前缀和计算出sum[i]+d[i]*b[i]即为结果。

import java.io.*;
import java.util.*;


public class Main {
	static Scanner scanner=new Scanner(System.in);
	public static void main(String[] args) {
		int n=scanner.nextInt();
		long a[]=new long [n+10];
		long b[]=new long [n+10];
		long c[]=new long [n+10];
		int d[]=new int [n+10];
		long ans[]=new long [n+10];
		long sum[]=new long [n+10];
		for(int i=1;i<=n;i++)a[i]=scanner.nextLong();
		for(int i=1;i<=n;i++) {
			b[i]=scanner.nextLong();
			c[i]=c[i-1]+b[i];
		}
		for(int i=1;i<=n;i++) {
			int l=i,r=n+1;
			while(l<r) {
				int mid=(l+r)/2;
				if(a[i]>c[mid]-c[i-1]) {
					l=mid+1;
				}else {
					r=mid;
				}
			}
			d[i]++;d[l]--;
			sum[l]+=a[i]-c[l-1]+c[i-1];
		}
		//System.out.println(Arrays.toString(d));
		//System.out.println(Arrays.toString(sum));
		for(int i=1;i<=n;i++) {
			d[i]+=d[i-1];
		}
		for(int i=1;i<=n;i++) {
			System.out.print(d[i]*b[i]+sum[i]+" ");
		}
	}
}