C. Square Subsets(状压+素因子分解)

题目

题意：

    给定n个数，要求选出任意多的数，使得这些数的乘积为平方数。
    $1\le n\le 10^5,1\le a_i\le 70$

分析：

    由于这些数都很小，所以我们必然是在它们身上做文章。对于完全平方数，它的素因子分解后每个素因子的指数均为偶数。所以我们就可以对这些数进行素因子分解来探讨了，由于70以内的素数只有19个，这么小的数字自然而然就想到了状压。
    每个位数上维护对应的素因子的指数为奇数和偶数即可。我们依次考虑每一个数，所以就是用dp计数了。我们考虑一个新的数字，这个数字选偶数个对结果不影响，选奇数就是异或上它对于的那个数字状态，选奇数个和偶数个的方案数均为2^(num [i] -1)种。所以dp[i][j] = 2^(num[i]-1)*(dp[i-1][j]+dp[i-1][j ^ tmp])，tmp为i这个数字对应的状态。由于空间不能开那么大，所以滚动数组压缩一下即可。

#include <iostream>
#include <vector>
using namespace std;

typedef long long ll;

int num[75];
ll dp[2][(1<<20)],pow[100005];
ll mod = 1e9 + 7;
vector<int> prime;

void init(int n)
{
	for (int i = 2; i <= 70; i++)
	{
		int flag = 0;
		for (int j = 2; j * j <= i; j++)
		{
			if( i % j == 0 ) 
			{
				flag = 1;
				break;
			}
		}
		if( flag == 0 ) prime.push_back(i); 
	}
	pow[0] = 1;
	for (int i = 1; i <= n; i++)
	{
		pow[i] = pow[i-1] * 2;
		pow[i] %= mod; 
	}
}

int divide(int n)
{
	int res = 0;
	for (int i = 0; i < prime.size(); i++)
	{
		int t = 0;
		if( n % prime[i] == 0 )
		{
			while( n % prime[i] == 0 )
			{
				t ^= 1;
				n /= prime[i];
			}
			res += (1<<i)*t;
		}
	}
	return res;
}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	int n;
	cin >> n;
	init(n);
	for (int i = 1; i <= n; i++)
	{
		int x;
		cin >> x;
		num[x] ++;
	}
	dp[0][0] = 1; 
	int now = 0,pre = 1;
	for (int i = 1; i <= 70; i++)
	{
		if( num[i] == 0 ) continue;
		int t = divide(i);
		swap(now,pre);
		for (int j = 0; j < (1<<20); j++)
		{
			dp[now][j] = pow[num[i]-1] * (dp[pre][j] + dp[pre][j^t]);
			dp[now][j] %= mod;
		}
	}
	cout << (dp[now][0] - 1 + mod) % mod << '\n';
	return 0;
}