并不对劲的复健训练-p3674

题目大意

给出序列$ 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 quest{int 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