知识点 - CDQ分治
解决问题类型:
三维偏序问题
前置知识
- 树状数组
- 分治思想
实现
一. cmp函数
cmp函数在第一比较元素相等时应该返回后面元素较大的
原来的CDQ分治是可以不严格的,所以在完全相等的时候我们的操作是直接将其合并为一个节点。
而现在不严格时,为了不让相同元素进行干扰,我们就让大的在前面,这样就一定不会有问题了。
二. 树状数组要修改
现在要求树状数组的功能时 d p [ i ] = m a x ( d p [ j ] ∣ j < i ) dp[i]=max(dp[j]|j<i) dp[i]=max(dp[j]∣j<i)这里求最大值是,把所有子节点的信息更新给父亲。父亲在查询时,查询自己所有子节点的最值。
三. CDQ分治顺序
时,因为这里是求最大值,而每次都需要清空数组,所以我们希望能先算前面的,在算完前面的后,在利用前面的信息更新后面的元素。
我们原先用的是后序遍历的分治,我们现在用了中序遍历。同时,如果我们使用了中序遍历,那么为了不影响后面的操作,我们又开了一个临时数组,来进行排序。
四. 排序
因为在对于每次分治区间时,我们都需要进行用树状数组来进行维护,总的复杂度为 O ( n l o g 2 n ) O(nlog^2n) O(nlog2n) 所以我们不必再中规中矩的进行归并排序了,我们可以直接用sort函数。
复杂度
O ( N l o g 2 N ) O(Nlog^2N) O(Nlog2N)
例题
代码
#include <bits/stdc++.h>
using namespace std;
#define rep(i,j,k) for(int i = (int)j;i <= (int)k;i ++)
#define debug(x) cerr<<#x<<":"<<x<<endl
#define pb push_back
typedef long long ll;
const int MAXN = (int)2e5+7;
template<typename T,size_t size>
struct BIT {
T a[size]; int n;
void init(int _n){n=_n;memset(a,0,sizeof(T)*(_n+1));}
void erase(int x) { for(int i=x;i<=n;i+=i&-i)a[i]=0; }
void update(int x,T y){ for(int i=x;i<=n;i+=i&-i)a[i]+=y;}
T query(int x){T re=0; for(int i=x;i;i-=i&-i)re+=a[i];return re;}
};
BIT<int,MAXN>bit;
int N,K;
struct Node {
int x,y,z,id,w;
}e[MAXN],te[MAXN];
bool cmpX (const Node&a,const Node&b) { return a.x==b.x?(a.y==b.y?a.z<b.z:a.y<b.y):a.x < b.x;}
bool cmpY (const Node&a,const Node&b) { return a.y==b.y?a.z<b.z:a.y < b.y;}
int num[MAXN];
int ans[MAXN];
void meg(int l,int r) {
if (l == r) return;
int m = l+r>>1;
meg(l,m);
meg(m+1,r);
sort(e+l,e+m+1,cmpY);
sort(e+m+1,e+r+1,cmpY);
//bit.init(K);
int pl = l;
rep(i,m+1,r) {
while (pl <= m && e[pl].y <= e[i].y) bit.update(e[pl].z,e[pl].w),pl++;
num[e[i].id] += bit.query(e[i].z);
}
rep(i,l,pl-1) bit.update(e[i].z,-e[i].w);
}
int main()
{
scanf("%d %d",&N,&K);bit.init(K);
rep(i,1,N) {
int a,b,c;
scanf("%d %d %d",&a,&b,&c);
e[i] = (Node){a,b,c,i,0};
}
sort(e+1,e+1+N,cmpX);
int tn = 0,cnt = 0;
rep(i,1,N) {
cnt ++;
if (e[i].x!=e[i+1].x||e[i].y!=e[i+1].y||e[i].z!=e[i+1].z) {
e[++tn]=e[i],e[tn].id=tn,e[tn].w=cnt,cnt = 0;
te[tn] = e[tn];
num[tn] += e[tn].w-1;
}
}
meg(1,tn);
rep(i,1,tn) ans[num[i]] += te[i].w;
rep(i,0,N-1) printf("%d\n",ans[i]);
}
/*
10 3
2 3 3
3 2 2
3 3 3
3 2 1
2 2 1
1 2 2
3 2 1
2 2 3
2 2 3
3 2 1
*/