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

小清新人渣的本愿

给定一个长度为 \(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;
}

猜你喜欢

转载自www.cnblogs.com/muronglin/p/luogu-p3674.html