BZOJ 3236: [Ahoi2013]作业

BZOJ 3236: [Ahoi2013]作业

标签(空格分隔): OI-BZOJ OI-莫队 OI-分块 OI-树状数组


Time Limit: 100 Sec
Memory Limit: 512 MB


Description

此处输入图片的描述

Input

此处输入图片的描述
Output

此处输入图片的描述

Sample Input

3 4

1 2 2

1 2 1 3

1 2 1 1

1 3 1 3

2 3 2 3

Sample Output

2 2

1 1

3 2

2 1
HINT

N=100000,M=1000000


Solution

莫队裸题
我偷了个懒,用树状数组统计出现次数,稍微优化了一下常数76s过去了。
复杂度\(O(n*\sqrt{n}*log_2n)\)
统计某个区间内的数字个数的时候可以使用分块,修改\(O(1)\)
查询的时候\(O(\sqrt{n})\)总复杂度\(O((n+m)*\sqrt{n})\)


Code

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<queue>
#include<set>
#include<map>
#include<bitset>
#include<vector>
using namespace std;
#define PA pair<int,int>
int read()
{
    int s=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){s=(s<<1)+(s<<3)+ch-'0';ch=getchar();}
    return s*f;
}
//smile please
struct que
{
    int l,r,a,b,h;
}q[1000005];
int n,m,L;
int t1[100005],t2[100005],a[100005],b[100005];
bool cmp(const que &a,const que &b)
{return a.l/L<b.l/L||(a.l/L==b.l/L&&a.r<b.r);}
void in(int x)
{
    int s=a[x];
    if((++b[s])==1)
      for(int x=s;x<=n;x+=x&-x)
         t2[x]++,t1[x]++;
    else
      for(int x=s;x<=n;x+=x&-x)
         t1[x]++;
}
void ou(int x)
{
    int s=a[x];
    if((--b[s])==0)
      for(int x=s;x<=n;x+=x&-x)
         t2[x]--,t1[x]--;
    else
      for(int x=s;x<=n;x+=x&-x)
         t1[x]--;
}
int an1[1000005],an2[1000005];
int main()
{
    //freopen(".in","r",stdin);
    //freopen(".out","w",stdout);
    n=read(),m=read();L=sqrt(n);
    for(int i=1;i<=n;i++)
        a[i]=read();
    for(int i=1;i<=m;i++)
        q[i]=(que){read(),read(),read(),read(),i};
    sort(&q[1],&q[m+1],cmp);
    int l=1,r=0;
    for(int i=1;i<=m;i++)
       {for(;l<q[i].l;ou(l++));
        for(;l>q[i].l;in(--l));
        for(;r<q[i].r;in(++r));
        for(;r>q[i].r;ou(r--));
        int H=q[i].h;
        for(int x=q[i].b;x;x-=x&-x)
            an1[H]+=t1[x],
            an2[H]+=t2[x];
        for(int x=q[i].a-1;x;x-=x&-x)
            an1[H]-=t1[x],
            an2[H]-=t2[x];
       }
    for(int i=1;i<=m;i++)
        printf("%d %d\n",an1[i],an2[i]);
    //fclose(stdin);
    //fclose(stdout);
    return 0;
}

posted on 2016-03-12 21:33  wuyuhan  阅读(...)  评论(...编辑  收藏

导航