Bzoj2639: 矩形计算

Time Limit: 20 Sec  Memory Limit: 256 MB
Submit: 178  Solved: 94

Description

　输入一个n*m的矩阵，矩阵的每一个元素都是一个整数，然后有q个询问，每次询问一个子矩阵的权值。矩阵的权值是这样定义的，对于一个整数x，如果它在该矩阵中出现了p次，那么它给该矩阵的权值就贡献p²。

Input

　　第一行两个整数n，m表示矩阵的规模。
　　接下来n行每行m个整数，表示这个矩阵的每个元素。
　　再下来一行一个整数q，表示询问个数。
　　接下来q行每行四个正整数x1，y1，x2，y2，询问以第x1行第y1列和第x2行第y2列的连线为对角线的子矩阵的权值。

Output

　　输出q行每行一个整数回答对应询问。

Sample Input

3 4
1 3 2 1
1 3 2 4
1 2 3 4
8
1 2 2 1
1 1 2 1
1 1 3 4
1 1 1 1
2 2 3 3
3 4 2 2
1 3 3 1
2 4 3 4

Sample Output

8
4
38
1
8
12
27
4

HINT

对于全部数据

　　1<=n,m<=200

　　q<=100000

　　|矩阵元素大小|<=2*10⁹

Source

 

分治 分块 二维莫队

二维的莫队算法

分块的时候按顶点坐标分块，belong= $ (x1/\sqrt {n} )* \sqrt m + (y1/\sqrt m)$

范围转移的时候稍有点麻烦

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<cmath>
#define LL long long
using namespace std;
const int mxn=310;
int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0' || ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0' && ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
inline void write(int x){
    if(x<0) putchar('-'),x=-x;
    if(x>9) write(x/10);
    putchar(x%10+'0');
    return;
}
int n,m,Q;
int mp[mxn][mxn];
int bas[mxn*mxn],cc=0;
int cnt[mxn*mxn],hs[mxn*mxn];
struct query{
    int x1,y1,x2,y2;
    int m,id;
    bool operator < (const query &b)const{
        return (m==b.m && ((x1==b.x1 && y1<b.y1) || x1<b.x1 )) || m<b.m;
    }
}q[100010];
int nowans=0,ans[100010];
inline void add(int x){
    nowans+=cnt[x]*2+1;  ++cnt[x]; return;
}
inline void del(int x){
    nowans-=cnt[x]*2-1;  --cnt[x]; return;
}
void solve(){
    int l=1,r=0,u=1,d=0;//初始位置 
    int i,j;
    for(i=1;i<=Q;i++){
        while(r<q[i].y2){++r;for(j=u;j<=d;j++)add(mp[j][r]);}
        while(l>q[i].y1){--l;for(j=u;j<=d;j++)add(mp[j][l]);}
        while(d<q[i].x2){++d;for(j=l;j<=r;j++)add(mp[d][j]);}
        while(u>q[i].x1){--u;for(j=l;j<=r;j++)add(mp[u][j]);}
        //
        //
        while(r>q[i].y2){for(j=u;j<=d;j++)del(mp[j][r]);--r;}
        while(l<q[i].y1){for(j=u;j<=d;j++)del(mp[j][l]);++l;}
        while(d>q[i].x2){for(j=l;j<=r;j++)del(mp[d][j]);--d;}
        while(u<q[i].x1){for(j=l;j<=r;j++)del(mp[u][j]);++u;}
        ans[q[i].id]=nowans;
    }
    return;
}
int main(){
    int i,j;
    n=read();m=read();
    for(i=1;i<=n;i++)
        for(j=1;j<=m;j++){
            mp[i][j]=read();
            bas[++cc]=mp[i][j];
        }
    sort(bas+1,bas+cc+1);
    int ed=unique(bas+1,bas+cc+1)-bas-1;
    for(i=1;i<=n;i++)
        for(j=1;j<=m;j++){
            int tmp=lower_bound(bas+1,bas+ed+1,mp[i][j])-bas;
            mp[i][j]=tmp;
        }//离散化 
    Q=read();
    int sn=sqrt(n),sm=sqrt(m);
    for(i=1;i<=Q;i++){
        q[i].x1=read(); q[i].y1=read();
        q[i].x2=read(); q[i].y2=read();
        if(q[i].x1>q[i].x2)swap(q[i].x1,q[i].x2);
        if(q[i].y1>q[i].y2)swap(q[i].y1,q[i].y2);
        q[i].m=(q[i].x1/sn)*sm+(q[i].y1/sm);
        q[i].id=i;
    }
    sort(q+1,q+Q+1);
    solve();
    for(i=1;i<=Q;i++){
        write(ans[i]);
        puts("");
    }
    return 0;
}