51nod 1463 找朋友

离线+线段树

将询问按右端点排序。这样对于(i,j)，j必定在当前询问范围内。可以将a[i]+a[j]累积到i上，那样就变成了求区间最大值。

#include <bits/stdc++.h>
using namespace std;

void read(int&a){
    char ch;while(!((ch=getchar())>='0')&&(ch<='9'));
    a=ch-'0';while(((ch=getchar())>='0')&&(ch<='9'))a*=10,a+=ch-'0';
}

inline void prin_d(int x)
{
    if (x > 9)
    {
        prin_d(x / 10);
    }
    putchar(x % 10 + '0');
    return ;
}

const int MAXN=100100;

struct node
{
	int l,r;
	long long nsum;
}segtree[MAXN<<2];

void build(int i,int l,int r)
{
	segtree[i].l=l;
	segtree[i].r=r;
	if(l==r)
	{
		segtree[i].nsum=0;
		return;
	}
	int mid=(l+r)>>1;
	build(i<<1,l,mid);
	build(i<<1|1,mid+1,r);
	segtree[i].nsum=0;
}
void update(int i,int a,long long c)
{
	if(segtree[i].l==a&&segtree[i].r==a)
	{
		if(segtree[i].nsum<c)
			segtree[i].nsum=c;
		return;
	}
	int mid=(segtree[i].l+segtree[i].r)>>1;
	if(a<=mid)
		update(i<<1,a,c);
	else
		update(i<<1|1,a,c);
	segtree[i].nsum=max(segtree[i<<1].nsum,segtree[i<<1|1].nsum);
}

long long query(int i,int a,int b)
{
	if(segtree[i].l==a&&segtree[i].r==b)
	{
		return segtree[i].nsum;
	}
	int mid=(segtree[i].l+segtree[i].r)>>1;
	if(b<=mid)
		return query(i<<1,a,b);
	else if(a>mid)
		return query(i<<1|1,a,b);
	else
		return max(query(i<<1,a,mid),query(i<<1|1,mid+1,b));
}

int a[MAXN],b[MAXN],pos[MAXN],k[12];

struct Query
{
	int id,l,r,ans;
}q[MAXN];

bool cmp1(Query q1,Query q2)
{
	if(q1.r!=q2.r)
		return q1.r<q2.r;
	return q1.l<q2.l;
}

bool cmp2(Query q1,Query q2)
{
	return q1.id<q2.id;
}

int main()
{
	int n,m,qsum,i,j,now;
	while(~scanf("%d%d%d",&n,&qsum,&m))
	{
		for(i=1;i<=n;i++)
			read(a[i]);
		for(i=1;i<=n;i++)
		{
			read(b[i]);
			pos[b[i]]=i;
		}
		for(i=0;i<m;i++)
			read(k[i]);
		for(i=0;i<qsum;i++)
		{
			read(q[i].l);
			read(q[i].r);
			q[i].id=i;
		}
		build(1,1,n);
		sort(q,q+qsum,cmp1);
		now=0;
		for(i=0;i<qsum;i++)
		{
			while(now<q[i].r)
			{
				now++;
				for(j=0;j<m;j++)
				{
					if(b[now]>k[j]&&pos[b[now]-k[j]]<=now)
						update(1,pos[b[now]-k[j]],a[now]+a[pos[b[now]-k[j]]]);
					if(b[now]+k[j]<=n&&pos[b[now]+k[j]]<=now)
						update(1,pos[b[now]+k[j]],a[now]+a[pos[b[now]+k[j]]]);
				}
			}
			q[i].ans=query(1,q[i].l,q[i].r);
		}
		sort(q,q+qsum,cmp2);
		for(i=0;i<qsum;i++)
		{
			prin_d(q[i].ans);
			puts("");
		}
	}
}