牛客練習賽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;
	}
}