POJ - 2065 SETI(高斯消元解方程(取模))
生活随笔
收集整理的這篇文章主要介紹了
POJ - 2065 SETI(高斯消元解方程(取模))
小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
題目鏈接:點(diǎn)擊查看
題目大意:給出一個(gè)質(zhì)數(shù)作為 mod,再給出一個(gè)字符串,每個(gè)字母對應(yīng)著一個(gè)數(shù)字:
假設(shè)字符串長度為 n,題目給出了 n 個(gè)線性同余方程需要求解
題目分析:因?yàn)槲粗獢?shù)和方程組的個(gè)數(shù)都不是很多,所以直接套上模
//#pragma GCC optimize(2) //#pragma GCC optimize("Ofast","inline","-ffast-math") //#pragma GCC target("avx,sse2,sse3,sse4,mmx") #include<iostream> #include<cstdio> #include<string> #include<ctime> #include<cmath> #include<cstring> #include<algorithm> #include<stack> #include<climits> #include<queue> #include<map> #include<set> #include<sstream> #include<cassert> #include<bitset> using namespace std;typedef long long LL;typedef unsigned long long ull;const LL inf=0x3f3f3f3f3f3f3f3f;const int N=110;int a[N][N];//增廣矩陣 int x[N];//解集 bool free_x[N];//標(biāo)記是否是不確定的變元 inline int gcd(int a,int b) {int t;while(b!=0){t=b;b=a%b;a=t;}return a; } inline int lcm(int a,int b) {return a/gcd(a,b)*b;//先除后乘防溢出 } // 高斯消元法解方程組(Gauss-Jordan elimination).(-2表示有浮點(diǎn)數(shù)解,但無整數(shù)解, //-1表示無解,0表示唯一解,大于0表示無窮解,并返回自由變元的個(gè)數(shù)) //有equ個(gè)方程,var個(gè)變元。增廣矩陣行數(shù)為equ,分別為0到equ-1,列數(shù)為var+1,分別為0到var. int Gauss(int equ,int var,int MOD) {int i,j,k;int max_r;// 當(dāng)前這列絕對值最大的行.int col;//當(dāng)前處理的列int ta,tb;int LCM;int temp;int free_x_num;int free_index;for(int i=0;i<=var;i++){x[i]=0;free_x[i]=true;}//轉(zhuǎn)換為階梯陣.col=0; // 當(dāng)前處理的列for(k = 0;k < equ && col < var;k++,col++){// 枚舉當(dāng)前處理的行. // 找到該col列元素絕對值最大的那行與第k行交換.(為了在除法時(shí)減小誤差)max_r=k;for(i=k+1;i<equ;i++){if(abs(a[i][col])>abs(a[max_r][col])) max_r=i;}if(max_r!=k){// 與第k行交換.for(j=k;j<var+1;j++) swap(a[k][j],a[max_r][j]);}if(a[k][col]==0){// 說明該col列第k行以下全是0了,則處理當(dāng)前行的下一列.k--;continue;}for(i=k+1;i<equ;i++){// 枚舉要?jiǎng)h去的行.if(a[i][col]!=0){LCM = lcm(abs(a[i][col]),abs(a[k][col]));ta = LCM/abs(a[i][col]);tb = LCM/abs(a[k][col]);if(a[i][col]*a[k][col]<0)tb=-tb;//異號(hào)的情況是相加for(j=col;j<var+1;j++){a[i][j] = ((a[i][j]*ta-a[k][j]*tb)%MOD+MOD)%MOD;}}}}// 1. 無解的情況: 化簡的增廣陣中存在(0, 0, ..., a)這樣的行(a != 0).for (i = k; i < equ; i++){ // 對于無窮解來說,如果要判斷哪些是自由變元,那么初等行變換中的交換就會(huì)影響,則要記錄交換.if ( a[i][col] != 0) return -1;}// 2. 無窮解的情況: 在var * (var + 1)的增廣陣中出現(xiàn)(0, 0, ..., 0)這樣的行,即說明沒有形成嚴(yán)格的上三角陣.// 且出現(xiàn)的行數(shù)即為自由變元的個(gè)數(shù).if (k < var){// 首先,自由變元有var - k個(gè),即不確定的變元至少有var - k個(gè).for (i = k - 1; i >= 0; i--){// 第i行一定不會(huì)是(0, 0, ..., 0)的情況,因?yàn)檫@樣的行是在第k行到第equ行.// 同樣,第i行一定不會(huì)是(0, 0, ..., a), a != 0的情況,這樣的無解的.free_x_num = 0; // 用于判斷該行中的不確定的變元的個(gè)數(shù),如果超過1個(gè),則無法求解,它們?nèi)匀粸椴淮_定的變元.for (j = 0; j < var; j++){if (a[i][j] != 0 && free_x[j]) free_x_num++, free_index = j;}if (free_x_num > 1) continue; // 無法求解出確定的變元.// 說明就只有一個(gè)不確定的變元free_index,那么可以求解出該變元,且該變元是確定的.temp = a[i][var];for (j = 0; j < var; j++){if (a[i][j] != 0 && j != free_index) temp -= a[i][j] * x[j]%MOD;temp=(temp%MOD+MOD)%MOD;}while(temp%a[i][free_index]!=0)temp+=MOD;x[free_index] = (temp / a[i][free_index])%MOD; // 求出該變元.free_x[free_index] = 0; // 該變元是確定的.}return var - k; // 自由變元有var - k個(gè).}// 3. 唯一解的情況: 在var * (var + 1)的增廣陣中形成嚴(yán)格的上三角陣.// 計(jì)算出Xn-1, Xn-2 ... X0.for (i = var - 1; i >= 0; i--){temp = a[i][var];for (j = i + 1; j < var; j++){if (a[i][j] != 0) temp -= a[i][j] * x[j];temp=(temp%MOD+MOD)%MOD;}while (temp % a[i][i] != 0) temp+=MOD;x[i] =( temp / a[i][i])%MOD ;}return 0; }int q_pow(int a,int b,int mod) {int ans=1;while(b){if(b&1)ans=ans*a%mod;a=a*a%mod;b>>=1;}return ans; }char s[N];int main() { #ifndef ONLINE_JUDGE // freopen("data.ans.txt","r",stdin); // freopen("data.out.txt","w",stdout); #endif // ios::sync_with_stdio(false);int w;cin>>w;while(w--){int mod;scanf("%d",&mod);scanf("%s",s);int n=strlen(s);for(int i=0;i<n;i++){if(s[i]=='*')a[i][n]=0;elsea[i][n]=s[i]-'a'+1;for(int j=0;j<n;j++)a[i][j]=q_pow(i+1,j,mod);}Gauss(n,n,mod);for(int i=0;i<n-1;i++)printf("%d ",x[i]);printf("%d\n",x[n-1]);}return 0; }板求解即可
代碼:
?
總結(jié)
以上是生活随笔為你收集整理的POJ - 2065 SETI(高斯消元解方程(取模))的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: Gym - 101471D Money
- 下一篇: HDU - 3364 Lanterns(