【jzoj3854】【CF173E】【Camping Groups】【分组】【数据结构】

题目大意

Bsny所在的精灵社区有n个居民，每个居民有一定的地位和年龄，ri表示第i个人的地位，ai表示第i个人的年龄。

最近社区里要举行活动，要求几个人分成一个小组，小组中必须要有一个队长，要成为队长有这样的条件：

1、队长在小组中的地位应该是最高的(可以并列第一)；

2、小组中其他成员的年龄和队长的年龄差距不能超过K。

有些人想和自己亲密的人组在同一个小组，同时希望所在的小组人越多越好。比如x和y想在同一个小组，同时希望它们所在的小组人越多越好，当然，它们也必须选一个符合上述要求的队长，那么问你，要同时包含x和y的小组，最多可以组多少人？

解题思路

先将人按地位排序，从小到大添加，年龄范围内的人数即当队长的贡献，注意地位相同的要先一起添加，再查询。

按两个人中较高地位排序，从大到小做，将地位高的人及其当队长的贡献添加进数据结构，年龄范围内的最大贡献即答案。

code

#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define LF double
#define LL long long
#define min(a,b) ((a<b)?a:b)
#define max(a,b) ((a>b)?a:b)
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
using namespace std;
int const maxn=1e5,maxa=1e9,inf=1e9;
int n,K,q,pon=1,pon2=1,num[maxn+10],ans[maxn+10],t[maxn*30+10],t2[maxn*30+10],
lson[maxn*30+10],lson2[maxn*30+10],rson[maxn*30+10],rson2[maxn*30+10];
struct rec{
    int r,a,p,x,y;
};
rec a[maxn+10],b[maxn+10];
bool cmp(rec x,rec y){
    return x.r<y.r;
    //return (x.r<y.r)||((x.r==y.r)&&(x.p<y.p));
}
void add(int now,int l,int r,int v){
    int mid=(l+r)>>1;
    t[now]++;
    if(l==r)return;
    if(v<=mid)add((lson[now])?lson[now]:(lson[now]=++pon),l,mid,v);
    else add((rson[now])?rson[now]:(rson[now]=++pon),mid+1,r,v);
}
void add2(int now,int l,int r,int v,int vv){
    int mid=(l+r)>>1;
    t2[now]=max(t2[now],vv);
    if(l==r)return;
    if(v<=mid)add2((lson2[now])?lson2[now]:(lson2[now]=++pon2),l,mid,v,vv);
    else add2((rson2[now])?rson2[now]:(rson2[now]=++pon2),mid+1,r,v,vv);
}
/*int qury(int now,int l,int r,int ll,int rr){
    int mid=(l+r)>>1;
    if(l==r)return t[now];
    if(rr<=mid)return (lson[now])?qury(lson[now],l,mid,ll,rr):0;
    else if(mid<ll) return (rson[now])?qury(rson[now],mid+1,r,ll,rr):0;
    else return ((lson[now])?qury(lson[now],l,mid,ll,mid):0)+((rson[now])?qury(rson[now],mid+1,r,mid+1,rr):0);
}*/
int qury(int now,int l,int r,int ll,int rr){
    int mid=(l+r)>>1;
    if((l==ll)&&(r==rr))return t[now];
    if(rr<=mid)return (lson[now])?qury(lson[now],l,mid,ll,rr):0;
    else if(mid<ll) return (rson[now])?qury(rson[now],mid+1,r,ll,rr):0;
    else return ((lson[now])?qury(lson[now],l,mid,ll,mid):0)+((rson[now])?qury(rson[now],mid+1,r,mid+1,rr):0);
}
int qury2(int now,int l,int r,int ll,int rr){
    int mid=(l+r)>>1;
    if((l==ll)&&(r==rr))return t2[now];
    if(rr<=mid)return (lson2[now])?qury2(lson2[now],l,mid,ll,rr):0;
    else if(mid<ll) return (rson2[now])?qury2(rson2[now],mid+1,r,ll,rr):0;
    else{
        int a1=(lson2[now])?qury2(lson2[now],l,mid,ll,mid):0,
            a2=(rson2[now])?qury2(rson2[now],mid+1,r,mid+1,rr):0;
        return max(a1,a2);
    }
}
int main(){
    //freopen("d.in","r",stdin);
    //freopen("d.out","w",stdout);
    scanf("%d%d",&n,&K);
    fo(i,1,n)scanf("%d",&a[i].r);
    fo(i,1,n)scanf("%d",&a[i].a),a[i].p=i;
    sort(a+1,a+n+1,cmp);
    fo(i,1,n)num[a[i].p]=i;
    for(int i=1,j;i<=n;i=j+1){
        for(j=i;(j<n)&&(a[j+1].r==a[i].r);j++);
        fo(k,i,j)add(1,1,maxa,a[k].a);
        fo(k,i,j)a[k].x=qury(1,1,maxa,max(a[k].a-K,1),min(a[k].a+K,maxa));
        //printf("%d\n",a[i].x);
    }
    scanf("%d",&q);int x,y;
    fo(i,1,q){
        scanf("%d%d",&x,&y);b[i].p=i;
        b[i].r=max(a[num[x]].r,a[num[y]].r);
        b[i].x=max(max(a[num[x]].a,a[num[y]].a)-K,1);
        b[i].y=min(min(a[num[x]].a,a[num[y]].a)+K,maxa);
    }
    sort(b+1,b+q+1,cmp);
    int j=n;
    fd(i,q,1){
        for(;(j>=1)&&(a[j].r>=b[i].r);add2(1,1,maxa,a[j].a,a[j].x),j--);
        if(b[i].x<=b[i].y)ans[b[i].p]=qury2(1,1,maxa,b[i].x,b[i].y);
    }
    fo(i,1,q)printf("%d\n",(ans[i])?ans[i]:-1);
    return 0;
}