HDU 5833 Zhu and 772002 高斯消元

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=5833
题意：给你N个数，取其中K个数相乘，问有多少种选择方法似的乘机为一个完全平方数

对每一个因子做一个方程，要求

对方程组 高斯消元，
有唯一解方案数为0（因为所有的数都不选得到的一定是一个完全平方数）
方程有多解，假设自由变元有K个那么方案数为2的K次方 - 1（减去全都不选的方案）

代码：

#include <bits/stdc++.h>
#define sf scanf
#define pf printf

using namespace  std;
const int maxn = 2000 + 50,mod = (int)(1e9) + 7;
typedef long long LL;
typedef int Matrix[maxn][maxn];
Matrix A;
LL Gauss(Matrix A,int n,int m){
    int i,j,k,p;
    for(i = j = 0;i < n && j < m;++j){
        k = i;
        for(p = i + 1;p < n;++p) if(A[p][j] > A[k][j]) k = p;
        if(A[k][j]){
            for(p = 0;p <= m;++p) swap(A[k][p],A[i][p]);
            for(k = i + 1;k < n;++k){
                if(A[k][j]){
                    for(p = j;p <= m;++p) A[k][p] = ( (A[k][p] - A[i][p]) % 2  + 2)  % 2;
                }
            }
            ++i;
        }
    }
    for(k = i;k < n;++k) if(A[k][m]) return 0;
    LL ret = 1;
    for(p = 0;p < m - i;++p) ret = ret * 2 % mod;
    return ret;
}
const int maxNode = 300 + 5;
int n;
LL Ar[maxNode];
void INIT(){
    memset(A,0,sizeof A);
    for(int i = 0;i < n;++i){
        for(int j = 2;j <= 3000;++j){
            if(Ar[i] % j == 0){
                int cnt = 0;
                while(Ar[i] % j == 0){
                    cnt++;
                    Ar[i] /= j;
                }
                A[j][i] = (cnt & 1);
            }
        }
    }
}
int main(){
    int T,ca = 0;sf("%d",&T);
    while(T--){
        sf("%d",&n);
        for(int i = 0;i < n;++i) sf("%lld",&Ar[i]);
        INIT();
        pf("Case #%d:\n",++ca);
        LL ans = Gauss(A,2000 + 1,n);
        pf("%lld\n",( (ans - 1) % mod  + mod) % mod );
    }
}