牛客练习赛46（Java版）

B.华华送奕奕小礼物。
题目链接：https://ac.nowcoder.com/acm/contest/894/B

这方面的题做的少，一开始想的思路会超时，参考了题解
设sum_a为a的前缀和，sum_b为b的前缀和。则以(x1,y1)为左上角，以(x2,y2)为右下角的矩阵的权值为(sum_a[x2]-sum_a[x1-1])*(sum_b[y2]-sum_b[y1-1])，其中要满足x1<=x2,y1<=y2。x1,x2的选择与y1,y2的选择相互独立。可以预处理所有y1,y2的组合，然后排序。接下来枚举x1,x2，对y二分即可。复杂度n^2*log(m2)

话说最开始并没有理解这个题解，但晚上睡前在想这个事，结果第二天早上起来发现突然懂了，果然睡前适合思考，梦中做题，qaq

ac代码：

import java.util.Arrays;
import java.util.Scanner;

public class Main {
	public static void main(String[] args) {
		Scanner sc=new Scanner(System.in);
		int n=sc.nextInt();
		int m=sc.nextInt();
		long L=sc.nextLong();
		long R=sc.nextLong();
		long a[]=new long[n+1];//行的前缀和
		long b[]=new long[m+1];//列的前缀和
		for(int i=1;i<=n;i++){
			int x=sc.nextInt();
			a[i]=a[i-1]+x;
		}
		for(int i=1;i<=m;i++){
			int x=sc.nextInt();
			b[i]=b[i-1]+x;
		}

		long c[]=new long[n*m+1];//存储y值的所有组合
		int index=1;
		for(int i=1;i<=m;i++){
			for(int j=i;j<=m;j++){
				c[index++]=b[j]-b[i-1];//预处理y1，y2的所有情况
			}
		}
		Arrays.sort(c,1,index);//对y进行排序
		long ans=0;		
		
		for(int i=1;i<=n;i++){
			for(int j=i;j<=n;j++){
				long sum=a[j]-a[i-1];//枚举x1，x2
				int res1=upper(c,0,index-1,R/sum);				
				int res2=lower(c,1,index,L%sum==0?L/sum:L/sum+1);
		
				if(res1>=1&&res1<=index-1&&res2>=1&&res2<=index-1)
					ans+=(res1-res2+1);		
			}
		}
		System.out.println(ans);
		sc.close();
	}
	//二分求最小后继
	static int lower(long s[],int l,int r,long k){
		while(l<r){
			int mid=(l+r)>>1;
			if(s[mid]>=k)
				r=mid;
			else
				l=mid+1;
		}
		return l;
	}
	//二分求最大前驱
	static int upper(long s[],int l,int r,long k){
		while(l<r){
			int mid=(l+r+1)>>1;
			if(s[mid]<=k)
				l=mid;
			else
				r=mid-1;
		}
		return l;
	}
}