POJ.2065.SETI(高斯消元 模线性方程组)

题目链接

\(Description\)

求\(A_0,A_1,A_2,\cdots,A_{n-1}\)，满足

\[A_0*1^0+A_1*1^1+\ldots+A_{n-1}*1^{n-1}\equiv B[1](mod\ p)
\]

\[A_0*2^0+A_1*2^1+\ldots+A_{n-1}*2^{n-1}\equiv B[2](mod\ p)
\]

\[\ldots\ldots\ldots
\]

\[A_0*n^0+A_1*n^1+\ldots+A_{n-1}*n^{n-1}\equiv B[n](mod\ p)
\]

其中\(B[i]\)为\(str[i-1]\)表示的数字。

\(Solution\)

　　模意义下的高斯消元，在初等行变换时把\(t=tar(A[i][j])/Anow(A[j][j])\)改为\(t=tag*inv(Anow)\)。(也可以在tar不整除Anow时把tar变为它们的lcm，即整行乘\(lcm(tar,Anow)/tar\))

　　最后求解回带时把除法用乘逆元替代即可。(也可用扩展欧几里得求出一个最小的ans[i])

#include <cstdio>
#include <cstring>
#include <algorithm>
#define mod p
typedef long long LL;
const int N=100;

namespace Gauss
{
	int p,n,A[N][N],ans[N];
	char s[N];
	int FP(LL x,int k)
	{
		LL t=1;
		for(;k;k>>=1, x=x*x%p)
			if(k&1) t=t*x%p;
		return t;
	}
	inline int inv(int x) {return FP(x,p-2);}
	void Init()
	{
		scanf("%d%s",&p,s);
		n=strlen(s);
		for(int i=0; i<n; ++i)
		{
			A[i][0]=1;
			for(int j=1; j<n; ++j)
				A[i][j]=(i+1)*A[i][j-1]%p;
			A[i][n]= s[i]=='*'?0:s[i]-'a'+1;
		}
	}
	void Solve()
	{
		Init();
		for(int j=0; j<n; ++j)
		{
			int mxrow=j;
			for(int i=j+1; i<n; ++i)
				if(A[i][j]>A[mxrow][j]) mxrow=i;
			if(mxrow!=j) std::swap(A[mxrow],A[j]);
			for(int i=j+1; i<n; ++i)
				if(A[i][j])
				{
					int t=A[i][j]*inv(A[j][j])%mod;
					for(int k=j; k<=n; ++k)
						A[i][k]=((A[i][k]-t*A[j][k]%mod)%mod+mod)%mod;
				}
		}
		for(int i=n-1; ~i; --i)
		{
			for(int j=i+1; j<n; ++j)
				A[i][n]=((A[i][n]-A[i][j]*ans[j]%mod)+mod)%mod;
			ans[i]=A[i][n]*inv(A[i][i])%mod;
		}
		for(int i=0; i<n-1; ++i) printf("%d ",ans[i]);
		printf("%d\n",ans[n-1]);
	}
}

int main()
{
	int t; scanf("%d",&t);
	while(t--) Gauss::Solve();
	return 0;
}/*
3
31 aaa
37 abc
29 hello*earth
*/