子矩阵权值和

牛客 华华送奕奕小礼物

链接：https://ac.nowcoder.com/acm/contest/894/B
来源：牛客网
 

题目描述

华华用数组a和数组b合成了矩阵c。其中a数组长度为n，b数组长度为m，c是n行m列的矩阵，且c[i][j]=a[i]*b[j]。定义矩阵的权值为矩阵中所有元素的和。然后他想把矩阵送给奕奕。然而他怕奕奕不喜欢。若矩阵的权值小于L,奕奕会讨厌它，因为奕奕不喜欢太小的数字。若矩阵的权值大于R，奕奕会生气因为奕奕不认识比R大的数字。所以奕奕只喜欢权值大于等于L并且小于等于R的矩阵。还好华华学过acm，他马上想到可以送奕奕一个子矩阵，并且他立马写程序从c矩阵中找出了所有奕奕喜欢的子矩阵。你只需要帮他算算这样的子矩阵有多少个即可。

输入描述:

第一行输入n,m,L,R。
第二行n个数表示a数组
第三行m个数表示b数组
1<=n,m<=1000,1<=L<=R<=1e18
1<=a[i],b[i]<=1e6

输出描述:

输出一个数表示子矩阵的个数

示例1

输入

复制

3 3 3 8
3 2 3
2 3 1

输出

复制

10

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll a[1005], b[1005], c[1000005];
int main()
{
    int n, m, tot = 0, tmp;
    ll l, r, ans = 0, x;
    scanf("%d%d%lld%lld",&n,&m,&l,&r);
    for(int i = 1; i <= n; i++){
        scanf("%lld",&x);
        a[i] = a[i-1]+x;
    }
    for(int i = 1; i <= m; i++){
        scanf("%lld",&x);
        b[i] = b[i-1]+x;
    }
    for(int i = 1; i <= m; i++){
        for(int j = i; j <= m; j++){
            c[++tot] = b[j]-b[i-1];
        }
    }
    sort(c+1,c+tot+1);
    for(int i = 1; i <= n; i++){
        for(int j = i; j <= n; j++){
            x = a[j]-a[i-1];
            tmp = upper_bound(c+1,c+tot+1,r/x)-upper_bound(c+1,c+tot+1,(l-1)/x);
            ans += tmp;
        }
    }
    printf("%lld\n",ans);
    return 0;
}