高斯消元模板

2019-08-20 17:03:21 阅读：185 来源： 互联网

//高斯消元 时间复杂度O(n^3) 使用浮点数计算
#include<iostream>
#include<string.h>
#include<math.h>
using namespace std;
const double eps=1e-8;
const int maxn=101;
double a[maxn][maxn];
bool l[maxn];
double ans[maxn];
 
int n,m;
int t;
void print();
//也可以避免实数运算 使用辗转相减的方法 多一个log的时间复杂度
//特别适合于行列式求值 取模操作 (除法要求逆元)
int solve()
{
    //a为方程组对应的矩阵
    //l,ans存储解  l[]表示是否为自由变元 1表示不是 0表示是
    //n为未知数的个数 m为方程的个数
    //如果无解返回-1 否则返回自由变元数
    int res=0,r=0;//r为第几行 res为自由变元数
    for(int i=0; i<n; ++i) l[i]=false; //开始都是自由变元
 
    for(int i=0; i<n; ++i) //枚举列
    {
 
        for(int j=r; j<m; ++j) //枚举行
            if(fabs(a[j][i])>eps)
            {
                //找到当前列下从第r行开始第一个不为零的元素并交换到第r行
                //如果一直为0 则下面会continue res++ 即自由变元+1
                //如果有不为0的 把它调上来（交换行）
                for(int k=i; k<=n; ++k) //第j行和第r行交换 因为a[j][i]!=0
                    swap(a[j][k],a[r][k]);
                break;
            }
//        print();
        //从a[r][i]这一个元素开始 往下的每个元素都是0了 所以这个元素xi是自由变元 i+1跳过即可
        if(fabs(a[r][i])<eps)
        {
            ++res;
            continue;
        }
 
        for(int j=0; j<m; ++j) //j是行数 从第一行（j=0）开始  让上三角更简洁 这样后面求ans就没有必要从下往上了
            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=n-res; i<m; ++i)
    {
        if(fabs(a[i][n])>eps) return -1;
    }
 
 
    //下面求结果
    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;//返回自由变元数
}
void print()
{
    for(int i=0; i<n; i++)
    {
        for(int j=0; j<=n; j++)
        {
            cout<<a[i][j]<<' ';
        }
        cout<<endl;
 
    }
 
}
 
int main()
{
    ios::sync_with_stdio(false);
    cin>>n;
    m=n;
    for(int i=0; i<n; i++)
    {
        for(int j=0; j<=m; j++)
        {
            cin>>a[i][j];
        }
 
    }
    cout<<solve()<<endl;
    print();
 
//    cin>>t;
//    while(t--)
//    {
//        cin>>n;
//        m=n;
//        int ta,tb;
//        memset(a,0,sizeof(a));
//        for(int i=0; i<n; ++i) cin>>a[i][n],a[i][i]=1;
//        for(int i=0; i<n; ++i) cin>>ta,a[i][n]=int(a[i][n])^ta;
//        while(cin>>ta>>tb&&(ta+tb))
//        {
//            a[tb-1][ta-1]=1;
//        }
//        res=solve(a,l,ans,n,m);
//        if(res==-1) cout<<"Oh,it's impossible~!!"<<endl;
//        else cout<<(1<<res)<<endl;
//    }
    return 0;
}

标签：int,res,maxn,高斯消,ans,tb,模板,ta
来源： https://www.cnblogs.com/cherish-lin/p/11383999.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

高斯消元模板