CodeForces - 633G Yash And Trees 线段树+dfs序+bitset移位

题目链接：点击查看

题意：n个点的一棵树，每个点都有一个权值，q次询问，1 v x,表示 1 这个子树上的所有节点权值都加x，2 v表示查询2这个子树上每个节点的权值表示成 au = p + m·k. 有多少种质数p，也就是mod m 后 有多少种质数，这个m最大为1000，直接开数组的话，

nlog（n）*m是不行的，这里权值加多少也就相当于移位 一个m长的0 1串s，权值+x时，也就是s<<x | s>>(m-x),这里的s我们就可以用bitset来实现，复杂度直接缩小了30倍    ->bitset:点击查看

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
struct node{
	int l,r;
	int laz;
	bitset<1010> s;
}tree[N<<2];
int n,m,q;
int a[N];
int p[N],in[N],out[N],cnt;
vector<int> v[N];
void dfs(int u,int fa)
{
	in[u]=++cnt;
	p[cnt]=u;
	for(int i=0;i<v[u].size();i++)
	{
		int to=v[u][i];
		if(to==fa) continue;
		dfs(to,u);
	}
	out[u]=cnt;
}
void pushup(int cur)
{
	tree[cur].s=tree[cur<<1].s|tree[cur<<1|1].s;
}
void build(int l,int r,int cur)
{
	tree[cur].l=l;
	tree[cur].r=r;
	tree[cur].laz=0;
	if(l==r)
	{		
		tree[cur].s[a[p[l]]%m]=1;
		return;
	}
	int mid=(r+l)>>1;
	build(l,mid,cur<<1);
	build(mid+1,r,cur<<1|1);
	pushup(cur);
//	cout<<l<<" "<<r<<endl;
//	for(int i=0;i<=15;i++) cout<<tree[cur].s[i];
//	cout<<endl;
}
void pushdown(int cur)
{
	if(tree[cur].laz)
	{
		tree[cur<<1].laz+=tree[cur].laz;
		tree[cur<<1].laz%=m;
		
		tree[cur<<1|1].laz+=tree[cur].laz;
		tree[cur<<1|1].laz%=m;
		
		tree[cur<<1].s=(tree[cur<<1].s<<tree[cur].laz)|(tree[cur<<1].s>>(m-tree[cur].laz));
		tree[cur<<1|1].s=(tree[cur<<1|1].s<<tree[cur].laz)|(tree[cur<<1|1].s>>(m-tree[cur].laz));
		
		tree[cur].laz=0;
	}
}
void update(int pl,int pr,int l,int r,int cur,int val)
{
	if(pl<=l&&r<=pr)
	{
		tree[cur].laz+=val;
		tree[cur].laz%=m; 
		tree[cur].s=(tree[cur].s<<val) | (tree[cur].s>>(m-val)); 
//	cout<<l<<" "<<r<<endl;
//	for(int i=0;i<=15;i++) cout<<tree[cur].s[i];
//	cout<<endl;
		return;
	}
	pushdown(cur);
	int mid=(r+l)>>1;
	if(pl<=mid) update(pl,pr,l,mid,cur<<1,val);
	if(pr>mid) update(pl,pr,mid+1,r,cur<<1|1,val);
	pushup(cur); 
//	cout<<l<<" "<<r<<endl;
//	for(int i=0;i<=15;i++) cout<<tree[cur].s[i];
//	cout<<endl;
}
bitset<1010> query(int pl,int pr,int l,int r,int cur)
{
	if(pl<=l&&r<=pr)
	{
		return tree[cur].s;
	}
	pushdown(cur);
	int mid=(r+l)>>1;
	bitset<1010> s(0);
	if(pl<=mid) s|=query(pl,pr,l,mid,cur<<1);
	if(pr>mid) s|=query(pl,pr,mid+1,r,cur<<1|1);
	pushup(cur);
	return s;
	
}
bitset<1010> prime;
void init()
{
	prime[0]=1;
	prime[1]=1;
	for(int i=2;i<=1000;i++)
	{
		if(prime[i]==0)
		{
			for(int j=i+i;j<=1000;j+=i)
				prime[j]=1;
		}
	}
	prime.flip();
}
int main()
{
	init();
	scanf("%d%d",&n,&m);
	for(int i=m;i<=1009;i++) prime[i]=0;
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	int x,y;
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&x,&y);
		v[x].push_back(y);
		v[y].push_back(x);
	}
	dfs(1,0);
	build(1,n,1);
	int op;
	scanf("%d",&q);
	while(q--)
	{
		scanf("%d",&op);
		if(op==1)
		{
			scanf("%d%d",&x,&y);
			y%=m;
			update(in[x],out[x],1,n,1,y);
		}
		else
		{
			scanf("%d",&x);
			bitset<1010> 
			s=query(in[x],out[x],1,n,1);
			s=s&prime;
			printf("%d\n",s.count());
		}
	}
	return 0;
}