poj 1830 开关问题 (高斯消元模板)

最新推荐文章于 2020-12-31 17:20:39 发布

0xFFF

最新推荐文章于 2020-12-31 17:20:39 发布

阅读量589

点赞数

分类专栏：数论文章标签： poj 高斯消元

本文链接：https://blog.csdn.net/Crazy_calf/article/details/60777143

版权

数论专栏收录该内容

6 篇文章 0 订阅

订阅专栏

开关问题

Time Limit: 1000MS		Memory Limit: 30000K
Total Submissions: 8188		Accepted: 3239

Description

有N个相同的开关，每个开关都与某些开关有着联系，每当你打开或者关闭某个开关的时候，其他的与此开关相关联的开关也会相应地发生变化，即这些相联系的开关的状态如果原来为开就变为关，如果为关就变为开。你的目标是经过若干次开关操作后使得最后N个开关达到一个特定的状态。对于任意一个开关，最多只能进行一次开关操作。你的任务是，计算有多少种可以达到指定状态的方法。（不计开关操作的顺序）

Input

输入第一行有一个数K，表示以下有K组测试数据。
每组测试数据的格式如下：
第一行一个数N（0 < N < 29）
第二行 N个0或者1的数，表示开始时N个开关状态。
第三行 N个0或者1的数，表示操作结束后N个开关的状态。
接下来每行两个数I J，表示如果操作第 I 个开关，第J个开关的状态也会变化。每组数据以 0 0 结束。

Output

如果有可行方法，输出总数，否则输出“Oh,it's impossible~!!” 不包括引号

Sample Input

Sample Output

4
Oh,it's impossible~!!

Hint

第一组数据的说明：
一共以下四种方法：
操作开关1
操作开关2
操作开关3

操作开关1、2、3 （不记顺序）

将问题转化为行数和列数都是N的行列式，然后利用高斯消元求出自由变元的个数k, 则答案ans = 2^k;

#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int MAXN = 100;
int a[MAXN][MAXN];//增广矩阵
int x[MAXN];//解集
bool free_x[MAXN];//标记是否是不确定的变元
int free_rec[MAXN];//依次记录自由变元 编号从0 - var-k-1
int gcd(int a, int b){
    return b == 0 ? a : gcd(b, a%b);
}
int lcm(int a, int b){
    return a / gcd(a, b) * b;
}
int equ, var;//方程数 变元数
int Gauss()
{
    int max_r;
    int col = 0;
    int k;
    int num = 0;
    for(k = 0; k < equ && col < var; k++, col++)
    {
        max_r = k;
        for(int i = k+1; i < equ; i++)
            if(abs(a[i][col]) > abs(a[max_r][col]))
                max_r = i;
        if(max_r != k)//找到——从当前处理的列开始 交换k和max_r两行
            for(int i = col; i < var+1; i++)
                swap(a[max_r][i], a[k][i]);
        if(a[k][col] == 0)//第col列在第k行下面全是0，处理下一列
        {
            k--;
            free_rec[num++] = col;//记录自由变元
            continue;
        }
        for(int i = k+1; i < equ; i++)
        {
            if(a[i][col] != 0)
            {
                int LCM = lcm(abs(a[i][col]),abs(a[k][col]));
                int ta = LCM / abs(a[i][col]);
                int tb = LCM / abs(a[k][col]);
                if(a[i][col] * a[k][col] < 0)
                    tb = -tb;//异号
                for(int j = col; j < var+1; j++)
                    a[i][j] = a[i][j] * ta - a[k][j] * tb;
            }
        }
    }
    for(int i = 0; i < var; i++)
        free_x[i] = true;//初始化所有变元都是不确定的
    int free_x_num;//记录每行自由变元数目
    int free_index;//记录每行唯一自由变元的下标
    int temp;
    for(int i = k; i < equ; i++)
        if(a[i][col] != 0)
            return -1;
    if(k < var)
    {
        //自由变元有var - k个，即不确定的变元至少有var - k个.
        for(int i = k-1; i >= 0; i--)
        {
            free_x_num = 0;
            for(int j = 0; j < var; j++)
                if(a[i][j] != 0 && free_x[j])
                    free_x_num++, free_index = j;
            if(free_x_num > 1) continue;
            temp = a[i][var];
            for(int j = 0; j < var; j++)
                if(a[i][j] != 0 && j != free_index)
                    temp -= a[i][j] * x[j];
            x[free_index] = temp / a[i][free_index];//求出该变元
            free_x[free_index] = 0;//该变元是确定的
        }
        return var - k;//返回自由变元数目
    }
    for(int i = var-1; i >= 0; i--)
    {
        temp = a[i][var];
        for(int j = i+1; j < var; j++)
            if(a[i][j] != 0)
                temp -= a[i][j] * x[j];
        x[i] = temp / a[i][i];
    }
    return 0;
}
int main()
{
    int t, n;
    scanf("%d", &t);
    while(t--)
    {
        int a1[MAXN], b1[MAXN];
        memset(a, 0, sizeof a);
        scanf("%d", &n);
        equ = n; var = n;
        for(int i = 0; i < n; i++)
            scanf("%d", &a1[i]);
        for(int i = 0; i < n; i++)
            scanf("%d", &b1[i]);
        for(int i = 0; i < n; i++)
        {
            if(a1[i] != b1[i]) a[i][n] = 1;
            a[i][i] = 1;
        }
        int t1, t2;
        while(~scanf("%d %d", &t1, &t2) && t1 && t2){
            a[t2 - 1][t1 - 1] = 1;
        }
        int ans = Gauss();
        if(ans == -1) printf("Oh,it's impossible~!!\n");
        else printf("%d\n", 1 << ans);
    }
    return 0;
}

