高斯消元模板

#include <iostream>
#include <cstdlib>
#include <algorithm>
#include <cstdio>
#include <string>
#include <cstring>
using namespace std;
#define CL(a,b) memset(a,b,sizeof(a))
#define INF 100000000
const int M(300);
int a[M][M];//方程
int free_x[M],num;//自由变元下标及个数
int x[M];//解
inline int gcd(int n,int m)
{
	if(m==0)
		return n;
	return gcd(m,n%m);
}
inline int LCM(int n,int m)
{
	return n*m/gcd(n,m);
}
int gauss(int equ,int var,int p)
{
	int i,j,k,col;
	int max_r;
	num=0;
	CL(free_x,0);
	CL(x,0);
	for(k=0,col=0;k<equ&&col<var;k++,col++)
	{
		max_r=k;
		for(i=k+1;i<equ;i++)
			if(a[i][col]>a[max_r][col])
				max_r=i;
		if(a[max_r][col]==0)
		{
			k--;
			free_x[num++]=col;
			continue;
		}
		if(k!=max_r)
			for(i=0;i<var+1;i++)
				swap(a[k][i],a[max_r][i]);
		for(i=k+1;i<equ;i++)
		{
			if(a[i][col]!=0)
			{
				int lcm,ta,tb;
				lcm=LCM(abs(a[k][col]),abs(a[i][col]));
				ta=lcm/abs(a[i][col]);
				tb=lcm/abs(a[k][col]);
				if(a[i][col]*a[k][col]<0)
				tb=-tb;
				for(j=col;j<var+1;j++)
				{
					a[i][j]=((a[i][j]*ta-a[k][j]*tb)%p+p)%p;
				}
			}
		}
		
	
	}
	
	//debug
	/*		
	printf("\n");
	for(i=0;i<equ;i++)
		{
			for(j=0;j<var+1;j++)
				printf("%d ",a[i][j]);
			printf("\n");
		}*/
	for(i=k;i<equ;i++)//(0,0,0,...,a)是否存在此种无解情况
	{
		if(a[i][col]!=0)
			return -1;
	}
	//printf("%d\n",num);
        
//var-k 自由变量的个数
	int s=(1<<(var-k)),index,cnt=0,l,ret=INF;
	// 枚举自由变量的状态,来确定解,此处为 mod 2 方程,
	
	/*for(i=0;i<s;i++)
	{
		index=i;cnt=0;
		CL(x,0);
		for(j=0;j<var-k;j++)
		{
			x[free_x[j]]=(index&1);
			if(x[free_x[j]])
				cnt++;
			index>>=1;
		}
		for(j=k-1;j>=0;j--)
		{
			int tmp=a[j][var];
			for(l=j+1;l<var;l++)
			{
				if(a[j][l])
					tmp ^= x[l] ;
			}
			x[j] = tmp ;
			if(x[j])
				cnt++;
		}
	//	printf("%d\n",cnt);
		if(ret>cnt)
			ret=cnt;
	}
	return ret;*/
	if(k<var)
		return var-k;
	for(i=var-1;i>=0;i--)
	{
		int tmp=a[i][var];
		for(j=i+1;j<var;j++)
			tmp=((tmp-a[i][j]*x[j])%p+p)%p;
		while(tmp%a[i][i]!=0)
			tmp+=p;
		x[i]=tmp/a[i][i];
	}
	return 0;

}
int main()
{
	int t,n,i,j,m,p;
	scanf("%d%d",&n,&m);
		
	/*	for(i=0;i<n*n;i++)
		{
			for(j=0;j<n*n+1;j++)
				printf("%d ",a[i][j]);
			printf("\n");
		}*/
	int k=gauss(n,m,p);//n个方程,m个变量 mod p 的方程
	if(k==-1)
		printf("无解\n");
	else if(k>0)	
		printf("无穷解\n");
	else
	{
		for(i=0;i<m;i++)
			printf("%d\n",x[i]);
	}
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值