bzoj 5016: [Snoi2017]一个简单的询问

题意：

求$\sum^{max}_{x=0}get(l1,r1,x)*get(l2,r2,x)$

题解：

分块+线段树。
一开始傻傻的想分颜色块，然后被ozy大佬D飞了。
按位置分块。
$f[i][j]$表示第i块，和1~j的答案。
那么对于第i块与l，r的答案，就是$f[i][r]-f[i][l-1]$
然后把块加起来，其他暴力扫，打个什么动态开点线段树查询就可以了。
code：

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#define LL long long
using namespace std;
struct trnode{
    int lc,rc,c;
}tr[1000010];int tot=0,root[50010];
struct color{
    int c[50010];
}g[250];
struct node{
    int a,b,num;
}a[50010];
int n,pre[50010],last[250];
LL f[250][50010];
bool cmp(node a,node b){return a.b<b.b;}
void update(int &x,int l,int r,int k)
{
    if(!x) x=++tot;
    if(l==r){tr[x].c++;return;}
    int mid=(l+r)/2;
    if(k<=mid) update(tr[x].lc,l,mid,k);
    else update(tr[x].rc,mid+1,r,k);
    tr[x].c=tr[tr[x].lc].c+tr[tr[x].rc].c;
}
int findans(int x,int l,int r,int fl,int fr)
{
    if(!x) return 0;
    if(fl==l&&fr==r) return tr[x].c;
    int mid=(l+r)/2;
    if(fr<=mid) return findans(tr[x].lc,l,mid,fl,fr);
    if(fl>mid) return findans(tr[x].rc,mid+1,r,fl,fr);
    return findans(tr[x].lc,l,mid,fl,mid)+findans(tr[x].rc,mid+1,r,mid+1,fr);
}
int check(int x,int l,int r){return findans(root[x],1,n,l,r);}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&a[i].b),a[i].num=i;
    sort(a+1,a+n+1,cmp);
    int num=0;
    for(int i=1;i<=n;i++)
    {
        if(i==1||a[i].b!=a[i-1].b) num++;
        a[a[i].num].a=num;
    }
    for(int i=1;i<=n;i++) update(root[a[i].a],1,n,i);
    int len=sqrt(n);
    for(int i=1;i<=n;i++)
    {
        pre[i]=(i/len)+1;
        if(pre[i]!=pre[i-1]) last[pre[i-1]]=i-1;
    }
    last[pre[n]]=n;
    for(int i=1;i<=n;i++)
    {
        g[pre[i]].c[a[i].a]++;
        if(i==n||pre[i]!=pre[i+1])
            for(int j=1;j<=n;j++) f[pre[i]][j]=f[pre[i]][j-1]+(LL)g[pre[i]].c[a[j].a];
    }
    int m;scanf("%d",&m);
    while(m--)
    {
        int x,y,l,r;scanf("%d %d %d %d",&x,&y,&l,&r);
        int i=x;
        LL ans=0;
        while(i<=y)
        {
            if((i==1||pre[i]!=pre[i-1])&&last[pre[i]]<=y) ans=ans+(f[pre[i]][r]-f[pre[i]][l-1]),i=last[pre[i]];
            else ans+=check(a[i].a,l,r);
            i++;
        }
        printf("%lld\n",ans);
    }
}