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

Link

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

Description

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

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

Output

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

Sample Input

3 3 3 8
3 2 3
2 3 1

Sample Output

10

Solution

$官方题解:设su{m}_a为a的前缀和，su{m}_b为b的前缀和。则以(x1,y1)为左上角，以(x2,y2)为右下角的矩阵的权值为(su{m}_a[x2]-su{m}_a[x1-1])\ast (su{m}_b[y2]-su{m}_b[y1-1])，其中要满足x1<=x2,y1<=y2。x1,x2的选择与y1,y2的选择相互独立。可以预处理所有y1,y2的组合，然后排序。接下来枚举x1,x2，对y二分即可。复杂度$$n^{2\ast log(m^2)}$

About

此题即求矩阵和介于[L,R]之间的子矩阵的个数,那么可以发散思维,如何求子序列和介于[L,R]之间的子区间的个数,或者复杂度很大时,怎么处理

Code

#include<iostream>
#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<map>
#include<unordered_map>
using namespace std;

#define maxn 200090
#define inf 0x3f3f3f3f
#define PI acos(-1)
typedef long long ll;
struct edge{
    int u,v,next,w; 
}e[2];

int head[1],cnt;
void add(int x,int y,int w){
    e[cnt].u=x;
    e[cnt].v=y;
    e[cnt].w=w;
    e[cnt].next=head[x];
    head[x]=cnt++;
    e[cnt].u=y;
    e[cnt].v=x;
    e[cnt].w=w;
    e[cnt].next=head[y];
    head[y]=cnt++;
}

ll read(){
    ll x=0,f=1;
    char c=getchar();
    while(c<'0'||c>'9'){
        if(c=='-')f=-1;
        c=getchar(); 
    }
    while(c>='0'&&c<='9'){
        x=x*10+c-'0';
        c=getchar();
    }
    return x*f;
}
ll n,m,L,R,a[1004],b[1006],cot,suma[1006],sumb[1006],sum_b[1000060];

ll find_l(ll x,ll l,ll r){
    ll res=0;
    while(r>=l){
        ll mid=(l+r)>>1;
        if(x*sum_b[mid]>=L){
            res=mid,r=mid-1;
        }
        else l=mid+1;
    }
    return res;
}   
ll find_r(ll x,ll l,ll r){
    ll res=0;
    while(r>=l){
        ll mid=(l+r)>>1;
        if(x*sum_b[mid]<=R)res=mid,l=mid+1;
        else r=mid-1;
    }
    return res;
}
int main(){
    n=read(),m=read(),L=read(),R=read();
    for(int i=1;i<=n;i++)
        a[i]=read(),suma[i]=suma[i-1]+a[i];
    for(int i=1;i<=m;i++)
        b[i]=read(),sumb[i]=sumb[i-1]+b[i];
    for(int i=1;i<=m;i++)
        for(int j=1;j<=i;j++)
            sum_b[++cot]=sumb[i]-sumb[j-1];
    sort(sum_b+1,sum_b+1+cot);
    ll x,l,r,res=0;;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=i;j++){
            x=suma[i]-suma[j-1];
            l=find_l(x,1,cot),r=find_r(x,1,cot);
            res+=(r-l+1);

        }
    cout<<res<<endl;
}

我们坚持一件事情，并不是因为这样做了会有效果，而是坚信，这样做是对的。
——哈维尔