此时己是凌晨两点,刚刚做了Codeforces的小A掏出了英语试卷。英语作业其实不算多,一个小时刚好可以做完。然后是一个小时可以做完的数学作业,接下来是分别都是一个小时可以做完的化学,物理,语文......小A压力巨大。
这是小A碰见了一道非常恶心的数学题,给定了一个长度为n的数列和若干个询问,每个询问是关于数列的区间表示数列的第1个数到第r个数),首先你要统计该区间内大于等于a,小于等于b的数的个数,其次是所有大于等于a,小于等于b的,且在该区间中出现过的数值的个数。
小A望着那数万的数据规模几乎绝望,只能向大神您求救,请您帮帮他吧。
输入输出格式
输入格式:
第一行n,m
接下来n个数表示数列
接下来m行,每行四个数l,r,a,b
N<=100000,M<=100000
输出格式:
输出m行,分别对应每个询问,输出两个数,分别为在1到i?这段区间中大小在[a,b]中的数的个数,以及大于等于a,小于等于b的,且在该区间中出现过的数值的个数(具体可以参考样例)。
输入输出样例
输入样例#1:
3 4
1 2 2
1 2 1 3
1 2 1 1
1 3 1 3
2 3 2 3
输出样例#1:
2 2
1 1
3 2
2 1
Sol:莫队加树状数组
/* 莫队+树状数组 用莫队解决[l,r]的问题 两个树状数组,以权值为下标,一个记录权值为i的是否出现, 一个记录权值为i的出现了几次 */ #include<cstdio> #include<cmath> #include<algorithm> #define N 100001 using namespace std; int n,m,size,l=1,r,tmp; int key[N],sum[N],c[2][N]; int ans[N*10][2]; struct node { int l,r,a,b; int bl,id; int ans1,ans2; }e[N*10]; bool cmp(node p,node q) { if(p.bl!=q.bl) return p.bl<q.bl; return p.r<q.r; } int lowbit(int x) { return x&(-x); } void change(int k,int w,int h) //h为数组下标,为0时用于统计某个数字是否出现过 //为1时统计数字出现的次数 { while(k<=n) { c[h][k]+=w; k+=lowbit(k);} } int query(int k,int h) { int tot=0; while(k) { tot+=c[h][k]; k-=lowbit(k);} return tot; } void update(int k,int w) { sum[key[k]]+=w; change(key[k],w,1); if(w==1&&sum[key[k]]==1) change(key[k],1,0); else if(w==-1&&!sum[key[k]]) change(key[k],-1,0); } int main() { scanf("%d%d",&n,&m); size=sqrt(n); for(int i=1;i<=n;i++) scanf("%d",&key[i]); for(int i=1;i<=m;i++) { scanf("%d%d%d%d",&e[i].l,&e[i].r,&e[i].a,&e[i].b); e[i].bl=(e[i].l-1)/size+1; e[i].id=i; } sort(e+1,e+m+1,cmp); for(int i=1;i<=m;i++) { while(e[i].l<l) update(--l,1); while(e[i].l>l) update(l++,-1); while(e[i].r<r) update(r--,-1); while(e[i].r>r) update(++r,1); e[i].ans1=query(e[i].b,0)-query(e[i].a-1,0); e[i].ans2=query(e[i].b,1)-query(e[i].a-1,1); } for(int i=1;i<=m;i++) ans[e[i].id][0]=e[i].ans2 , //出现了多少个数字在规定范围内 ans[e[i].id][1]=e[i].ans1; //出现了多少种数字在规定范围内 for(int i=1;i<=m;i++) printf("%d %d ",ans[i][0],ans[i][1]); }