P4396 [AHOI2013]作业 cdq分治

传送门

题意：

给你一个长度为$n$的数列$a$，有$q$个询问，每次询问$[l,r]$中值域在$[a,b]$中的数出现的次数和在$[a,b]$中出现过的数值个数。

$n\le 1e5,1\le a\le 1e5$

思路：

这个题可以值域分块来写，经典模型了。

这里介绍$cdq$的写法，考虑分的三维是那三维。

比较容易想到前两维是$l\le pos\le r,a\le val\le b$，第二个答案的统计是难点，考虑如何统计第二个答案。记$pre[val]$代表$val$之前出现的位置，我们如果能知道有多少$0\le pre[val]\le l-1$是不是就可以了，因为$[a,b]$中的每个值的$pre$在$[0,l-1]$最多出现一次，所以我们第三维就是这个？

并不是，不能忽略的一个很重要的问题就是$cdq$需要保证先进行修改，再进行查询，所以第三维应该保证修改在查询之前，与$pre[val]$是多少无关，因为第一第二维已经定序。

这里给一组$hack$样例

2 1

2 2

1 2 2 2

当然这是对于不离散化来说的，如果你离散化了，那么就另当别论了。

复杂度$O(nlo{g}^{2}n)$

// Problem: P4396 [AHOI2013]作业
// Contest: Luogu
// URL: https://www.luogu.com.cn/problem/P4396
// Memory Limit: 125 MB
// Time Limit: 1000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

//#pragma GCC optimize("Ofast,no-stack-protector,unroll-loops,fast-math")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4.1,sse4.2,avx,avx2,popcnt,tune=native")
//#pragma GCC optimize(2)
#include<cstdio>
#include<iostream>
#include<string>
#include<cstring>
#include<map>
#include<cmath>
#include<cctype>
#include<vector>
#include<set>
#include<queue>
#include<algorithm>
#include<sstream>
#include<ctime>
#include<cstdlib>
#include<random>
#include<cassert>
#define X first
#define Y second
#define L (u<<1)
#define R (u<<1|1)
#define pb push_back
#define mk make_pair
#define Mid ((tr[u].l+tr[u].r)>>1)
#define Len(u) (tr[u].r-tr[u].l+1)
#define random(a,b) ((a)+rand()%((b)-(a)+1))
#define db puts("---")
#define lowbit(x) ((x)&(-x))
using namespace std;

//void rd_cre() { freopen("d://dp//data.txt","w",stdout); srand(time(NULL)); }
//void rd_ac() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//AC.txt","w",stdout); }
//void rd_wa() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//WA.txt","w",stdout); }

typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int,int> PII;

const int N=6000010,mod=1e9+7,INF=0x3f3f3f3f;
const double eps=1e-6;

int n,m;
int a[N],pre[N],tot;
struct Node {
	int x,y,z,op,id,val;
}q[N],p[N];
vector<int>v;

bool cmp(Node a,Node b) {
	if(a.x!=b.x) return a.x<b.x;
	else if(a.y!=b.y) return a.y<b.y;
	else return a.op<b.op;
	
}

int tr[N],ans[N],ans1[N];

void add(int x,int c) {
	for(int i=x;i<N;i+=lowbit(i)) tr[i]+=c;
}

int sum(int x) {
	int ans=0;
	for(int i=x;i;i-=lowbit(i)) ans+=tr[i];
	return ans;
}

void cdq(int l,int r) {
	if(l>=r) return;
	
	int mid=(l+r)>>1;
	cdq(l,mid); cdq(mid+1,r);
	int i=l,j=mid+1;
	int now=0,cnt=0;
	
	while(i<=mid&&j<=r) {
		// if(l==1&&r==3&&q[i].y==q[j].y) {
			// cout<<q[i].op<<' '<<q[j].op<<"**"<<endl;
		// } 
		if(q[i].y<=q[j].y) {
			p[++now]=q[i];
			if(q[i].op==0) {
				add(q[i].z+1,1);
				cnt++;
			}
			i++;
		} else {
			p[++now]=q[j];
			if(q[j].op==1) {
				ans[q[j].id]+=sum(q[j].z+1)*q[j].val;
				ans1[q[j].id]+=cnt*q[j].val;
			}
			j++;
		}
	}
	
	while(i<=mid) {
		p[++now]=q[i];
		if(q[i].op==0) {
			add(q[i].z+1,1);
			cnt++;
		}
		i++;
	}
	
	while(j<=r) {
		p[++now]=q[j];
		if(q[j].op==1) {
			ans[q[j].id]+=sum(q[j].z+1)*q[j].val;
			ans1[q[j].id]+=cnt*q[j].val;
		}
		j++;
	}
	
	for(int i=l;i<=mid;i++) if(q[i].op==0) add(q[i].z+1,-1);
	for(int i=l,j=1;i<=r;i++,j++) q[i]=p[j];
}

int main()
{
//	ios::sync_with_stdio(false);
//	cin.tie(0);

	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) {
		int x; scanf("%d",&x);
		a[i]=x; v.pb(x);
	}
	v.pb(0);
	sort(v.begin(),v.end());
	for(int i=1;i<=n;i++) {
		int x; x=a[i];
		q[++tot]={i,x,pre[x],0,0,0};
		pre[x]=i;
	}
	for(int i=1;i<=m;i++) {
		int l,r,a,b; scanf("%d%d%d%d",&l,&r,&a,&b);
		q[++tot]={r,b,l-1,1,i,1};
		q[++tot]={l-1,a-1,l-1,1,i,1};
		q[++tot]={l-1,b,l-1,1,i,-1};
		q[++tot]={r,a-1,l-1,1,i,-1};
		
	}
	sort(q+1,q+1+tot,cmp);
	cdq(1,tot);
	for(int i=1;i<=m;i++) printf("%d %d\n",ans1[i],ans[i]);



	return 0;
}
/*
l<=x<=r
a<=val<=b
0<=pos<=l-1

x y z op val

0 1 0 1 1
0 2 0 1 -1
1 2 0 0 0

2 1 0 1 -1
2 2 0 1 1
2 2 1 0 0

0 1 0 1 1
0 2 0 1 -1
1 2 0 0 0
2 1 0 1 -1
2 2 0 1 1
2 2 1 0 0

*/