并不对劲的复健训练-p3674

Posted 2022-11-06 xzyf

题目大意

给出序列$ a_1,...,a_n $ ( $ n\leq10^5，a\leq 10^5 $ )，有\(m\) ( \(m\leq 10^5\))个以下三类询问：

(1)给出\(l,r,k\)(\(k\leq 10^5\))，问是否存在\(x,y\)使\(x\in[l,r],y\in[l,r],a_x-a_y=k\)
(2)给出\(l,r,k\)(\(k\leq 10^5\))，问是否存在\(x,y\)使\(x\in[l,r],y\in[l,r],a_x+a_y=k\)
(3)给出\(l,r,k\)(\(k\leq 10^5\))，问是否存在\(x,y\)使\(x\in[l,r],y\in[l,r],a_x\times a_y=k\)

题解

发现\(a、k\)的值域较小，而且没有修改或强制在线，考虑莫队。
维护表示每个值在当前区间中是否存在的bitset，记为\(c\)。
对于(1)操作，将式子变形为\(a_x=a_y+k\)，判断\((c<<k)| c\)中第\(k\)位是否为1
对于(2)操作，将式子变形为\(a_x=k-a_y\)，发现\(-a_y\)不好直接算，但是可以维护\((10^5-a_y)\)，使式子变为\(a_x=(10^5-a_y)-10^5+k\)，记\((10^5-a_y)\)为\(d\)，判断\((d>>(10^5-k))|c\)中第\(k\)位是否为1
对于(3)操作，难以变形，可以考虑把\(k\)拆成\(\sqrt k\)个约数，判断\(\sqrt k\)个约数中是否有存在于\(a_l,...,a_r\)的。这样听上去是 莫队的时间复杂度\(\times \sqrt k=n^2\)级别的。但是发现这么写的代码长这样：

rep i 1 to m
    移动左右端点;
    判断根号k个约数;

所以判断约数的 \(\sqrt k\)和莫队的\(\sqrt n\)是相加的关系。

代码

#include<algorithm>
#include<bitset>
#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<ctime>
#include<iomanip>
#include<iostream>
#include<map>
#include<queue>
#include<set>
#include<stack>
#include<vector>
#define rep(i,x,y) for(register int i=(x);i<=(y);++i)
#define dwn(i,x,y) for(register int i=(x);i>=(y);--i)
#define view(u,k) for(int k=fir[u];k!=-1;k=nxt[k])
#define maxn 100007
#define maxk 100000
#define LL long long
using namespace std;
int read()

    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)&&ch!='-')ch=getchar();
    if(ch=='-')f=-1,ch=getchar();
    while(isdigit(ch))x=(x<<1)+(x<<3)+ch-'0',ch=getchar();
    return x*f;

void write(int x)

    if(x==0)putchar('0'),putchar('\n');return;
    int f=0;char ch[20];
    if(x<0)putchar('-'),x=-x;
    while(x)ch[++f]=x%10+'0',x/=10;
    while(f)putchar(ch[f--]);
    putchar('\n');
    return;

bitset<100001>now,pnow;
int a[maxn],n,m,num[maxn],ans[maxn],blo=300;
struct questint f,l,r,x,id;q[maxn];
bool cmp(quest x,quest y)return (x.l/blo==y.l/blo)?(x.r<y.r):(x.l<y.l);
void add(int id,int f)

    num[a[id]]+=f;
    if(f==1&&num[a[id]]==1)now[a[id]]=1,pnow[maxk-a[id]]=1;
    if(f==-1&&!num[a[id]])now[a[id]]=0,pnow[maxk-a[id]]=0;

int main()

    n=read(),m=read();
    rep(i,1,n)a[i]=read();
    rep(i,1,m)q[i].f=read(),q[i].l=read(),q[i].r=read(),q[i].x=read(),q[i].id=i;
    int nowl=1,nowr=1;num[a[1]]=1,now[a[1]]=1,pnow[maxk-a[1]]=1;
    sort(q+1,q+m+1,cmp);
    rep(i,1,m)
    
        while(nowl<q[i].l)add(nowl,-1),nowl++;
        while(nowr>q[i].r)add(nowr,-1),nowr--;
        while(nowl>q[i].l)nowl--,add(nowl,1);
        while(nowr<q[i].r)nowr++,add(nowr,1);
        if(q[i].f==1)if(((now<<q[i].x)&now).any())ans[q[i].id]=1;
        else if(q[i].f==2)if(((pnow>>(maxk-q[i].x))&now).any())ans[q[i].id]=1;
        else
        
            int lim=sqrt(q[i].x);
            rep(j,1,lim)if(q[i].x%j==0&&now.test(j)&&now.test(q[i].x/j))ans[q[i].id]=1;break;
        
    
    rep(i,1,m)puts(ans[i]?"hana":"bi");
    return 0;

一些感想

之前轻视bitset的人被bitset教做人了
bitset常用操作：

#include<bitset>//bitset的头文件
bitset<6342>a; //开一个下标从0~6341的bitset
a[63]=1,a[42]=0,a.set(63,1),a.set(42,0),a.reset(42);//赋值
a.set(),a.reset();//全变成1或全变成0
((a<<634)|(a>>23)).any();//bitset可以直接进行位运算，any()返回是否存在一位是1
a.none(),a.all(),a.test(5);//是否任意一位都为0；是否任意一位都为1；是否第5位为1
a.flip(32),a[32].flip();//翻转第32位
a.flip();//翻转全部
a.count(),a.size();//返回一共有几位是1；返回一共有几位
a.to_string(),a.to_ulong(),a.to_ullong();//把bitset变成string；unsigned long；unsigned long long