Hdu 5213-Lucky 莫队,容斥原理,分块

题目:http://acm.hdu.edu.cn/showproblem.php?pid=5213

Lucky

Time Limit: 6000/3000 MS (Java/Others)    Memory Limit: 65536/65536 K (Java/Others)
Total Submission(s): 763    Accepted Submission(s): 249

Problem Description

WLD is always very lucky.His secret is a lucky number K.k is a fixed odd number. Now he meets a stranger with N numbers:a1,a2,...,aN.The stranger asks him Mquestions.Each question is like this:Given two ranges [Li,Ri] and [Ui,Vi],you can choose two numbers X and Y to make aX+aY=K.The X you can choose is between Li and Ri and the Y you can choose is between Ui and Vi.How many pairs of numbers(X,Y) you can choose?
If WLD can answer all the questions correctly,he'll be the luckiest man in the world.Can you help him?

 

Input

There are multiple cases.(At MOST 5)

For each case:

The first line contains an integer N(1≤N≤30000).

The following line contains an integer K(2≤K≤2∗N),WLD's lucky number.K is odd.

The following line contains N integers a1,a2,...,aN(1≤ai≤N).

The following line contains an integer M(1≤M≤30000),the sum of the questions WLD has to answer.

The following M lines,the i-th line contains 4 numbers Li,Ri,Ui,Vi(1≤Li≤Ri<Ui≤Vi≤N),describing the i-th question the stranger asks.

 

Output

For each case:

Print the total of pairs WLD can choose for each question.

 

Sample Input

5 3 1 2 1 2 3 1 1 2 3 5

 

Sample Output

2

Hint

a1+a4=a2+a3=3=K. So we have two pairs of numbers (1,4) and (2,3). Good luck!

 

Source

BestCoder Round #39 ($) 

 

Recommend

hujie   |   We have carefully selected several similar problems for you:  5659 5658 5657 5656 5655 

 

题意：给你一个a数列，再给出一些询问，每次询问给两个区间，分别为[l,r]和[u,v]，且1<=l<=r<u<=v<=n，让你从[l,r]中找一个a[i],在[u,v]中找一个a[j],使得a[i]+a[j]＝K，问有多少对。

题解：

好几天没发题解了。。。

来个莫队压压惊。。。

莫队＋容斥

莫队很好想的，主要是如何用容斥。

我们把每组询问的两个区间写出来。

         l-----r--------u-----v

         |<A>|

                            |<B>|

                |<--C-->|

然后定义f(x,y)为第一个数i在x区间，第二个数j在y区间的a[i]+a[j]=K的方案数。

我们要求的为f(A,B)=f(A+B+C,A+B+C)－f(A+C,A+C)－f(B+C,B+C)+f(C,C)

然后就可以用莫队做了。

这里有些小技巧：在加区间的时候，我们可以把区间的 两端点 和 当前区间的值是加还是减 记录下来，这样比较简单。

另外这道题的数组大小好离奇。。。

具体看程序:

#include<bits/stdc++.h>
using namespace std;
#define MAXN 30010
#define MAXM 30010
struct node
{
    int l,r,id,fh;
}q[MAXM*4];
int a[MAXN],pos[MAXN],sum[MAXN*2],N,ans[MAXM*2];
int read()
{
    int s=0,fh=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')fh=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){s=s*10+(ch-'0');ch=getchar();}
    return s*fh;
}
void Add(int ll,int rr,int ii,int ff){q[++N].l=ll;q[N].r=rr;q[N].id=ii;q[N].fh=ff;}
bool cmp(node aa,node bb)
{
    if(pos[aa.l]==pos[bb.l])return aa.r<bb.r;
    return aa.l<bb.l;
}
int main()
{
    int n,k,i,m,block,tot,L,R,U,V;
    while(scanf("%d",&n)!=EOF)
    {
        k=read();
        for(i=1;i<=n;i++)a[i]=read();
        m=read();
        N=0;
        for(i=1;i<=m;i++)
        {
            L=read();R=read();U=read();V=read();
            Add(L,V,i,1);Add(L,U-1,i,-1);Add(R+1,V,i,-1);Add(R+1,U-1,i,1);
        }
        block=(int)sqrt(n);
        for(i=1;i<=n;i++)pos[i]=(int)(i-1)/block+1;
        sort(q+1,q+N+1,cmp);
        memset(ans,0,sizeof(ans));
        L=1;R=0;
        tot=0;//当前区间有多少对a[i]+a[j]=k.
        memset(sum,0,sizeof(sum));//当前区间数字为i的有sum[i]个.
        for(i=1;i<=N;i++)
        {
            while(L<q[i].l)
            {
                sum[a[L]]--;
                tot-=sum[k-a[L]];
                //sum[a[L]]--;
                //if(k==a[L]*2)tot++;
                L++;
            }
            while(L>q[i].l)
            {
                L--;
                tot+=sum[k-a[L]];
                sum[a[L]]++;
            }
            while(R<q[i].r)
            {
                R++;
                tot+=sum[k-a[R]];
                sum[a[R]]++;
            }
            while(R>q[i].r)
            {
                sum[a[R]]--;
                tot-=sum[k-a[R]];
                //sum[a[R]]--;
                //if(k==a[R]*2)tot++;
                R--;
            }
            ans[q[i].id]+=q[i].fh*tot;
        }
        for(i=1;i<=m;i++)printf("%d
",ans[i]);
    }
    fclose(stdin);
    fclose(stdout);
    return 0;
}