洛谷#P3674#小清新人渣的本愿

muronglin
muronglin 2019年11月09日
  • 在其它设备中阅读本文章

小清新人渣的本愿

给定一个长度为 $n$ 的数组 $A$ ,有三种询问:
1. 询问是否存在 $A_i+A_j=x(i,j \in [l,r])$
2. 询问是否存在 $A_i-A_j=x(i,j \in [l,r])$
3. 询问是否存在 $A_i*A_j=x(i,j \in [l,r])$

解法

我们使用莫队来解这道题 对于减法, $A_i-A_j=x$ 化简为 $A_i=A_j+x$ 维护一个 bitset 存区间内有的值,等价于 s &(s<<x)存在一个 $1$ 对于加法, $A_i+A_j=x$ 化简为 $A_i-(N-A_j)=x-N$ 和减法一样,维护一个 bitset 存 $N-x$ 是否出现即可, $N$ 取最大值 对于乘法,暴力枚举因子即可

ac 代码

#include<bits/stdc++.h>
using namespace std;
bitset<100010>bi,bii;
int n,m,K,a[100010],to[100010],cnt[100010],ans[100010];
struct node
{
    int opt,l,r,x,id;
    void init(int i){scanf("%d%d%d%d",&opt,&l,&r,&x),id=i;}
    bool operator<(const node&a)const{return to[l]==to[a.l]?r<a.r:l<a.l;}
}q[100010];
void add(int x){if(cnt[x]++==0)bi[x]=1,bii[100000-x]=1;}
void del(int x){if(--cnt[x]==0)bi[x]=0,bii[100000-x]=0;}
int main()
{
    scanf("%d%d",&n,&m),K=sqrt(n);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]),to[i]=(i-1)/K+1;
    for(int i=1;i<=n;i++)q[i].init(i);
    sort(q+1,q+m+1);
    int l=1,r=0;
    for(int i=1;i<=m;i++)
    {
        while(q[i].l<l)add(a[--l]);
        while(q[i].r>r)add(a[++r]);
        while(q[i].l>l)del(a[l++]);
        while(q[i].r<r)del(a[r--]);
        if(q[i].opt==1)
        {
            if((bi&(bi<<q[i].x)).count())
                ans[q[i].id]=1;
        }
        else if(q[i].opt==2)
        {
            if((bi&(bii>>(100000-q[i].x))).count())
                ans[q[i].id]=1;
        }
        else if(q[i].opt==3)
        {
            int y=sqrt(q[i].x);
            for(int j=1;j<=y;j++)
                if(q[i].x%j==0&&bi[j]&&bi[q[i].x/j])
                    ans[q[i].id]=1;
        }
    }
    for(int i=1;i<=m;i++)
        if(ans[i])puts("hana");
        else puts("bi");
    return 0;
}