洛谷P4783 (高斯消元求矩阵的逆 模板)

/**
    高斯消元矩阵求逆
**/
#include <cstdio>
#include <stack>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <iostream>
#include <map>
#include <vector>
#include <queue>
#include <set>
#define eps 1e-8
typedef long long ll;
const double PI = acos(-1.0);
const int maxn = 1005;
const int INF = 0x3f3f3f;
const ll mod = 1e9+7;
const ll linf = 0x3f3f3f3f3f3f3f3f;
using namespace std;
int n,m;
ll f[maxn][maxn];
ll r,ret;
ll ksm(ll a, ll b)//求逆元
{
    ret = 1;
    while(b)
    {
        if(b&1)
            ret = ret*a%mod;
        a = a*a%mod;
        b >>= 1;
    }
    return ret;
}
int main()
{
    scanf("%d",&n);
    m = n*2;
    for(int i = 1; i<=n; i++)
    {
        for(int j = 1; j<=n; j++)
        {
            scanf("%lld",&f[i][j]);
        }
        f[i][n+i] = 1;
    }
    //高斯消元模板
    for(int i = 1; i<=n; i++)
    {
        for(int j = i; j<=n; j++)
        {
            //找到xi的系数不为0的一个方程
            if(f[j][i])
            {
                for(int k = 1; k<=m; k++)
                    swap(f[i][k],f[j][k]);
                break;
            }
        }
        if(!f[i][i])
        {
            cout<<"No Solution"<<endl;
            return 0;
        }
        r = ksm(f[i][i],mod-2);
        //将该方程的xi的系数变为1
        for(int j = i; j<=m; j++)
            f[i][j] = f[i][j]*r%mod;
        //消去其他方程的xi的系数
        for(int j = 1; j<=n; j++)
        {
            if(j != i)
            {
                for(int k = i; k<=m; k++)
                    f[j][k] = (f[j][k] - f[j][i]*f[i][k]%mod + mod)%mod;
            }
        }
    }
        for(int i = 1; i<=n; i++)
        {
            for(int j = n+1; j<=m; j++)
                cout<<f[i][j]<<" ";
            cout<<endl;
        }

    return 0;
}

 

以下是一个用C语言实现高斯消元复数矩阵的示例代码: ```c #include <stdio.h> #include <stdlib.h> #include <math.h> #define MAX_SIZE 100 void print_matrix(int n, double A[][2*MAX_SIZE]) { int i, j; for (i = 0; i < n; i++) { for (j = 0; j < 2*n; j++) { printf("%lf ", A[i][j]); } printf("\n"); } } void swap_rows(int n, double A[][2*MAX_SIZE], int i, int j) { int k; for (k = 0; k < 2*n; k++) { double temp = A[i][k]; A[i][k] = A[j][k]; A[j][k] = temp; } } void scale_row(int n, double A[][2*MAX_SIZE], int i, double scalar) { int j; for (j = 0; j < 2*n; j++) { A[i][j] *= scalar; } } void add_rows(int n, double A[][2*MAX_SIZE], int i, int j, double scalar) { int k; for (k = 0; k < 2*n; k++) { A[i][k] += scalar * A[j][k]; } } int find_pivot(int n, double A[][2*MAX_SIZE], int k) { int i, max_row = k; double max_val = fabs(A[k][k]); for (i = k+1; i < n; i++) { double val = fabs(A[i][k]); if (val > max_val) { max_val = val; max_row = i; } } return max_row; } void gaussian_elimination(int n, double A[][2*MAX_SIZE]) { int i, j, k; for (k = 0; k < n; k++) { int pivot_row = find_pivot(n, A, k); if (pivot_row != k) { swap_rows(n, A, k, pivot_row); } double pivot = A[k][k]; scale_row(n, A, k, 1.0/pivot); for (i = k+1; i < n; i++) { double factor = -A[i][k]; add_rows(n, A, i, k, factor); } } } void backward_substitution(int n, double A[][2*MAX_SIZE]) { int i, j, k; for (k = n-1; k >= 0; k--) { for (i = 0; i < k; i++) { double factor = -A[i][k]; add_rows(n, A, i, k, factor); } } } void invert_matrix(int n, double A[][2*MAX_SIZE]) { int i, j; for (i = 0; i < n; i++) { for (j = n; j < 2*n; j++) { if (i == j-n) { A[i][j] = 1.0; } else { A[i][j] = 0.0; } } } gaussian_elimination(n, A); backward_substitution(n, A); } int main() { int n, i, j; double A[MAX_SIZE][2*MAX_SIZE]; printf("Enter the size of the matrix: "); scanf("%d", &n); printf("Enter the matrix:\n"); for (i = 0; i < n; i++) { for (j = 0; j < n; j++) { printf("A[%d][%d] = ", i, j); scanf("%lf", &A[i][j]); } } invert_matrix(n, A); printf("The inverse matrix is:\n"); for (i = 0; i < n; i++) { for (j = n; j < 2*n; j++) { printf("%lf ", A[i][j]); } printf("\n"); } return 0; } ``` 该程序使用了高斯消元算法,分为三个部分: 1. `gaussian_elimination`:将矩阵进行高斯消元,将其变为上三角矩阵; 2. `backward_substitution`:将上三角矩阵进行回带解,得到矩阵; 3. `invert_matrix`:将原矩阵和单位矩阵合并,进行高斯消元和回带解,得到矩阵。 该程序支持任意复数矩阵矩阵解,其中复数使用一个长度为2的数组表示,第一个元素表示实部,第二个元素表示虚部。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值