题目大意:
有n颗星星,每一颗星星都有一个坐标,对于这颗星星左下方(包括正下方和正左方)有k颗星星,就说这颗星星是k等级的星星。现在叫你求出每个等级各有多少颗星星。
输入保证每颗星星的纵坐标升序排列
(上面这句话很重要)
输入样例:
5
1 1
5 1
7 1
3 3
5 5
输出样例:
1 2 1 1 0
先贴代码:
#include <iostream>
#include <vector>
#include <cstdio>
#include <ctime>
using namespace std;
int n;
const int N=32000;
struct node{
int l,r;
int cnt;
}tree[4*N+1];
int ans[N]={0};
int read()
{
int x=0,f=1; char ch=getchar();
while(ch<'0' || ch>'9') {if(ch=='-') f=-1; ch=getchar();}
while(ch>='0' && ch<='9') {x=x*10+ch-'0'; ch=getchar();}
return x*f;
}
int cnt=0;
void sum(int k,int x,int y)//求区间和模板
{
if(tree[k].l>=x && tree[k].r<=y)
{
cnt+=tree[k].cnt;
return ;
}
int mid=(tree[k].l+tree[k].r)>>1;
if(x<=mid) sum(2*k,x,y);
if(y>mid) sum(2*k+1,x,y);
}
void updata(int k,int x)
{
if(tree[k].l==tree[k].r)
{
tree[k].cnt++;
return ;
}
int mid=(tree[k].l+tree[k].r)>>1;
if(x<=mid) updata(2*k,x);
else updata(2*k+1,x);
tree[k].cnt=tree[2*k+1].cnt+tree[2*k].cnt;//一定要更新回去啊!!!
}
void build(int k,int l,int r)
{
tree[k].l=l,tree[k].r=r;
if(l==r) return;
int mid=(l+r)>>1;
build(2*k,l,mid);
build(2*k+1,mid+1,r);
}
int main()
{
/*
freopen("stars.in","r",stdin);
freopen("stars.out","w",stdout);
*/
n=read();
int maxx=INT_MIN;
vector<int> x;
for(int i=1;i<=n;i++)
{
int tmp=read();
x.push_back(tmp);
maxx=max(maxx,tmp);
tmp=read();//读掉纵坐标
}
build(1,0,maxx);
for(int i=0;i<n;i++)
{
cnt=0;//有多少个覆盖点,每次都要重置一次,好像可以放进递归的函数参数里面
sum(1,0,x[i]);//求0~x的区间和(找有多少个覆盖点)
ans[cnt]++;//累入答案
updata(1,x[i]);//更新叶节点
}
for(int i=0;i<n;i++)
cout<<ans[i]<<" ";
//cout<<endl<<(double)clock()/CLOCKS_PER_SEC<<"s";
return 0;
}
总体思路:线段树思路上变一下形
怎么看出来和线段树有关系呢?
先来看一遍题目:找有多少个点在当前点的左下方(正左正下也算)
我们把在它左下方的点称之为“覆盖点”
输入给定了点的坐标升序排列
那么我们就只要找出比当前点横坐标要小的点就好了
(纵坐标是没有用的信息)
然后仔细想想,把0~每个点的横坐标作为线段
那么找一个点的覆盖点),就是相当于查询0~x的区间和(假设覆盖点坐标为x1,x2,x3……,那么它们就是叶节点,找有多少个叶节点存在,就相当于求区间和了——画图理解)
或者换个角度解释:
对于每一个节点,状态cnt存的是该节点表示的区间有多少个纵坐标在区间范围内的星星存在
举个例子:
有且只有两颗星星坐标为(1,1)(3,1)
对于区间【0,4】这个节点的cnt的值就是2,因为只有1和3在区间之内
然后对于后一个点,千万别忘了把当前这个点更新
省时间(吧)的做法:取横坐标的最大值作为取件最右端建树