HDU - 3364 Lanterns(高斯消元解方程(取模))
生活随笔
收集整理的這篇文章主要介紹了
HDU - 3364 Lanterns(高斯消元解方程(取模))
小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
題目鏈接:點(diǎn)擊查看
題目大意:給出 n 盞燈和 m 個(gè)開關(guān),每個(gè)開關(guān)可以控制多個(gè)燈,每次詢問給定燈的最終狀態(tài),問有多少種方案可以到達(dá)
題目分析:以開關(guān)為變元,燈為方程列出方程組,每次求出有多少個(gè)自由元,因?yàn)樽杂稍梢蚤_也可以關(guān),所以方案數(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 int inf=0x3f3f3f3f;const int N=55;int maze[N][N]; 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 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;int kase=0;while(w--){int equ,var;memset(maze,0,sizeof(maze));scanf("%d%d",&equ,&var);//燈,開關(guān) for(int i=0;i<var;i++)//每個(gè)開關(guān)控制的燈 {int k;scanf("%d",&k);while(k--){int x;scanf("%d",&x);maze[x-1][i]=true;}}printf("Case %d:\n",++kase);int q;scanf("%d",&q);while(q--){for(int i=0;i<equ;i++)for(int j=0;j<var;j++)a[i][j]=maze[i][j];for(int i=0;i<equ;i++)scanf("%d",&a[i][var]);int ans=Gauss(equ,var,2);if(ans==-1)puts("0");elseprintf("%lld\n",1LL<<ans);}}return 0; }?
總結(jié)
以上是生活随笔為你收集整理的HDU - 3364 Lanterns(高斯消元解方程(取模))的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: POJ - 2065 SETI(高斯消元
- 下一篇: 中石油训练赛 - Switches(高斯