题意:给定n个数,求取出任意个数的乘积是平方数的取法数。
n<=100,每个数不超过10^15,而且每个数的最大素因子不超过500
解法:
结论:平方数的所有素因子的幂次都是偶数
题目中每个数的素因子都不超过500,因此只要考虑每个素因子在各个数中奇偶性,每次只能选择偶数个含奇次素因子的数。
xi=1,代表选择第i个数,xi=0,代表不选第i个数。
对每个素因子如果第i1,i2…ik个数的幂次为奇,则
xi1
^
xi2
^…^
xik
=0。由此得到p个异或方程,p是n个数中所有的素因子的并。
用gauss消元可以解出方程。复杂度O(n^3)
拓展:
1.平方数改为p次方(p为任意素数)如何做?
增加限制条件:每个数最多可以取p-1次。
与上面做法类似xi=j,代表选择第i个数j次。对于任意素因子d,如果xi的幂次%p=ai,则方程中xi前的系数为ai,于是得到方程:
∑ai∗xi=0(modp)
方程个数是不同素因子的总数
于是得到了模方程组!如何解模线性方程组?
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
#include<map>
#include<vector>
#include<queue>
#include<stack>
#define ll long long
using namespace std;
const int maxp = 500+10;
typedef ll Matrix[maxp][maxp];
ll t,n;
ll x;
ll tmpn;
bool not_prime[maxp+10];
vector<ll> prime,factor;
Matrix aim;
void make_prime(){
not_prime[0]=not_prime[1]=1;
prime.clear();
for(ll i=2;i<maxp;i++){
if(!not_prime[i]) prime.push_back(i);
for(ll j=0;j<prime.size()&&i*prime[j]<maxp;j++){
not_prime[i*prime[j]]=1;
if(!(i%prime[j])) break;
}
}
}
void divided(ll x,ll j){
for(ll i=0;i<prime.size();i++){
while(x%prime[i]==0){
aim[i][j]^=1;
tmpn=max(tmpn,i);
x/=prime[i];
}
}
}
int gauss(Matrix A,ll m,ll n){//m行n列矩阵,n+1列为b向量
int i=0,j=0,r;
while(i<m&&j<n){
r=i;
for(ll k=i;k<m;k++) if(A[k][j]) { r=k; break; }
if(A[r][j]){
if(r!=i) for(ll jj=0;jj<=n;jj++) swap(A[i][jj],A[r][jj]);
for(ll ii=i+1;ii<m;ii++){
if(A[ii][j]){
for(ll jj=i;jj<=n;jj++){//此处i改为j似乎也可以,因为j列之前都已经为0了(i<=j)
A[ii][jj]^=A[i][jj];
}
}
}
i++;
}
j++;
}
return i;
}
int main(){
make_prime();
//freopen("a.txt","r",stdin);
scanf("%lld",&t);
while(t--){
scanf("%lld",&n);
memset(aim,0,sizeof(aim));
tmpn=-1;
for(ll i=0;i<n;i++){
scanf("%lld",&x);
divided(x,i);
}
ll r=gauss(aim,tmpn+1,n);
printf("%lld\n",(1LL<<(n-r))-1);
}
}