HDU 6534 치카와 친절한 쌍 (모 + 팀 펜윅 나무)

문제의 의미

• 사용자가 간격을 추구 만족 수 n 및 m 간격 질의를 제공 | 인공 AJ | <= k는 (L <= i가 J를 <<= R) 인 로그

문제 해결

• 각 숫자 범위는 이산 AI는 다음-K 인공 지능되도록 + K 이산화 후 직접 모 + 펜윅 트리 해결 팀 AI, 어떤 섹션에서의 다수의 조건을 만족하도록 결정된다

코드

#include<bits/stdc++.h>
using namespace std;
const int maxx = 27010;
struct node{int l,r,id,block;}q[maxx];
bool cmp(node a,node b)
{
    if(a.block==b.block)return a.r<b.r;
    return a.block<b.block;
}
int a[maxx],b[maxx],t[maxx],ll[maxx],rr[maxx];
int tree[maxx],res[maxx];
int n,m,k,s;
void add(int x,int c)
{
    for(int i=x;i<=s;i+=(i&-i))
        tree[i]+=c;
}
int getsum(int x)
{
    int res=0;
    for(int i=x;i>0;i-=(i&-i))
        res+=tree[i];
    return res;
}
int main()
{
    while(~scanf("%d%d%d",&n,&m,&k))
    {
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&t[i]);
            b[i]=t[i];
            tree[i]=0;
        }
        sort(b+1,b+1+n);
        s=unique(b+1,b+1+n)-b-1;
        for(int i=1;i<=n;i++)
        {
            a[i]=lower_bound(b+1,b+1+s,t[i])-b;
            ll[i]=lower_bound(b+1,b+1+s,t[i]-k)-b;//找第一个大于或等于的位置
            rr[i]=upper_bound(b+1,b+1+s,t[i]+k)-b-1;//找第一个大于的位置
            //找不到返回end
        }
        int siz=sqrt(n);
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d",&q[i].l,&q[i].r);
            q[i].id=i;
            q[i].block=(q[i].l-1)/siz+1;
        }
        sort(q+1,q+1+m,cmp);
        int l=1,r=0,ans=0;
        for(int i=1;i<=m;i++)
        {
            while(l<q[i].l)add(a[l],-1),ans-=(getsum(rr[l])-getsum(ll[l]-1)),l++;
            while(l>q[i].l)l--,ans+=(getsum(rr[l])-getsum(ll[l]-1)),add(a[l],1);
            while(r<q[i].r)r++,ans+=(getsum(rr[r])-getsum(ll[r]-1)),add(a[r],1);
            while(r>q[i].r)add(a[r],-1),ans-=(getsum(rr[r])-getsum(ll[r]-1)),r--;
            res[q[i].id]=ans;
        }
        for(int i=1;i<=m;i++)
            printf("%d\n",res[i]);
    }
    return 0;
}