【康复训练】【51nod】1463 找朋友

Description

给定：
两个长度为n的数列A 、B
一个有m个元素的集合K
询问Q次
每次询问[l,r]，输出区间内满足|Bi-Bj|∈K 的最大Ai+Aj

数据约定：
n,Q<=100000
m <= 10
0<=A[i]<=1000000000
1<=B[i]<=n
1<=K[i]<=n
保证B[i]互不相等
Input
n Q m
A1 A2 ….An
B1 B2 ….Bn
K1 K2 ….Km
l1 r1
l2 r2
.
.
lQ rQ
Output
Q行，每行一个整数表示相对应的答案。
如果找不到这样的两个数则输出0。

题解

离线，树状数组，或主席树。

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxn 100006
#define lowbit(x) (x&-x)
using namespace std;
inline char nc(){
    static char buf[100000],*i=buf,*j=buf;
    return i==j&&(j=(i=buf)+fread(buf,1,100000,stdin),i==j)?EOF:*i++;
}
inline int _read(){
    char ch=nc();int sum=0;
    while(!(ch>='0'&&ch<='9'))ch=nc();
    while(ch>='0'&&ch<='9')sum=sum*10+ch-48,ch=nc();
    return sum;
}
int n,T,m,a[maxn],b[maxn],h[maxn],f[maxn],K[11];
struct data{
    int l,r,id,ans;
    bool operator <(const data&x)const{return r<x.r;}
}q[maxn];
void add(int x,int y){for(;x<=n;x+=lowbit(x))f[x]=max(f[x],y);}
int get(int x){
    int s=0;
    for(;x;x-=lowbit(x))s=max(s,f[x]);
    return s;
}
bool cmp(data i,data j){return i.id<j.id;}
int main(){
    freopen("sport.in","r",stdin);
    freopen("sport.out","w",stdout);
    n=_read();T=_read();m=_read();
    for(int i=1;i<=n;i++)a[i]=_read();
    for(int i=1;i<=n;i++)b[i]=_read(),h[b[i]]=i;
    for(int i=1;i<=m;i++)K[i]=_read();
    for(int i=1;i<=T;i++)q[i].l=_read(),q[i].r=_read(),q[i].id=i;
    sort(q+1,q+1+T);q[0].r=0;
    for(int k=1;k<=m;k++){
     memset(f,0,sizeof(f));
     for(int i=1;i<=T;i++){
         for(int j=q[i-1].r+1;j<=q[i].r;j++){
            if(K[k]<=b[j]&&h[b[j]-K[k]]&&h[b[j]-K[k]]<j)add(n-h[b[j]-K[k]]+1,a[j]+a[h[b[j]-K[k]]]);
            if(b[j]+K[k]<=n&&h[b[j]+K[k]]<j)add(n-h[b[j]+K[k]]+1,a[j]+a[h[b[j]+K[k]]]);
         }
         q[i].ans=max(q[i].ans,get(n-q[i].l+1));
     }
    }
    sort(q+1,q+1+T,cmp);
    for(int i=1;i<=T;i++)printf("%d\n",q[i].ans);
    return 0;
}