poj2065 SETI

题目描述：

多组数据，每次给出一个模数$p$和一个表示答案的字符串，

相当于给出一个方程组：$$a_1*1^{1}+a_2*1^{2}+……+a_n*1^{n}=f_1$$

$$a_1*2^{1}+a_2*2^{2}+……+a_n*2^{n}=f_2$$

等等共$n$个方程。

求$a_i$。

题解：

高消时搞一下逆元即可。

代码：

#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N = ;
int t,p,n,a[N][N];
char ch[N];
ll fastpow(ll x,int y)
{
    ll ret = ;
    while(y)
    {
        if(y&)ret=ret*x%p;
        x=x*x%p;
        y>>=;
    }
    return ret;
}
void gs()
{
    for(int i=;i<=n;i++)
    {
        int tmp = i;
        for(int j=i+;j<=n;j++)
            if(abs(a[j][i])>abs(a[tmp][i]))tmp=j;
        if(tmp!=i)
            for(int j=i;j<=n+;j++)swap(a[i][j],a[tmp][j]);
        int now=fastpow(a[i][i],p-);
        for(int j=i;j<=n+;j++)a[i][j]=a[i][j]*now%p;
        for(int j=i+;j<=n;j++)
        {
            now = a[j][i];
            for(int k=i;k<=n+;k++)
                a[j][k]=(a[j][k]-now*a[i][k]%p+p)%p;
        }
    }
    for(int i=n;i>=;i--)
        for(int j=i-;j>=;j--)
            a[j][n+]=(a[j][n+]-a[j][i]*a[i][n+]%p+p)%p;
}
int main()
{
    scanf("%d",&t);
    while(t--)
    {
        memset(a,,sizeof(a));
        scanf("%d%s",&p,ch+);
        n = strlen(ch+);
        for(int k=;k<=n;k++)
        {
            int K = ;
            for(int i=;i<=n;i++,K=K*k%p)
                a[k][i]=K;
            if(ch[k]>='a'&&ch[k]<='z')a[k][n+]=ch[k]-'a'+;
        }
        gs();
        for(int i=;i<=n;i++)
            printf("%d ",a[i][n+]);
        puts("");
    }
    return ;
}