[POJ2065]SETI（高斯消元）

题目描述

传送门
题意：
良心博主才会给出题意…
直接说方程是什么吧…
首先a…z=1..26，*=0
读入p（模数且为质数），s（下标从0开始），s长度为n
那么求方程组
$1^0x_1+1^1x_2...1^{n-1}x_n\equiv s_1\pmod p$
$2^0x_1+2^1x_2...2^{n-1}x_n\equiv s_2\pmod p$
$3^0x_1+3^1x_2...3^{n-1}x_n\equiv s_3\pmod p$
…
$n^0x_1+n^1x_2...n^{n-1}x_n\equiv s_n\pmod p$
的一组合法解
没有无解或多解的情况

题解

高斯消元求同余方程裸题
就是普通的高斯消元把除法变成乘逆元

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
#define N 100

char s[N];
int T,Mod,n,K;
int a[N][N],b[N],ans[N];

void clear()
{
    Mod=n=K=0;
    memset(a,0,sizeof(a));memset(b,0,sizeof(b));memset(ans,0,sizeof(ans));
}
int fast_pow(int x,int p)
{
    int ans=1;
    for (;p;p>>=1,x=x*x%Mod)
        if (p&1)
            ans=ans*x%Mod;
    return ans;
}
int inv(int x)
{
    return fast_pow(x,Mod-2);
}
void gauss()
{
    for (int i=1;i<=n;++i)
    {
        int num=i;
        for (int j=i+1;j<=n;++j)
            if (a[j][i]>a[num][j]) num=j;
        if (num!=i)
        {
            for (int j=1;j<=n;++j) swap(a[i][j],a[num][j]);
            swap(b[i],b[num]);
        }
        for (int j=i+1;j<=n;++j)
        {
            int t=a[j][i]*inv(a[i][i])%Mod;
            for (int k=1;k<=n;++k)
                a[j][k]=((a[j][k]-a[i][k]*t%Mod)%Mod+Mod)%Mod;
            b[j]=((b[j]-b[i]*t)%Mod+Mod)%Mod;
        }
    }
    for (int i=n;i>=1;--i)
    {
        for (int j=i+1;j<=n;++j)
            b[i]=((b[i]-a[i][j]*ans[j]%Mod)%Mod+Mod)%Mod;
        ans[i]=b[i]*inv(a[i][i])%Mod;
    }
}
int main()
{
    scanf("%d",&T);
    while (T--)
    {
        clear();
        scanf("%d",&Mod);
        scanf("%s",s);n=strlen(s);
        for (int i=1;i<=n;++i)
        {
            ++K;a[i][1]=1;
            for (int j=2;j<=n;++j)
                a[i][j]=a[i][j-1]*K%Mod;
            if (s[i-1]!='*') b[i]=s[i-1]-'a'+1;
        }
        gauss();
        for (int i=1;i<=n;++i) printf("%d%c",ans[i]," \n"[i==n]);
    }
}