208. 开关问题 —— 高斯消元

Link
高斯消元问题
其实就是对于每盏灯 $x_i$，均有一个数组$a$ 如 $a$ = {0, 1, 1, 0}表示对第 $i$ 盏灯操作，则第2，3盏灯也会变化，于是构成一个二维矩阵$a[i][j]$，$a[i][j]=1$则表示操作开关 j ,开关 i 也会变化，特别的，$a[i][i]=1$，等号右侧即为每盏灯的初始状态xor最终状态。

代码

xor+状压，用xor+状压的话就是代码会简略些，比如第i行消第j行的时候可以直接
$$a_j=a_{j}xor{a}_i$$
值得注意的是，a[i]中二进制的第0位表示增广矩阵最后一列的常数，之后的n位分别表示增广阵1~n列的系数。

int a[110];
int n;
void solve() {
    cin >> n;
    memset(a, 0, sizeof(a));
    for(int i = 1; i <= n; i++) {
        cin >> a[i];
    }
    for(int i = 1; i <= n; i++) {
        int x;
        cin >> x;
        a[i] ^= x;
        a[i] |= 1 << i;
    }
    int x, y;
    while(cin >> x >> y && x) {
        a[y] |= 1 << x;
    }
    int ans = 1;
    for(int i = 1; i <= n; i++) {
        for(int j = i + 1; j <= n; j++)
            if(a[j] > a[i]) swap(a[i], a[j]);
        if(a[i] == 0) {
            ans = 1 << (n - i + 1);
            break;
        }
        if(a[i] == 1) {
            ans = 0;
            break;
        }
        for(int k = n; k; k--)
            if(a[i] >> k & 1) {
                for(int j = 1; j <= n; j++)
                    if(i != j && (a[j] >> k & 1)) a[j] ^= a[i];
                break;
            }
    }
    if(ans == 0) puts("Oh,it's impossible~!!");
    else cout << ans << endl;
}

一开始写的，等号左侧既没有用状压也没有用xor，就是正常Gauss消元。

inline int gauss(double a[][N], bool l[], double ans[], const int& n) {
    int res = 0, r = 0;
    for(int i = 0; i < n; i++)
        l[i] = false;
    for(int i = 0; i < n; i++) {
        for(int j = r; j < n; j++)
            if(fabs(a[j][i]) > EPS) {
                for(int k = i; k <= n; k++)
                    swap(a[j][k], a[r][k]);
                break;
            }
        if(fabs(a[r][i]) < EPS) {
            ++res;
            continue;
        }
        for(int j = 0; j < n; j++)
            if(j != r && fabs(a[j][i]) > EPS) {
                double tmp = a[j][i] / a[r][i];
                for(int k = i; k <= n; k++)
                    a[j][k] -= tmp * a[r][k];
            }
        l[i] = true;
        ++r;
    }
    for(int i = 0; i < n; i++)
        if(l[i])
            for(int j = 0; j < n; j++)
                if(fabs(a[j][i]) > EPS)
                    ans[i] = a[j][n] / a[j][i];
    return res;
}
double a[N][N];
int b[N], c[N];
double ans[N];
bool l[N];
int n;
bool issolve(double a[][N], const int& n) {
    for(int i = 0; i < n; i++) {
        bool flag = true;
        for(int j = 0; j < n; j++) {
            if(a[i][j] != 0) {
                flag = false;
                break;
            }
        }
        if(flag && a[i][n] != 0) return false;
    }
    return true;
}
void solve() {
    cin >> n;
    memset(a, 0, sizeof(a));
    memset(ans, 0, sizeof(ans));
    for(int i = 0; i < n; i++) {
        cin >> b[i];
    }
    for(int i = 0; i < n; i++) {
        int x;
        cin >> x;
        b[i] ^= x;
    }
    int x, y;
    while(cin >> x >> y && x) {
        a[y-1][x-1] = 1;
    }
    for(int i = 0; i < n; i++) {
        a[i][n] = b[i];
        a[i][i] = 1;
    }
    int cnt = gauss(a, l, ans, n);
    if(!issolve(a, n)) {
        cout << "Oh,it's impossible~!!" << endl;
        return;
    }
    cnt = 1 << cnt;
    cout << cnt << endl;
}