COGS：1822. [AHOI2013]作业

1822. [AHOI 2013] 作业

★★★   输入文件：ahoi2013_homework.in   输出文件：ahoi2013_homework.out   简单对比
时间限制：20 s   内存限制：512 MB

【题目描述】

【输入格式】

【输出格式】

【样例输入】

3 4

1 2 2

1 2 1 3

1 2 1 1

1 3 1 3

2 3 2 3

【样例输出】

2 2

1 1

3 2

2 1

【提示】

N=100000,M=1000000

数据极弱（和BZOJ相比），请放心A

【来源】

BZOJ 3236

分析

莫队，用树状数组来维护求值，一个求出现不同数的个数，另一个满足的数字个数。复杂度$O(n\sqrt n log_2n)$

另一种做法：由于这些数都是小于等于n的，所以对权值进行分块，修改$O(1)$，查询$O(\sqrt n)$，总复杂度$O(m \sqrt n)$

交了几次，最后一个点老是TLE，然后想尽办法，读入优化，inline内联函数，最后写上了输出优化。。。还是TLE，之后才发现ans数组开小了，直接开了MAXN，QAQ

代码

 #include<cstdio>
 #include<algorithm>
 #include<cmath>
 using namespace std;

 const int MAXN = ;
 struct Que{
     int l,r,a,b,block,id;
     bool operator < (const Que &a) const
     {
         if (block==a.block) return r < a.r;
         return block < a.block;
     }
 }q[];
 int a[MAXN],cnt[MAXN],tr[][MAXN],pr[];
 int ans1[],ans2[];
 int n,m,pos,len;

 inline int read()
 {
     int x=,f=;char ch=getchar();
     while(ch<'' || ch>''){if(ch=='-')f=-;ch=getchar();}
     while(ch>=''&& ch<=''){x=x*+ch-'';ch=getchar();}
     return x*f;
 }
 inline int lowbit(int x)
 {
     return x & (-x);
 }
 inline void change(int x,int c,int t)
 {
     while (x<=n)
     {
         tr[t][x] += c;
         x += lowbit(x);
     }
 }
 inline int query(int x,int t)
 {
     int res = ;
     while (x)
     {
         res += tr[t][x];
         x -= lowbit(x);
     }
     return res;
 }
 inline void update(int x,int c)
 {
     if (cnt[x]==) change(x,,);
     cnt[x] += c;
     if (cnt[x]==) change(x,-,);
     change(x,c,);
 }
 inline void solve()
 {
     int l = , r = ;
     for (int i=; i<=m; ++i)
     {
         for (; l>q[i].l; ) update(a[--l], );
         for (; r<q[i].r; ) update(a[++r], );
         for (; l<q[i].l; ) update(a[l++], -);
         for (; r>q[i].r; ) update(a[r--], -);
         ans1[q[i].id] = query(q[i].b, ) - query(q[i].a-, );
         ans2[q[i].id] = query(q[i].b, ) - query(q[i].a-, );
     }
 }
 inline void print(int x)
 {
     while (x)
         pr[++len] = x%, x/=;
     if (!len) putchar('');
     while (len)
         putchar(pr[len--]+'');
 }
 int main()
 {
     freopen("ahoi2013_homework.in","r",stdin);
     freopen("ahoi2013_homework.out","w",stdout);
     n = read(); m = read();
     pos = (int)sqrt(n);
     for (int i=; i<=n; ++i)
         a[i] = read();
     for (int i=; i<=m; ++i)
     {
         q[i].l = read(); q[i].r = read();
         q[i].a = read(); q[i].b = read();
         q[i].block = (q[i].l-)/pos+;
         q[i].id = i;
     }
     sort(q+,q+m+);
     solve();
     for (int i=; i<=m; ++i)
     {
         print(ans1[i]),putchar(' ');
         print(ans2[i]),putchar('\n');
         //printf("%d %d\n",ans1[i],ans2[i]);
     }
     return ;
 }

备注：luogu提交地址