萌新求助CDQ分治,玄关
查看原帖
萌新求助CDQ分治,玄关
276588
lonely_cyx楼主2023/8/20 11:34

rt。

本地跑不出来

#include<bits/stdc++.h>
#define lowbit(x)  x&(-(x))
using namespace std;
struct node
{
	int a,b,c,cnt,ans;
}a1[2000010],a2[2000010];
int n,k,cnt,m,c[2000010],ans[2000010];
bool cmp1(node a,node b)
{
	if(a.a==b.a)
	{
		if(a.b==b.b)
			return a.c<b.c;
		else
			return a.b<b.b;
	}
	else
		return a.a<b.a;
}
bool cmp2(node a,node b)
{
	if(a.b==b.b)
		return a.c<b.c;
	else
		return a.b<b.b;
}
void add(int x,int y)
{
	for(;x<=k;x-=lowbit(x))
		c[x]+=y;
}
int ask(int x)
{
	int ans=0;
	for(;x;x-=lowbit(x))
		ans+=c[x];
	return ans;
}
void cdq(int l,int r)
{
	cout<<l<<" "<<r<<endl;
	if(l==r)
		return;
	int mid=l+r>>1;
	cdq(l,mid);
	cdq(mid+1,r);
	sort(a2+l,a2+mid+1,cmp2);
	sort(a2+mid+1,a2+r+1,cmp2);
	int j=l;
	for(int i=mid+1;i<=r;i++)
	{
		while(a2[i].b>=a2[j].b&&j<=mid)
		{
			add(a2[j].c,a2[j].cnt);
			j++;
		}
		a2[i].ans+=ask(a2[i].c);
	}
	for(int i=l;i<j;i++)
	{
		add(a2[i].c,-a2[i].cnt);
	}
}
int main()
{
	cin>>n>>k;
	for(int i=1;i<=n;i++)
		cin>>a1[i].a>>a1[i].b>>a1[i].c;
	sort(a1+1,a1+1+n,cmp1);
	for(int i=1;i<=n;i++)
	{
		cnt++;
		if(a1[i].a!=a1[i+1].a||a1[i].b!=a1[i+1].b||a1[i].c!=a1[i+1].c)
		{
			m++;
			a2[m]=a1[i];
			a2[m].cnt=cnt;
			cnt=0;
		}
	}
	cdq(1,m);
	for(int i=1;i<=m;i++)
		ans[a2[i].ans+a2[i].cnt-1]+=a2[i].cnt;
	for(int i=1;i<=n;i++)
		cout<<ans[i]<<endl;
	return 0;
}
2023/8/20 11:34
加载中...