HDU - 5213 Lucky(莫队算法+容斥思想)

题目大意:

多次询问,每次询问两个区间 [l1,r1][l2,r2] 个选出一个元素,有多少种选择方法可以使选出的两数的和为定值 k 。

分析:

f([a,b],[c,d]) 表示区间 [a,b],[c,d] 的选择方式数,那么,就可以推得一个公式:

f([a,b],[c,d])=f([a,d],[a,d])f([a,c1],[a,c1])f([b+1,d],[b+1,d])+f([b+1,c1],[b+1,c1])

然后只要求每个区间 [l,r] 对应的 f([l,r],[l,r]) 就可以了。

代码:

#include<bits/stdc++.h>
using namespace std;
#define maxn 30050
struct mo
{
    int l;int r;int id;int ans;
}q[4*maxn];
int m,n,cnt,k,ans,num[2*maxn],a[maxn],out[maxn],pos[maxn],temp_a,temp_b,temp_c,temp_d;
void add_mo(int l,int r)
{
    q[cnt].l=l;q[cnt].r=r;q[cnt].id=cnt;
}
bool cmp_mo(mo a,mo b)
{
    return pos[a.l]==pos[b.l] ? a.r<b.r : a.l<b.l;
}
bool cmp_id(mo a,mo b)
{
    return a.id<b.id;
}
void move_mo(int x,int add)
{
    ans+=(add*num[k-a[x]]);num[a[x]]+=add;
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        cnt=0;ans=0;
        memset(num,0,sizeof(num));
        int unit=sqrt(n);
        for(int i=0;i<=n;i++)pos[i]=i/unit;

        scanf("%d",&k);
        for(int i=1;i<=n;i++)scanf("%d",&a[i]);
        scanf("%d",&m);
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d%d",&temp_a,&temp_b,&temp_c,&temp_d);
            add_mo(temp_a,temp_d);cnt++;
            add_mo(temp_a,temp_c-1);cnt++;
            add_mo(temp_b+1,temp_d);cnt++;
            add_mo(temp_b+1,temp_c-1);cnt++;
        }
        sort(q,q+cnt,cmp_mo);
        int l=1,r=0;
        for(int i=0;i<cnt;i++)
        {
            while(r<q[i].r)move_mo(r+1,1),r++;
            while(l>q[i].l)move_mo(l-1,1),l--;
            while(r>q[i].r)move_mo(r,-1),r--;
            while(l<q[i].l)move_mo(l,-1),l++;
            q[i].ans=ans;
        }
        sort(q,q+cnt,cmp_id);
        for(int i=0;i<cnt;i+=4)
        {
            //printf("[%d,%d]=%d\n",q[i].l,q[i].r,q[i].ans);
            out[q[i].id/4]=q[i].ans-q[i+1].ans-q[i+2].ans+q[i+3].ans;
            //printf("[%d,%d]=%d\n",q[i+3].l,q[i+3].r,q[i+3].ans);
            //if(a[q[i+3].l]+a[q[i+3].r]==k)out[q[i].id/4]++;
        }
        for(int i=0;i<m;i++)
        {
            printf("%d\n",out[i]);
        }
        //cout<<"ans="<<ans<<endl;
    }
}
全部评论

相关推荐

11-11 16:40
已编辑
门头沟学院 人工智能
不知道怎么取名字_:这个有点不合理了,相当于已经毕业了,但还是没转正,这不就是白嫖
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
12-17 16:48
今天九点半到公司,我跟往常一样先扫了眼电脑,屁活儿没有。寻思着没事干,就去蹲了个厕所,回来摸出手机刷了会儿。结果老板刚好路过,拍了我一下说上班别玩手机,我吓得赶紧揣兜里。也就过了四十分钟吧,我的直属领导把我叫到小隔间,上来就给我一句:“你玩手机这事儿把老板惹毛了,说白了,你可以重新找工作了,等下&nbsp;HR&nbsp;会来跟你谈。”&nbsp;我当时脑子直接宕机,一句话都没憋出来。后面&nbsp;HR&nbsp;找我谈话,直属领导也在旁边。HR&nbsp;说我这毛病不是一次两次了,属于屡教不改,不光上班玩手机,还用公司电脑看论文、弄学校的事儿。我当时人都傻了,上班摸鱼是不对,可我都是闲得发慌的时候才摸啊!而且玩手机这事儿,从来没人跟我说过后果这么严重,更没人告诉我在公司学个习也算犯错!连一次口头提醒都没有,哪儿来的屡教不改啊?更让我膈应的是,昨天部门刚开了会,说四个实习生里留一个转正,让大家好好表现。结果今天我就因为玩手机被开了。但搞笑的是,开会前直属领导就把我叫去小会议室,明明白白告诉我:“转正这事儿你就别想了,你的学历达不到我们部门要求,当初招你进来也没打算给你这个机会。”合着我没入贵厂的眼是吧?可我都已经被排除在转正名单外了,摸个鱼至于直接把我开了吗?真的太离谱了!
rush$0522:转正名单没进,大概率本来就没打算留你
摸鱼被leader发现了...
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务