HDU 5833 -- 高斯消元法+XOR方程组

题目链接: http://acm.hdu.edu.cn/showproblem.php?pid=5833

题目大意：给你一堆数，让你从这些数中随机的选取一些数，使得这些数的乘积所得结果为一个平方数 (可以只选一个数)，问有多少种选法，最后取模

思路：依据质因数<=2000这个点，用01向量表示一个数，再用n个01变量Xi来表示我们的选择，其中xi = 1表示要选的第i个数，xi = 0 表示不选它，则可对每个素数的幂列出一个模2的方程。

举个例子：以{4, 6, 10, 15}为例， 它们的质因子只有2, 3, 5这三种，改写成01向量形式, 4 ->  (2, 0, 0)，6 -> (1, 1, 0), 10 -> (1, 0, 1) , 15 -> (0, 1, 1), 所以我们可以知道，选出来的数乘积为 2^(2x1+x2+x3), 3^(x2+x4), 5^(x3+x4), 如果让这个数为一个平方数，那么每个幂都应该是偶数，并且我们可以进一步优化一下，将原本偶数的部分直接看成0，奇数部分的系数一律看成1，那么我们就可以的到如下的xor方程组

 x2 xor x3 = 0        x2 xor x4 = 0        x3 xor x4 = 0,   我们就可以转换成矩阵，求秩来做

AC代码：

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
const int maxn = 2005;
const int maxp = 305;
const ll mod = 1000000007;

ll vis[maxn], prime[maxn], a[maxp];
ll A[maxn][maxp];
ll cnt, n;

void init()
{
    cnt = 0;
    memset(vis, 0, sizeof(vis));
    memset(prime, 0, sizeof(prime));
    for (int i = 2; i <= 2000; ++i) {
        if(!vis[i]) prime[++cnt] = i;
        for (int j = i; j <= 2000; j += i) vis[j] = 1;
    }
}

int ranks()
{
    int i = 0, j = 0, k, r, u;
    while (i <= cnt && j <= n) {
        r = i;
        while (!A[r][j] && r <= cnt)  r++;
        if(A[r][j]){
            swap(A[i], A[r]);
            for(u = i+1; u <= cnt; ++u) {
                if(A[u][j]) {
                    for(k = i; k <= n; ++k) A[u][k] ^= A[i][k];
                }
            }
            i++;
        }
        j++;
    }
    return i;
}

int main()
{
    init();
    int T, cas = 1;
    cin >> T;
    while (T--) {
        scanf("%lld", &n);
        memset(A, 0, sizeof(A));
        for (int i = 1; i <= n; ++i) {
            scanf("%lld", &a[i]);
        }
        for (int i = 1; i <= n; ++i) {
            for (int j = 1; j <= cnt; ++j) {
                 ll num = a[i];
                 if(num % prime[j] == 0){
                    while (num % prime[j] == 0) {
                        num /= prime[j];
                        A[j][i] ^= 1;
                    }
                 }
            }
        }
        printf("Case #%d:\n",cas);
        ll r = n - ranks();
        ll ans = 1;
        for(int i = 1; i <= r; i++) ans = (ans*2)%mod;
        cout << ans-1 << '\n';
        cas++;
    }
    return 0;
}

更多带图讲解见蓝书P160