bzoj 3262: 陌上花开 （CDQ +树状数组）

Description

有n朵花,每朵花有三个属性:花形(s)、颜色(c)、气味(m),用三个整数表示。

现在要对每朵花评级,一朵花的级别是它拥有的美丽能超过的花的数量。

定义一朵花A比另一朵花B要美丽,当且仅Sa>=Sb,Ca>=Cb,Ma>=Mb。

显然,两朵花可能有同样的属性。需要统计出评出每个等级的花的数量。

Input

第一行为N,K (1 <= N <= 100,000, 1 <= K <= 200,000 ), 分别表示花的数量和最大属性值。

以下N行，每行三个整数si, ci, mi (1 <= si, ci, mi <= K)，表示第i朵花的属性

Output

包含N行，分别表示评级为0...N-1的每级花的数量。

Sample Input

10 3
3 3 3
2 3 3
2 3 1
3 1 1
3 1 2
1 3 1
1 1 2
1 2 2
1 3 2
1 2 1

Sample Output

3
1
3
0
1
0
1
0
0
1

HINT

思路： 

CDQ分治 + 树状数组。

降维。 

这里一共是三维。 

首先排序将一维， 

第二维用 CDQ 维护， 

第三位用树状数组维护 

CDQ 分治的时候， 分成左右两个区间， 左边区间的 b 一定比 右边区间的 b 小。 

所以实现  b 是有顺序的， 在此基础上就相当于求 逆序对， 不过此题是顺序的。

每次 CDQ 之后要记得还原树状数组。 

 

 

 

 

 

#include<bits/stdc++.h>
#define low(x) (x & (-x))
using namespace std;
const int N = 2e5+100;
struct node
{
   int a,b,c,size,ans;
   bool operator < (const node &A) const {
        if (A.a != a) return a < A.a;
        if (A.b != b) return b < A.b;
        return c < A.c;
   }
}f[N],t[N];
int n,m,k,ans[N],top = 1,g[N];
void add(int x, int c){for (; x <= k; x += low(x)) g[x] += c; }
int sum(int x) {int ans = 0; for (; x > 0; x -= low(x)) ans += g[x]; 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, p = l;
    while(i <= mid || j <= r){
        if (j > r || (i <= mid && f[i].b <= f[j].b)) add(f[i].c,f[i].size),t[p++] = f[i++];
        else f[j].ans += sum(f[j].c),t[p++] = f[j++];
    } 
    for (i = l; i <= mid; ++i) add(f[i].c,-f[i].size);
    for (i = l; i <= r; ++i) f[i] = t[i];
}

int main(){
    scanf("%d%d",&n,&k); m = n;
    for (int i = 1; i <= n; ++i)
        scanf("%d%d%d",&f[i].a,&f[i].b,&f[i].c),f[i].size = 1;
    sort(f+1,f+n+1);
    for (int i = 2; i <= n; ++i)
        if (f[i].a == f[top].a && f[i].b == f[top].b && f[i].c == f[top].c)
            f[top].size ++;   else f[++top] = f[i];
    n = top;
    CDQ(1,n);
    for (int i = 1; i <= n; ++i) ans[f[i].ans + f[i].size -1] += f[i].size;
    for (int i = 0; i < m; ++i) printf("%d\n",ans[i]);
    return 0;
}

 第二种 CDQ 的写法。 

每次二分的时候，先把整个区间排序。 

用 vis 标记一下左区间，和右区间。

这种写法，排序的时候， b 的值有可能一样，

b 的值一样的时候，我们排序要确保 左区间的值在左边，右区间的值在右边。 这样才能得到正确的答案。 

为什么b 值一样的时候，右区间的值有可能靠左呢？ 是因为有可能右区间的 c 值比较小。所以比较的时候一定要考虑清楚。 

在这里，我们还定义了一个 id ， 因为 id 代表是哪朵花。 

如果 第一个是时间戳，而且时间戳没有重复，就可以不用 id 了。 

这种写法比较好理解，

但是时间会长一点。 

#include<bits/stdc++.h>
#define low(x) (x & (-x))
using namespace std;
const int N = 2e5+100;
struct node
{
   int a,b,c,size,ans,id;
   bool vis;
   bool operator < (const node &A) const {
        if (A.a != a) return a < A.a;
        if (A.b != b) return b < A.b; 
        return c < A.c;
   }
}f[N],g[N];
bool cmp(const node &A, const node &B){
	if (A.b != B.b) return A.b < B.b;
	if (A.id != B.id)	return A.id < B.id;
	return A.c < B.c;
}
int n,m,k,ans[N],top = 1,c[N];
void add(int x, int w){for (; x <= k; x += low(x)) c[x] += w; }
int sum(int x) {int ans = 0; for (; x; x -= low(x)) ans += c[x]; return ans;}
 
void CDQ(int l, int r){
    if (l >= r) return;
    int mid = (l + r) >> 1,cnt = 0;
    for (int i = l; i <= mid; ++i) g[++cnt] = f[i],g[cnt].vis = 0,g[cnt].id = i;
    for (int i = mid+1; i <= r; ++i) g[++cnt] = f[i],g[cnt].vis = 1,g[cnt].id = i;
    sort(g+1,g+cnt+1,cmp);
	for (int i = 1; i <= cnt; ++i)
		if (g[i].vis == 0) add(g[i].c,g[i].size);
		else f[g[i].id].ans += sum(g[i].c);
	for (int i = 1; i <= cnt; ++i)
		if(!g[i].vis) add(g[i].c,-g[i].size);
	CDQ(l,mid); if (mid < r) CDQ(mid+1,r);
}
 
int main(){
    scanf("%d%d",&n,&k); m = n;
    for (int i = 1; i <= n; ++i)
        scanf("%d%d%d",&f[i].a,&f[i].b,&f[i].c),f[i].size = 1;
    sort(f+1,f+n+1);

    for (int i = 2; i <= n; ++i)
        if (f[i].a == f[top].a && f[i].b == f[top].b && f[i].c == f[top].c)
            f[top].size ++;   else f[++top] = f[i];
    n = top;
    CDQ(1,n);
    for (int i = 1; i <= n; ++i) ans[f[i].ans + f[i].size -1] += f[i].size;
    for (int i = 0; i < m; ++i) printf("%d\n",ans[i]);
    return 0;
}