1. 程式人生 > 實用技巧 >【模板】三維偏序(陌上花開)

【模板】三維偏序(陌上花開)

題目

求滿足 \(a_i \leq a_j,b_i \leq b_j,c_i \leq c_j\) 的三元組的個數
輸出個數範圍為 \([0..n-1]\) 的每種個數出現的次數

解析

\(cdq\) 分治版題,三維數點
先讓第一維排序,然後歸併思想排序第二維
考慮分治的左區間對右區間的貢獻
每次用左區間的點更新右區間點的答案
特別注意如果一個點與另一個點三個屬性都相同
由於我們是算他左邊點對他的貢獻
所以我們最後記錄最終的答案時取三個屬性都相同排在最後的

\(Code\)

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

const int N = 2e5 + 5;
int n , k , ans[N] , c[N];

struct node{
	int a , b , c , f;	
}fl[N] , nfl[N];

inline bool cmp(node x , node y)
{
	if (x.a == y.a)
	{
		if (x.b == y.b) return x.c < y.c;
		return x.b < y.b;
	}
	return x.a < y.a;
}

inline int lowbit(int x){return x & (-x);}
inline void add(int x , int v){for(; x <= k; x += lowbit(x)) c[x] += v;}

inline int query(int x)
{
	int res = 0;
	for(; x; x -= lowbit(x)) res += c[x];
	return res;
}

inline void solve(int l , int r)
{
	if (l == r) return;
	int mid = (l + r) >> 1;
	solve(l , mid) , solve(mid + 1 , r);
	int i = l , j = mid + 1 , k = l;
	while (i <= mid && j <= r)
	{
		if (fl[i].b <= fl[j].b) add(fl[i].c , 1) , nfl[k++] = fl[i] , i++;
		else fl[j].f += query(fl[j].c) , nfl[k++] = fl[j] , j++;
	}
	while (i <= mid) add(fl[i].c , 1) , nfl[k++] = fl[i] , i++;
	while (j <= r) fl[j].f += query(fl[j].c) , nfl[k++] = fl[j] , j++;
	for(register int p = l; p <= mid; p++) add(fl[p].c , -1);
	for(register int p = l; p <= r; p++) fl[p] = nfl[p];
}

int main()
{
	scanf("%d%d" , &n , &k);
	for(register int i = 1; i <= n; i++) scanf("%d%d%d" , &fl[i].a , &fl[i].b , &fl[i].c);
	sort(fl + 1 , fl + n + 1 , cmp);
	solve(1 , n);
	for(register int i = 1; i <= n; i++)
	{
		int j = i;
		while (fl[i].a == fl[i + 1].a && fl[i].b == fl[i + 1].b && fl[i].c == fl[i + 1].c && i < n) i++;
		ans[fl[i].f] += i - j + 1;
	}
	for(register int i = 0; i < n; i++) printf("%d\n" , ans[i]);
}