• 周四. 8月 11th, 2022

5G编程聚合网

5G时代下一个聚合的编程学习网

热门标签

P4783-[模板]矩阵求逆

admin

11月 28, 2021

正题

题目链接:https://www.luogu.com.cn/problem/P4783


题目大意

给出一个矩阵,求它的逆矩阵。

(1leq nleq 400)


解题思路

记给出矩阵(P),记单位矩阵(E)

[P imes P^{-1}=ERightarrow P imes (E imes P^{-1})=E
]

虽然看上去上面那个式子是废话,但是这是一个提示。

因为(P)进行初等变化变为(E)的过程中相当于乘上了一个(P^{-1}),而(P^{-1} imes E=P^{-1})。所以如果我们拿一个(E)(P)做一样的初等变化就变为了(P^{-1})

写个高斯消元就好了,但是需要注意因为一般的消元会自动省略已经消掉的部分,但是因这里要处理(P^{-1})矩阵所以不能这么做。

时间复杂度(O(n^3))


code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const ll N=410,P=1e9+7;
ll n,a[N][N],b[N][N];
ll power(ll x,ll b){
	ll ans=1;
	while(b){
		if(b&1)ans=ans*x%P;
		x=x*x%P;b>>=1;
	}
	return ans;
}
bool work(){
	for(ll i=1;i<=n;i++){
		for(ll j=i;j<=n;j++)
			if(a[j][i]){
				if(i!=j)swap(a[j],a[i]),swap(b[j],b[i]);
				break;
			}
		if(!a[i][i])return 1;
		ll inv=power(a[i][i],P-2);
		for(ll j=1;j<=n;j++)
			a[i][j]=a[i][j]*inv%P,b[i][j]=b[i][j]*inv%P;
		for(ll j=i+1;j<=n;j++){
			ll rate=P-a[j][i];
			for(ll k=1;k<=n;k++){
				(a[j][k]+=rate*a[i][k]%P)%=P;
				(b[j][k]+=rate*b[i][k]%P)%=P;
			}
		}
	}
	for(int i=n;i>=1;i--)
		for(int j=1;j<i;j++){
			for(int k=1;k<=n;k++)
				(b[j][k]+=P-a[j][i]*b[i][k]%P)%=P;
			a[j][i]=0;
		}
	return 0;
}
signed main()
{
	scanf("%lld",&n);
	for(ll i=1;i<=n;i++){
		for(ll j=1;j<=n;j++)
			scanf("%lld",&a[i][j]);
		b[i][i]=1;
	}
	if(work())return 0&puts("No Solution");
	for(ll i=1;i<=n;i++,putchar('
'))
		for(ll j=1;j<=n;j++)
			printf("%lld ",b[i][j]);
	return 0;
}

发表评论

您的电子邮箱地址不会被公开。