牛客小白月赛18——G.Forsaken的三维数点

前言
总所周知小白赛对真小白一点也不友好。所以赛后又来补题了。

题目链接

https://ac.nowcoder.com/acm/contest/1221/G

题目理解

题目中的两个操作，一个是增加单点，一个是单点查询总和，看和这两个条件，就可以想到线段树或者树状数组，由于不涉及到区间修改和区间查询，所以可以用朴素的线段树或者树状数组。选择更好敲代码的树状数组。

• op==1时，将增加的点离原点的距离当成下标，然后在树状数组中增加这个点的信息。
• op==2时，将所要找的半径当成需要查询的点，直接用树状数组查询就行了。由于题目给定的是总和大于等于k，所以这里可以用二分答案来找到这个半径。由于k只有1e5的量级，二分并不会超时。

需要注意的地方：题目中的k可以等于0，但是树状数组下标不能为零，所以在查询的时候要把sum初始化成树状数组下标为零的元素，表示所查询的范围包括了(0,0,0)这个原点。

AC代码

#include<iostream>
#include<cmath>
#include<cstdio>
using namespace std;

#define int long long

const int maxn=2e5+5;
int c[maxn]={0};

int lowbit(int i){
	return i&(-i);
}
void updata(int k){
	for(int i=k;i<maxn;i+=lowbit(i)){
		c[i]+=1;
		if(i==0) return;
	}
}
int query(int k){
	int sum=c[0];
	for(int i=k;i>0;i-=lowbit(i)){
		sum+=c[i];
	}
	return sum;	
}
signed main(){
	int n,ans=0;
	scanf("%lld",&n);
	for(int i=0;i<n;i++){
		int op;
		scanf("%lld",&op);
		if(op==1){
			double x,y,z;
			scanf("%lf%lf%lf",&x,&y,&z);
			ans++; //记录增加多少点能量
			double dis=ceil(sqrt(x*x+y*y+z*z));
			int p=ceil(dis); //距离向上取整
			updata(p);
		}else if(op==2){
			int k;
			scanf("%lld",&k);
			if(ans<k){
				printf("-1\n");	
				continue;
			}
			//二分找答案
			int l=0,r=maxn,mid;
			while(l<=r){
				mid=(l+r)>>1;
				if(query(mid)>=k) r=mid-1;
				else l=mid+1;
			}
			printf("%lld\n",l);
		}
	}
	return 0;
}