nefu503矩阵求解+十进制高斯消元

nefu503矩阵求解+高斯消元:http://acm.nefu.edu.cn/JudgeOnline/problemShow.php?problem_id=503

题目描述:

矩阵求解

Problem:503
Time Limit:1000ms
Memory Limit:65536K

Description

给你系数矩阵和增广矩阵,让你求出矩阵代表方程的解,有唯一解直接输出就行,没解输出"no sovle!",有多个解请输出"more sovle!".

Input

数据有多组,每组第1行为N,表示是N行N列的矩阵,接下来输入N行每行N个系数,最后1行为这N行方程对应的值(Bi).

Output

有唯一解直接输出就行,没解输出"no sovle!",有多个解请输出"more sovle!".

Sample Input

5
3 1 -1 2 1
1 1 1 1 1
-1 1 2 1 -2
1 -1 -1 1 1
3 2 1 1 1
-2 3 10 -7 7

Sample Output

1 2 3 -1 -2

题目分析:

求解十进制的线性方程组,注意此题要用long long类型,防止数据有溢出~

代码实现:

#include <iostream>
#include <cstdio>

using namespace std;

const long long maxn=500;

long long a[maxn][maxn+1],x[maxn];//a 是系数矩阵和增广矩阵,x 是最后存放的解
// a[][maxn]中存放的是方程右面的值(bi)
long long equ,var;//equ 是系数阵的行数,var 是系数矩阵的列数(变量的个数)
long long free_num,ans=100000000;
long long abs1(long long num) //取绝对值
{
    if (num>=0) return num;
    else
        return -1*num;
}
void Debug(void) //调试输出,看消元后的矩阵值,提交时,就不用了
{
    long long i, j;
    for (i = 0; i < equ; i++)
    {
        for (j = 0; j < var + 1; j++)
        {
            cout << a[i][j] << " ";
        }
        cout << endl;
    }
    cout << endl;
}
inline long long gcd(long long a, long long b) //最大公约数
{
    long long t;
    while (b != 0)
    {
        t = b;
        b = a % b;
        a = t;
    }
    return a;
}
inline long long lcm(long long a, long long b) //最小公倍数
{
    return a * b / gcd(a, b);
}
void swap(long long &a,long long &b)
{
    long long temp=a;    //交换2 个数
    a=b;
    b=temp;
}
long long Gauss()
{
    long long k,col = 0; //当前处理的列
    for(k = 0; k < equ && col < var; ++k,++col)
    {
        long long max_r = k;
        for(long long i = k+1; i < equ; ++i)
            if(abs1(a[i][col]) > abs1(a[max_r][col]))//注意要求绝对值
                max_r = i;
        if(max_r != k)
        {
            for(long long i = k; i < var + 1; ++i)
                swap(a[k][i],a[max_r][i]);
        }
        if(a[k][col] == 0)
        {
            k--;
            continue;
        }
        for(long long i = k+1; i < equ; ++i)
        {
            if(a[i][col] != 0)
            {
                long long LCM = lcm(a[i][col],a[k][col]);
                long long ta = LCM/a[i][col], tb = LCM/a[k][col];
                if(abs1(a[i][col])*abs1(a[k][col]) < 0)<span style="font-size: 13.3333px; font-family: Arial, Helvetica, sans-serif;">//注意要求绝对值</span>
                    tb = -tb;
                for(long long j = col; j < var + 1; ++j)
                    a[i][j] = ( (a[i][j]*ta) - (a[k][j]*tb) ) ; //a[i][j]只有0 和1 两种状态
            }
        }
    }

//上述代码是消元的过程,行消元完成
//解下来2 行,判断是否无解
//注意K 的值,k 代表系数矩阵值都为0 的那些行的第1 行
    for(long long i = k; i < equ; ++i)
        if(a[i][col] != 0) return -1; // 无解返回-1
//Debug();
//唯一解或者无穷解,k<=var
//var-k==0 唯一解;var-k>0 无穷多解,自由解的个数=var-k
//能执行到这,说明肯定有解了,无非是1 个和无穷解的问题。
//下面这几行很重要,保证秩内每行主元非0,且按对角线顺序排列,就是检查列
    for(long long i = 0; i <equ; ++i)//每一行主元素化为非零
        if(!a[i][i])
        {
            long long j;
            for(j = i+1; j<var; ++j)
                if(a[i][j])
                    break;
            if(j == var)
                break;
            for(long long k = 0; k < equ; ++k)
                swap(a[k][i],a[k][j]);
        }
// ----处理保证对角线主元非0 且顺序,检查列完成
// free_num=k;
    if (var-k>0)
    {
//无穷多解,先枚举解,然后用下面的回带代码进行回带;
//这里省略了下面的回带的代码;不管唯一解和无穷解都可以回带,只不过无穷解
//回带时,默认为最后几个自由变元=0 而已。
        return 1;
    }
    if (var-k==0)//唯一解时
    {
//下面是回带求解代码,当无穷多解时,最后几行为0 的解默认为0;
        for(long long i = k-1; i >= 0; --i) //从消完元矩阵的主对角线非0 的最后1 行,开始往
//回带
        {
            long long tmp = a[i][var] ;
            for(long long j = i+1; j < var; ++j) //x[i]取决于x[i+1]--x[var]啊,所以后面的解对前面的解有影响。
                if(a[i][j] != 0)
                    tmp = ( tmp - a[i][j]*x[j]  ) ;
//if (a[i][i]==0) x[i]=tmp; //最后的空行时,即无穷解得
//else
            x[i] = tmp/a[i][i]; //上面的正常解
        }
//回带结束了
        return 0;
    }
}

int main()
{
    long long n;
    while(scanf("%lld",&n)!=EOF)
    {
        equ=var=n;
        for(long long i=0; i<n; i++)
        {
            for(long long j=0; j<n; j++)
            {
                scanf("%lld",&a[i][j]);
            }
        }
        for(long long i=0; i<n; i++)
        {
            scanf("%lld",&a[i][n]);
        }
        //Debug();
        long long kkk=Gauss();
        if(kkk==0)
        {
            printf("%lld",x[0]);
            for(long long i=1; i<var; i++)
            {
                printf(" %lld",x[i]);
            }
            printf("\n");
        }
        else if(kkk==1)
        {
            printf("more sovle!\n");
        }
        else
        {
            printf("no sovle!\n");
        }
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值