Codeforces Round #694 (Div. 2) D. Strange Definition 质因子分解 + 平方数

最新推荐文章于 2022-11-22 23:15:14 发布

原创最新推荐文章于 2022-11-22 23:15:14 发布 · 123 阅读

0 ·

CC 4.0 BY-SA版权

Codeforces 同时被 2 个专栏收录

77 篇文章

订阅专栏

数学

60 篇文章

订阅专栏

本文探讨了一种算法问题，涉及到相邻数的定义，即两个数的最简公倍数除以最大公约数是一个平方数。文章通过分析完全平方数的质因数特性，提出将质因子的幂次模2来判断两个数是否相邻。接着，通过动态维护质因子计数，解决询问最大相邻数个数的问题。文章提供了高效的解决方案，包括预处理质因子和优化计算过程。

传送门

题意： 定义相邻数为 $\frac{lcm(x,y)}{gcd(x,y)}$ 是一个平方数，则 $x$ 和 $y$ 是相邻的。现在给出q个询问，每次询问一个 $i$ ，表示询问第 $i$ 秒后 $max_{1<=i<=n}d_i$ ， $d_i$ 表示与 $a_i$ 为相邻数的个数。注意每一秒数组都会改变成与 $a_i$ 是相邻数的乘积。

思路： 先考虑怎么化简 $\frac{lcm(x,y)}{gcd(x,y)}$ ，我们知道 $lcm(x,y)=\frac{x*y}{gcd(x,y)}$ ，带入得 $\frac{x*y}{gcd(x,y)^2}$ ，继续化简 $(\frac{\sqrt{x*y}}{gcd(x,y)})^2$ ，所以我们可以知道如果两个数是相邻的，那么他们乘积一定是完全平方数。我们需要知道完全平方数有什么特点，显然他们的质因子的幂数都是偶数，那么两个数相乘为平方数需要满足什么条件呢？也比较容易想到他们的质因子幂数相加为偶数，那么我们如果在乘之前就把他们的质因子的幂数模2，得到两个新的数，这两个新数必须相等的时候乘起来才能是平方数。那么这就比较显然了，我们可以用 $m p [i]$ 表示 $i$ 出现的次数，第0秒的时候只需要取一个 $m a x$ 即可。考虑第零秒到第二秒什么变了，什么没变。我们下面把相等的数的集合称为一个团，且以下的数的质因子幂数默认是模2意义下的。 如果一个团的数量为偶数，那么把他们乘起来之后，他们的质因数的幂数都为偶数，就变成1了。如果一个团的数量为奇数，乘起来之后的质因数的幂数仍为奇数。让后特殊处理下1的情况就好啦。可以发现从第1秒之后他们的数量不会变，只需要先得到第零秒的，再得到偶数变乘1的数量，让后加上原来1的数量，与第零秒取 $m a x$ 即可。
让后分解质因子肯定不能 $\sqrt{n}$ 分解，这里可以记一下每个数的最小质因数，让后 $l o g n$ 分解即可。

//#pragma GCC optimize(2)
#include<cstdio>
#include<iostream>
#include<string>
#include<cstring>
#include<map>
#include<cmath>
#include<cctype>
#include<vector>
#include<set>
#include<queue>
#include<algorithm>
#include<sstream>
#include<ctime>
#include<cstdlib>
#define X first
#define Y second
#define L (u<<1)
#define R (u<<1|1)
#define pb push_back
#define mk make_pair
#define Mid (tr[u].l+tr[u].r>>1)
#define Len(u) (tr[u].r-tr[u].l+1)
#define random(a,b) ((a)+rand()%((b)-(a)+1))
#define db puts("---")
using namespace std;

//void rd_cre() { freopen("d://dp//data.txt","w",stdout); srand(time(NULL)); }
//void rd_ac() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//AC.txt","w",stdout); }
//void rd_wa() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//WA.txt","w",stdout); }

typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int,int> PII;

const int N=2000010,mod=1e9+7,INF=0x3f3f3f3f;
const double eps=1e-6;

int n;
int prime[N],cnt;
int com[N];
bool st[N];
map<int,int>mp;

void get_prime(int n)
{
    for(int i=2;i<=n;i++)
    {
        if(!st[i]) prime[cnt++]=i,com[i]=i;
        for(int j=0;prime[j]<=n/i;j++)
        {
            st[prime[j]*i]=true;
            com[prime[j]*i]=min(com[prime[j]*i],prime[j]);
            if(i%prime[j]==0) break;
        }
    }
}

void divide(int x)
{
    map<int,int>has;
    while(x!=1)
    {
        has[com[x]]++;
        x/=com[x];
    }
    int ans=1;
    for(auto x:has) if(x.Y%2==1) ans*=x.X;
    mp[ans]++;
}

int main()
{
//	ios::sync_with_stdio(false);
//	cin.tie(0);

    memset(com,0x3f,sizeof(com));
    get_prime(1000000);
    int _; scanf("%d",&_);
    while(_--)
    {
        scanf("%d",&n); mp.clear();
        for(int i=1;i<=n;i++)
        {
            int x; scanf("%d",&x);
            divide(x);
        }
        int ans1,ans2; ans1=ans2=0;
        for(auto x:mp)
        {
            ans1=max(ans1,x.Y);
            if(x.X==1) ans2+=x.Y;
            else if(x.Y%2==0) ans2+=x.Y;
        }
        int q; scanf("%d",&q);
        while(q--)
        {
            LL op; scanf("%lld",&op);
            if(op==0) printf("%d\n",ans1);
            else printf("%d\n",max(ans1,ans2));
        }
    }


	return 0;
}
/*

*/