刷题记录:牛客NC50428数星星 Stars

传送门:牛客

题目描述:

一句话题意 给定n个点，定义每个点的等级是在该点左下方（含正左、正下）的点的数目，试统计每个等级有多少
个点。
输入:
5
1 1
5 1
7 1
3 3
5 5
输出:
1
2
1
1
0

一道线段树维护区间和+区间加修改的题

看完题目我们会发现对于每一个星星来说,我们此时的等级等价于之前所有横坐标<=它的所有星星的数量(我们按照$y$坐标从小到大来).所以我们可以使用线段树来维护每一个横坐标已有的星星的数量以及区间和.那么对于每一个新加的星星来说,我们可以用线段树直接查询区间和,并且对于这个新加的星星,我们将其横坐标加一即可

需要注意的是我们需要维护的横坐标可能为0,但是我们的线段树是无法维护的,此时将横纵坐标+1来避免出现这种情况即可

下面是具体的代码部分:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define root 1,n,1
#define ls rt<<1
#define rs rt<<1|1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
	ll x=0,w=1;char ch=getchar();
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
	return x*w;
}
#define maxn 1000000
const double eps=1e-8;
#define	int_INF 0x3f3f3f3f
#define ll_INF 0x3f3f3f3f3f3f3f3f
map<int,int>mp;
struct Segment_tree{
	int l,r,sum;
}tree[maxn];
int n;
void pushup(int rt) {
	tree[rt].sum=tree[ls].sum+tree[rs].sum;
}
void build(int l,int r,int rt) {
	tree[rt].l=l;tree[rt].r=r;
	if(l==r) {
		return ;
	}
	int mid=(l+r)>>1;
	build(lson);build(rson);
	pushup(rt);
}
void update(int pos,int rt,int v) {
	if(tree[rt].l==pos&&tree[rt].r==pos) {
		tree[rt].sum+=v;
		return ;
	}
	int mid=(tree[rt].l+tree[rt].r)>>1;
	if(pos<=mid) update(pos,ls,v);
	else update(pos,rs,v);
	pushup(rt);
}
int query(int l,int r,int rt) {
	if(tree[rt].l==l&&tree[rt].r==r) {
		return tree[rt].sum;
	}
	int mid=(tree[rt].l+tree[rt].r)>>1;
	if(r<=mid) return query(l,r,ls);
	else if(l>mid) return query(l,r,rs);
	else return query(l,mid,ls)+query(mid+1,r,rs);
}
int main() {
	n=read();
	build(1,40000,1);
	for(int i=1;i<=n;i++) {
		int x=read(),y=read();
		x++;y++;
		int num=query(1,x,1);
		mp[num]++;
		update(x,1,1);
	}
	for(int i=0;i<=n-1;i++) {
		printf("%d\n",mp[i]);
	}
	return 0;
}