首页 > 代码库 > 【bzoj4026】dC Loves Number Theory 可持久化线段树

【bzoj4026】dC Loves Number Theory 可持久化线段树

题目描述

dC 在秒了BZOJ 上所有的数论题后,感觉萌萌哒,想出了这么一道水题,来拯救日益枯竭的水题资源。 
给定一个长度为 n的正整数序列A,有q次询问,每次询问一段区间内所有元素乘积的φ(φ(n)代表1~n 中与n互质的数的个数) 。由于答案可能很大,所以请对答案 mod 10^6 + 777。 (本题强制在线,所有询问操作的l,r都需要 xor上一次询问的答案 lastans,初始时,lastans = 0) 

输入

第一行,两个正整数,N,Q,表示序列的长度和询问的个数。 

第二行有N 个正整数,第i个表示Ai. 
下面Q行,每行两个正整数,l r,表示询问[l ^ lastans,r ^ lastans]内所有元素乘积的φ 

输出

Q行,对于每个询问输出一个整数。 

样例输入

5 10
3 7 10 10 5
3 4
42 44
241 242
14 9
1201 1201
0 6
245 245
7 7
6 1
1203 1203

样例输出

40
240
12
1200
2
240
4
4
1200
4


题解

可持久化线段树(个人觉得不应该叫做主席树,因为本题是区间线段树)

前置技能:HH的项链 强制在线

标准的做法是离线+树状数组,但是由于要在线处理,所以只能尽量使用可持久化数据结构。

我们还是按照那道题的思路,对于每个数,当它出现时就把它上一次出现的位置删除掉。

那么我们就可以使用可持久化区间线段树完成这样类似的操作。

对于每个数a[i],记录一下它上一个出现的位置last,在i对应的线段树中,把last位置的数-1,并把i位置的数+1。

这样在查询区间[l,r]的时候,在r对应的线段树中,查询[l,r]([l,n]同理)这段区间内数的和即为答案。

比较复杂,稍微理解一下~

就当是理解完了

回过头来看这道题,要求区间乘积的欧拉函数。

欧拉函数是什么请自行百度,怎么求请自行找博客。

这里还是简单说一下吧,$\varphi(x)=x\sum\limits_{prime(i)\& i|x}\frac{i-1}i$。

所以说对于这道题可以维护一个前缀乘积,然后只需要求出区间内的质因子即可。

使用类似于HH的项链的方法,对于一个位置i,枚举a[i]的质因子,如果它在之前last位置出现过,则将i位置对应的线段树的last位置的数除以$\frac{i-1}i$;在i位置的数乘以$\fraci-1{i}$。

然后使用可持久化线段树维护区间乘积,查询[l,r]时求出r对应的线段树中[l,r]([l,n]同理)的乘积,再乘上$\frac{mul[r]}{mul[l-1]}$(前缀积)即为答案。

需要筛逆元什么的,参见代码。

#include <cstdio>#include <algorithm>#define N 100010#define M 2100000using namespace std;typedef long long ll;const ll mod = 1000777 , k = 1000005;ll a[N] , prime[M] , cnt , last[M] , ls[N * 80] , rs[N * 80] , tot , root[N] , inv[M] , sum[N * 80] , mul[N];bool np[M];void update(ll p , ll v , ll l , ll r , ll x , ll &y){	y = ++tot , sum[y] = sum[x] * v % mod;	if(l == r) return;	ll mid = (l + r) >> 1;	if(p <= mid) rs[y] = rs[x] , update(p , v , l , mid , ls[x] , ls[y]);	else ls[y] = ls[x] , update(p , v , mid + 1 , r , rs[x] , rs[y]);}ll query(ll p , ll l , ll r , ll x){	if(l >= p) return sum[x];	if(r < p) return 1;	ll mid = (l + r) >> 1;	return query(p , l , mid , ls[x]) * query(p , mid + 1 , r , rs[x]) % mod;}int main(){	ll n , m , i , j , x , y;	ll lastans = 0;	scanf("%lld%lld" , &n , &m);	for(i = 2 ; i <= k ; i ++ )	{		if(!np[i]) prime[++cnt] = i;		for(j = 1 ; j <= cnt && i * prime[j] <= k ; j ++ )		{			np[i * prime[j]] = 1;			if(i % prime[j] == 0) break;		}	}	mul[0] = sum[0] = inv[1] = 1;	for(i = 2 ; i < mod ; i ++ ) inv[i] = (mod - mod / i) * inv[mod % i] % mod;	for(i = 1 ; i <= n ; i ++ )	{		scanf("%lld" , &a[i]) , mul[i] = mul[i - 1] * a[i] % mod , root[i] = root[i - 1];		for(j = 1 ; prime[j] * prime[j] <= a[i] ; j ++ )		{			if(a[i] % prime[j] == 0)			{				if(!last[prime[j]]) update(i , (prime[j] - 1) * inv[prime[j]] % mod , 1 , n , root[i] , root[i]);				else update(last[prime[j]] , prime[j] * inv[prime[j] - 1] % mod , 1 , n , root[i] , root[i]) , update(i , (prime[j] - 1) * inv[prime[j]] % mod , 1 , n , root[i] , root[i]);				last[prime[j]] = i;				while(a[i] % prime[j] == 0) a[i] /= prime[j];			}		}		if(a[i] > 1)		{			if(!last[a[i]]) update(i , (a[i] - 1) * inv[a[i]] % mod , 1 , n , root[i] , root[i]);			else update(last[a[i]] , a[i] * inv[a[i] - 1] % mod , 1 , n , root[i] , root[i]) , update(i , (a[i] - 1) * inv[a[i]] % mod , 1 , n , root[i] , root[i]);			last[a[i]] = i;		}	}	while(m -- ) scanf("%lld%lld" , &x , &y) , x ^= lastans , y ^= lastans , printf("%lld\n" , lastans = query(x , 1 , n , root[y]) * mul[y] % mod * inv[mul[x - 1]] % mod);	return 0;}

 

 

【bzoj4026】dC Loves Number Theory 可持久化线段树