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
*/

转载于:https://www.cnblogs.com/SovietPower/p/8446229.html