【BZOJ2639】矩形计算（二维普通莫队）

题意：输入一个n*m的矩阵，矩阵的每一个元素都是一个整数，然后有q个询问，每次询问一个子矩阵的权值。

矩阵的权值是这样定义的，对于一个整数x，如果它在该矩阵中出现了p次，那么它给该矩阵的权值就贡献p^2。

n,m<=200，m<=1e5，abs(a[i][j])<=2e9

思路：学习资料见https://www.cnblogs.com/ouuan/p/2DMoDui.html

和一维普通莫队差不多，先扩大区间再缩小保证不会出错

这个块长很有特色，和n，m，q都有关

map离散化真的不靠谱，换预处理离散化即可

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned int uint;
typedef unsigned long long ull;
typedef pair<int,int> PII;
typedef pair<ll,ll> Pll;
typedef vector<int> VI;
typedef vector<PII> VII;
//typedef pair<ll,ll>P;
#define N  210
#define M  200010
#define fi first
#define se second
#define MP make_pair
#define pb push_back
#define pi acos(-1)
#define mem(a,b) memset(a,b,sizeof(a))
#define rep(i,a,b) for(int i=(int)a;i<=(int)b;i++)
#define per(i,a,b) for(int i=(int)a;i>=(int)b;i--)
#define lowbit(x) x&(-x)
#define Rand (rand()*(1<<16)+rand())
#define id(x) ((x)<=B?(x):m-n/(x)+1)
#define ls p<<1
#define rs p<<1|1

const int MOD=1e9+7,inv2=(MOD+1)/2;
      double eps=1e-4;
      int INF=1e9;
      int inf=0x7fffffff;
      int dx[4]={-1,1,0,0};
      int dy[4]={0,0,-1,1};

struct Q
{
    int x1,y1,x2,y2,id;
}t[M];

int cnt[N*N],c[N*N];
int ans[M],pos[250],a[N][N],sum,L1,L2,R1,R2;

int read()
{
   int v=0,f=1;
   char c=getchar();
   while(c<48||57<c) {if(c=='-') f=-1; c=getchar();}
   while(48<=c&&c<=57) v=(v<<3)+v+v+c-48,c=getchar();
   return v*f;
}

bool cmp(Q a,Q b)
{
    return pos[a.x1]<pos[b.x1]
         ||pos[a.x1]==pos[b.x1]&&pos[a.y1]<pos[b.y1]
         ||pos[a.x1]==pos[b.x1]&&pos[a.y1]==pos[b.y1]&&pos[a.x2]<pos[b.x2]
         ||pos[a.x1]==pos[b.x1]&&pos[a.y1]==pos[b.y1]&&pos[a.x2]==pos[b.x2]&&pos[a.y2]<pos[b.y2];
}

void add(int op,int x,int y,int z)
{
    if(op==1)
    {
        if(!x) return;
        rep(j,y,z)
         if(j)
         {
            sum+=(cnt[a[x][j]]<<1)+1;
            cnt[a[x][j]]++;
         }
    }
     else
     {
         if(!z) return;
         rep(i,x,y)
          if(i)
          {
              sum+=(cnt[a[i][z]]<<1)+1;
              cnt[a[i][z]]++;
          }
     }
}

void del(int op,int x,int y,int z)
{
    if(op==1)
    {
        if(!x) return;
        rep(j,y,z)
         if(j)
         {
             sum=sum-(cnt[a[x][j]]<<1)+1;
             cnt[a[x][j]]--;
         }
    }
     else
     {
         if(!z) return;
         rep(i,x,y)
          if(i)
          {
              sum=sum-(cnt[a[i][z]]<<1)+1;
              cnt[a[i][z]]--;
          }
     }
}

int main()
{
    int n=read(),m=read();
    int tot=0;
    rep(i,1,n)
     rep(j,1,m)
     {
         a[i][j]=read();
         c[++tot]=a[i][j];
     }
    sort(c+1,c+tot+1);
    int k=unique(c+1,c+tot+1)-c-1;
    rep(i,1,n)
     rep(j,1,m) a[i][j]=lower_bound(c+1,c+k+1,a[i][j])-c;
    int q=read();
    int S=pow(n*m,0.5)/pow(q,0.25)+1;
    rep(i,1,200) pos[i]=(i-1)/S;
    rep(i,1,q)
    {
        int x1=read(),y1=read(),x2=read(),y2=read();
        t[i].x1=min(x1,x2);
        t[i].x2=max(x1,x2);
        t[i].y1=min(y1,y2);
        t[i].y2=max(y1,y2);
        t[i].id=i;
    }
    sort(t+1,t+q+1,cmp);
    L1=0,L2=0,R1=0,R2=0;
    sum=0;
    rep(i,1,q)
    {
        while(L1>t[i].x1)
        {
            L1--;
            add(1,L1,R1,R2);
        }
        while(L2<t[i].x2)
        {
            L2++;
            add(1,L2,R1,R2);
        }
        while(R1>t[i].y1)
        {
            R1--;
            add(2,L1,L2,R1);
        }
        while(R2<t[i].y2)
        {
            R2++;
            add(2,L1,L2,R2);
        }

        while(L1<t[i].x1)
        {
            del(1,L1,R1,R2);
            L1++;
        }
        while(L2>t[i].x2)
        {
            del(1,L2,R1,R2);
            L2--;
        }
        while(R1<t[i].y1)
        {
            del(2,L1,L2,R1);
            R1++;
        }
        while(R2>t[i].y2)
        {
            del(2,L1,L2,R2);
            R2--;
        }
        ans[t[i].id]=sum;
    }
    rep(i,1,q) printf("%d\n",ans[i]);
    return 0;
}