前言
总所周知小白赛对真小白一点也不友好。所以赛后又来补题了。
题目链接
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;
}