牛客练习赛46 B 华华送奕奕小礼物

牛客练习赛46 B 华华送奕奕小礼物

https://ac.nowcoder.com/acm/contest/894/B

题意： 给两个数组，其中a数组长度为n，b数组长度为m，c是n行m列的矩阵，且c[i][j]=a[i]*b[j]。求有多少子矩阵，满足子矩阵的和>=L&&子矩阵的和<=R。

可以先确定行，一共有n^2种，再确定列的时候二分所有可能的行即可。

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pai acos(-1)
const LL mod = 1000000007;
const int MXN = 1e6+5;
LL sa[1005],sb[1005];
vector<LL>v;
int main(){
    LL n,m,L,R;
    cin>>n>>m>>L>>R;int sc;
    for(int i=1;i<=n;i++){scanf("%d",&sc);sa[i]=(LL)sc+sa[i-1];}
    for(int i=1;i<=m;i++){scanf("%d",&sc);sb[i]=(LL)sc+sb[i-1];}
    for(int i=1;i<=n;i++)for(int j=i;j<=n;j++)v.push_back(sa[j]-sa[i-1]);
    sort(v.begin(),v.end());
    LL ans=0;
    for(int i=1;i<=m;i++){
        for(int j=i;j<=m;j++){
            LL nowsb=sb[j]-sb[i-1];
            LL l=L/nowsb+(L%nowsb!=0);
            LL r=R/nowsb;
            LL idl=lower_bound(v.begin(),v.end(),l)-v.begin();
            LL idr=upper_bound(v.begin(),v.end(),r)-v.begin()-1;
            if(idr>=idl) ans+=(idr-idl+1);
        }
    }
    printf("%lld",ans);
    return 0;
}