Codeforces Round #705 (Div. 2)D. GCD of an Array（线段树动态开点or set+map）

Codeforces Round #705 (Div. 2)D. GCD of an Array（线段树动态开点or set+map）

题意：

有n个数，每回合可以对某个位置的数乘x，并求出所有数的gcd

思路：

把a1质因数分解，pi表示各质因子，xi表示幂

$a1={p_1}^{x_{11}}\ast {p_2}^{x_{12}}\ast {p_3}^{x_{13}}$…

$a2={p_1}^{x_{21}}\ast {p_2}^{x_{22}}\ast {p_3}^{x_{23}}$…

$...........$

$gcd={p_1}^{min(x_{11},x_{21}......)}\ast {p_2}^{min(x_{12},x_{22}......)}$

1.线段树动态开点：暴力，以每个质因子为根，建立一颗线段树，维护最小值。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 2e5 + 5, mod = 1e9 + 7;
int a[maxn], n, q;
int p[maxn], isp[maxn], tot, minfact[maxn];
ll qpow(ll base, ll p = mod - 2){
    ll ret = 1;
    for (; p; p >>= 1)
    {
        if (p & 1)
            ret = ret * base % mod;
        base = base * base % mod;
    }
    return ret;
}
void make_prime(){
    for(int i=2;i<=maxn-2;i++){
        if (!isp[i])
            p[++tot] = i, minfact[i] = tot;
        for (int j = 1; i * p[j] < maxn; ++j){
            isp[i * p[j]] = 1;
            minfact[i * p[j]] = j;//维护一个数的最小质因子以再在分解时避免遍历无用的质数
            if (i % p[j] == 0) break;
        }
    }
}
const int maxp = 2e7 + 5;//空间往不会炸的卡是绝对没错的
int cnt, lson[maxp], rson[maxp], val[maxp];//动态开点线段树
int root[maxn];
void modify(int &now, int l, int r, int x, int add)
{
    if (!now)now = ++cnt;
    if (l == r){
        val[now] += add;
        return;
    }
    int mid = (l + r) >> 1;
    if (x <= mid)
        modify(lson[now], l, mid, x, add);
    else modify(rson[now], mid + 1, r, x, add);
    val[now] = min(val[lson[now]], val[rson[now]]);
}
int main(){
    make_prime();
    scanf("%d%d",&n,&q);
    ll gcd = 1;
    for(int i=1,tmp;i<=n;i++){
    	scanf("%d",&tmp);
        for (int j = minfact[tmp]; tmp > 1;)//j 的初始值要是最小质因子，避免无谓的循环
        {
            int k = 0;
            while (tmp > 1 && tmp % p[j] == 0)
                ++k, tmp /= p[j];
            gcd = gcd * qpow(qpow(p[j], val[root[j]])) % mod;//首先 gcd 删除 p[j] 的贡献
            modify(root[j], 1, n, i, k);
            gcd = gcd * qpow(p[j], val[root[j]]) % mod;//然后把贡献乘回去
            if (tmp > 1) j = minfact[tmp];//如果不这么写是会 FST 的
        }
    }
    while (q--){
    	int x,v;
    	cin>>x>>v;
        for (int j = minfact[v]; v > 1;){
            while (v > 1 && v % p[j]) ++j;
            int k = 0;
            while (v > 1 && v % p[j] == 0)
                ++k, v /= p[j];
            gcd = gcd * qpow(qpow(p[j], val[root[j]])) % mod;
            modify(root[j], 1, n, x, k);
            gcd = gcd * qpow(p[j], val[root[j]]) % mod;
            if (v > 1) j = minfact[v];
        }
        printf("%lld\n", gcd);
    }
}

2.unordered_map+multiset：

unordered_map<int,int>mp[maxn];//每个位置的质因子的数量

multisetst[maxn];//每个质因子 在每个位置的数量

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e5+10;
const ll mod=1e9+7;
int n,m,gcd=1,tot,isp[maxn],prime[maxn];
unordered_map<int,int>mp[maxn];//每个位置的质因子的数量 
multiset<int>st[maxn];//每个质因子 在每个位置的数量 
ll ksm(ll x,ll y){
	ll ans=1;
	while(y){
		if(y&1)ans=ans*x%mod;
		y>>=1;
		x=x*x%mod;
	}
	return ans;
}
void make_prime(){
	for(int i=2;i<=maxn;i++){
		if(!isp[i])prime[++tot]=i;
		for(int j=1;j<=tot&&i*prime[j]<maxn;j++){
			isp[i*prime[j]]=1;
			if(i%prime[j]==0)break;
		}
	}
}
void update(int op,int x,int num){
	if(mp[op][x]){
		if(st[x].size()==n)gcd=1ll*gcd*ksm(ksm(x,*st[x].begin()),mod-2)%mod;
		st[x].erase(st[x].find(mp[op][x]));
    }
	mp[op][x]+=num;
	st[x].insert(mp[op][x]);
	if(st[x].size()==n)gcd=1ll*gcd*ksm(x,*st[x].begin())%mod;
}
void solve(int k,int x)
{
	for(int i = 2;i <= x/i;i++)
		if(x%i == 0)
		{
			int cnt = 0;
			while(x%i == 0) cnt++,x /= i;
			update(k,i,cnt);
		}
	if(x>1) update(k,x,1);
}
int main(){
	make_prime();
    scanf("%d%d",&n,&m);
    for(int i=1,x;i<=n;i++){
    	scanf("%d",&x);
    	solve(i,x);
	}
	while(m--){
		int x,op;
		scanf("%d%d",&op,&x);
		solve(op,x);
		printf("%d\n",gcd);
	}
}