高斯消元模板

int a[MAXN][MAXN];//增广矩阵
int x[MAXN];//解集
bool free_x[MAXN];//标记是否是不确定的变元
int free_rec[MAXN];//依次记录自由变元 编号从0 - var-k-1
int gcd(int a, int b){
    return b == 0 ? a : gcd(b, a % b);
}
int lcm(int a, int b){
    return a / gcd(a, b) * b;
}
int equ, var;//方程数 变元数
int Gauss()
{
    int max_r;//记录当前列 绝对值最大的行号
    int col = 0;//当前处理的列
    int k;
    int num = 0;//自由变元在free_rec里面的编号
    //将增广矩阵 转变为 阶梯矩阵
    for(k = 0; k < equ && col < var; k++, col++)
    {
        /***列主消元法***/
        /*找到第col列 绝对值最大的行i(i > k)*/
        max_r = k;
        for(int i = k+1; i < equ; i++)
            if(abs(a[i][col]) > abs(a[max_r][col]))
                max_r = i;
        if(max_r != k)//找到——从当前处理的列开始 交换k和max_r两行
            for(int i = col; i < var+1; i++)
                swap(a[max_r][i], a[k][i]);
        if(a[k][col] == 0)//第col列在第k行下面全是0，处理下一列
        {
            k--;
            free_rec[num++] = col;//记录自由变元
            continue;
        }
        /***加减消元***/
        for(int i = k+1; i < equ; i++)
        {
            if(a[i][col] != 0)
            {
                /*避免浮点数出现的处理*/
                /*如果不在乎浮点数的出现，直接减就可以了*/
                int LCM = lcm(abs(a[i][col]),abs(a[k][col]));
                int ta = LCM / abs(a[i][col]);
                int tb = LCM / abs(a[k][col]);
                if(a[i][col] * a[k][col] < 0)
                    tb = -tb;//异号
                for(int j = col; j < var+1; j++)
                    a[i][j] = a[i][j] * ta - a[k][j] * tb;
            }
        }
    }
    for(int i = 0; i < var; i++)
        free_x[i] = true;//初始化所有变元都是不确定的
    int free_x_num;//记录每行自由变元数目
    int free_index;//记录每行唯一自由变元的下标
    int temp;
    /***分析解的情况***/
    // 1. 无解的情况: 化简的增广阵中存在(0, 0, ..., a)这样的行(a != 0).
    for(int i = k; i < equ; i++)
        if(a[i][col] != 0)
            return -1;
    // 2. 无穷解的情况: 在var * (var + 1)的增广阵中出现(0, 0, ..., 0)这样的行
    //说明没有形成严格的上三角阵.出现的行数即为自由变元的个数且这样的行只出现在k~equ.
    if(k < var)
    {
        //自由变元有var - k个，即不确定的变元至少有var - k个.
        for(int i = k-1; i >= 0; i--)
        {
            free_x_num = 0;//用于判断该行中不确定变元的个数，如果超过1个，则无法求解，它们仍然为不确定的变元.
            for(int j = 0; j < var; j++)
                if(a[i][j] != 0 && free_x[j])
                    free_x_num++, free_index = j;
            if(free_x_num > 1) continue;//不能求解
            //说明就只有一个不确定的变元free_index，那么可以求解出该变元，且该变元是确定的.
            temp = a[i][var];
            //求解该行唯一变元free_index
            for(int j = 0; j < var; j++)
                if(a[i][j] != 0 && j != free_index)
                    temp -= a[i][j] * x[j];
            x[free_index] = temp / a[i][free_index];//求出该变元
            free_x[free_index] = 0;//该变元是确定的
        }
        return var - k;//返回自由变元数目
    }
    //3. 唯一解的情况: 在var * (var + 1)的增广阵中形成严格的上三角阵.
    //回代求出所有解
    for(int i = var-1; i >= 0; i--)
    {
        temp = a[i][var];
        for(int j = i+1; j < var; j++)
            if(a[i][j] != 0)
                temp -= a[i][j] * x[j];
        if (temp % a[i][i] != 0)
            return -2; // 说明有浮点数解，但无整数解.
        x[i] = temp / a[i][i];
    }
    return 0;
}