hdu Pair Sum and Perfect Square

 题意：

给你一个1~n组成的排列p，有q个询问，每次询问[l,r]之间有多少对i和j满足pi+pj是平方数（i<j）

思路：

用树状数组来解决

先将询问的区间按右端点从小到大，左端点从小到大的顺序排序

我们设求的满足pi+pj的对数i<j，那么我们可以从左到右枚举每个j，列举出平方数把所有i<j,满足pi+pj的i都+1，用树状数组tr记录每个pi的个数。那么当遍历到j并且询问是l~r，r=j的时候，我们就可以将答案算出：sum[r]-sum[l-1]。遍历到j控制右端点r，使其不会算到右端点在r右边的ij对数，tr求出左端点在l~r的范围内，左端点的个数，满足条件的左端点的个数就是ij的对数。

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=2e5+10;
int n,m;
int p[N];
int pos[N];
int tr[4*N];
int lowbit(int x){
	return x&(-x);
}
int sum(int x){
	int res=0;
	for(int i=x;i;i-=lowbit(i) )res+=tr[i];
	return res;
}
void add(int x,int c){
	for(int i=x;i<=n;i+=lowbit(i))tr[i]+=c;
}

struct name{
	int l,r;
	int id;
	int ans;
}q[N];
bool cmp1(name a,name b){
	if(a.r !=b.r )return a.r <b.r ;
	return a.l <b.l ;
}
bool cmp2(name a,name b){
	return a.id <b.id ;
}
void sove(){
	cin>>n;
	memset(tr,0,sizeof tr);
	memset(pos,0,sizeof pos);
	for(int i=1;i<=n;i++){
		cin>>p[i];
	}
	cin>>m;
	for(int i=1;i<=m;i++){
		cin>>q[i].l >>q[i].r ;
		q[i].id =i;
		q[i].ans =0;
	}
	std::sort(q+1,q+1+m,cmp1);
	int qid=1;
	for(int j=1;j<=n;j++){
		for(int i=1;i<=500;i++){
			int ii=i*i;
			if(ii-p[j]>0&&pos[ii-p[j]]&&ii-p[j]<=n){
				add(pos[ii-p[j]],1);
			}
		}
		while(q[qid].r ==j){
			q[qid].ans +=sum(j)-sum(q[qid].l -1);
			qid++;
		}
		pos[p[j]]=j;
	}
	std::sort(q+1,q+1+m,cmp2);
	for(int i=1;i<=m;i++){
		cout<<q[i].ans <<endl;
	}
}
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(),cout.tie();
	int t;
	cin>>t;
	while(t--){
		sove();
	}
	return 0;
}