借鉴了一位大佬的博客:https://www.cnblogs.com/HDUjackyan/p/8996172.html
题目链接:http://acm.hdu.edu.cn/userloginex.php
题意:给我们n个数,然后给我们一个数字K,再给我们两个区间[L,R],[U,V],问我们从这两个区间里分别取一个数,有多少种取法,可以使得所取的两个数和为K。
分析:莫队算法+容斥定理。因为莫队算法是知道[l,r]的值后可以求[l’,r’]的值,因为每次都会给两个区间,所以我们需要将两个不相关的区间想办法让他们产生连续。对于[x1,y1][x2,y2]我们这时候可以通过容斥定理将其变为F[x1,y1,x2,y2]=F[x1,y2]+F[y1+1,x2-1]-F[y1+1,y2]-F[x1,x2-1]将一次求解变成4个区间分别求解,每次在一个区间中求解有多少种取法可以使得两个数和为k。
AC代码如下:
#include <iostream>
#include <stdio.h>
#include <math.h>
#include <string.h>
#include <algorithm>
#include <iostream>
#include <stdlib.h>
#include <queue>
#include <map>
#include <vector>
using namespace std;
const int maxn = 30000+10;
int N,K,a[maxn],m,belong[maxn],ans;
int sum[2*maxn];
struct Query
{
int x1,y1,x2,y2,ans;
}q[maxn];///储存查询
struct Query1
{
int l,r,ans,flag,fa;
}arr[maxn*4];///储存区间
bool cmp1(Query1 a, Query1 b)///分块排序
{
if(belong[a.l]==belong[b.l])
return a.r<b.r;
return belong[a.l]<belong[b.l];
}
void updata(int pos, int flag)
{
sum[a[pos]]+=flag;
if(K>a[pos])
ans+=sum[K-a[pos]]*flag;
}
int main()
{
while(scanf("%d",&N)!=EOF)
{
scanf("%d",&K);
for(int i=1; i<=N; i++)
scanf("%d",&a[i]);
int block =sqrt(N);
for(int i=1; i<=N; i++)
belong[i]=(i-1)/block+1;
scanf("%d",&m);
for(int i=1; i<=m; i++)
{
scanf("%d %d %d %d",&q[i].x1,&q[i].y1,&q[i].x2,&q[i].y2);
q[i].ans=0;
for(int j=0; j<4; j++)
{
arr[i+j*m].fa=i;
arr[i+j*m].ans=0;
}///初始化各区间
arr[i].flag=1,arr[i].l=q[i].x1,arr[i].r=q[i].y2;///[x1,y2];
arr[i+m].flag=1,arr[i+m].l=q[i].y1+1,arr[i+m].r=q[i].x2-1;///[y1+1,x2-1]
arr[i+m*2].flag=0,arr[i+m*2].l=q[i].y1+1,arr[i+m*2].r=q[i].y2;///[y1+1,y2]
arr[i+m*3].flag=0,arr[i+m*3].l=q[i].x1,arr[i+m*3].r=q[i].x2-1;///[x1,x2-1]
}
sort(arr+1,arr+4*m+1,cmp1);
ans=0;
int l=1,r=0;
memset(sum,0,sizeof(sum));
sum[a[1]]=1;
for(int i=1; i<=4*m; i++)
{
while(l<arr[i].l)
{
updata(l,-1);
l++;
}
while(l>arr[i].l)
{
l--;
updata(l,1);
}
while(r<arr[i].r)
{
r++;
updata(r,1);
}
while(r>arr[i].r)
{
updata(r,-1);
r--;
}
arr[i].ans=ans;
}
for(int i=1; i<=4*m; i++)
{
int x=arr[i].fa;
int y=arr[i].ans;
if(arr[i].flag) q[x].ans+=y;
else
q[x].ans-=y;
}
for(int i=1; i<=m; i++)
printf("%d\n",q[i].ans);
}
return 0;
}