[bzoj1770][Usaco2009 Nov]lights 燈——Gauss消元法
题意
给定一个无向图,初始状态所有点均为黑,如果更改一个点,那么它和与它相邻的点全部会被更改。一个点被更改当它的颜色与之前相反。
题解
第一道Gauss消元题。所谓gauss消元,就是使用初等行列式变换把原矩阵转化为上三角矩阵然后回套求解。
给定一个矩阵以后,我们考察每一个变量,找到它的系数最大的一行,然后根据这一行去消除其他的行。具体地代码如下面所示。
double a[N][N]
void Gauss(){
for(int i=1;i<=n;i++){
int r=i;
for(int j=i+1;j<=n;j++)
if(abs(a[j][i])>abs(a[r][i])) r=j;
if(r!=i) for(int j=1;j<=n+1;j++) swap(a[i][j],a[r][j]);
for(int j=i+1;j<=n;j++){
double t=a[j][i]/a[i][i];
for(int k=i;k<=n+1;k++) a[j][k]-=a[i][k]*t;
}
}
for(int i=n;i>=1;i--){
for(int j=n;j>i;j--) a[i][n+1]-=a[j][n+1]*a[i][j];
a[i][n+1]/=a[i][i];
}
}
对于xor运算,我们可以使用同样的方法消元。
另外,xor的话可以使用bitset压位以加速求解。
代码(附有详细注释)
#include <algorithm>
#include <cstdio>
const int maxn = 45;
int a[maxn][maxn], b[maxn];
int n, m, tot, mn = 0x3f3f3f;
void gauss() {
for (int i = 1; i <= n; i++) { //依次考察每一个未知数
int j = i; //开始选中第i行
while (j <= n && !a[j][i]) //选中系数最大的一行(减小精度误差)
j++;
if (j > n)
continue;
if (i != j)
for (int k = 1; k <= n + 1; k++) //交换两行,使得第i行成为最大系数
std::swap(a[j][k], a[i][k]);
for (int j = 1; j <= n;
j++) // gauss消元核心代码:使用第i行消除所有行的第i个未知数
if (i != j && a[j][i]) //以此来形成一个上三角矩阵,为之后的消元作准备
for (int k = 1; k <= n + 1; k++)
a[j][k] ^=
a[i][k]; //如果是普通的线性方程组,这里需要使用别的方法把系数置零
}
}
void dfs(int now) { //由于gauss消元后有一些自由元,我们需要进行最优解暴力搜索
if (tot >= mn)
return;
if (!now) {
mn = std::min(mn, tot);
return;
}
if (a[now][now]) { //确定的情况
int t = a[now][n + 1];
for (int i = now + 1; i <= n; i++)
if (a[now][i])
t ^= b[i]; //由于是上三角矩阵,所以逆向消元
b[now] = t;
if (t)
tot++;
dfs(now - 1);
if (t)
tot--; //回溯
} else { //自由元的情况,随意确定
b[now] = 0;
dfs(now - 1);
b[now] = 1;
tot++;
dfs(now - 1);
tot--;
}
}
int main() {
#ifdef D
freopen("input", "r", stdin);
#endif
scanf("%d %d", &n, &m);
for (int i = 1; i <= n; i++)
a[i][i] = 1, a[i][n + 1] = 1;
for (int i = 1; i <= m; i++) {
int x, y;
scanf("%d %d", &x, &y);
a[x][y] = a[y][x] = 1;
}
gauss(); //判定是否无解:系数矩阵全0,常数矩阵不全为0
dfs(n);
printf("%d\n", mn);
return 0;
}
附:如何使用bitset
首先,声明bitset:
#include <bitset>
using std::bitset;
初始化:
bitset<n> b;
bitset<n> b(unsigned long u);
上述语句声明了一个n位全部为0的bitset,第二个语句用一个unsigned long long变量去初始化bitset。
bitset的更多操作:
b1 = b2 & b3;//按位与
b1 = b2 | b3;//按位或
b1 = b2 ^ b3;//按位异或
b1 = ~b2;//按位补
b1 = b2 << 3;//移位