hdu5213（莫队算法+容斥定理）

借鉴了一位大佬的博客：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;
}