天天看點

poj EXTENDED LIGHTS OUT

開關問題是典型的高斯消元問題

不過這道題我一直錯了好久,後來發現在對系數矩陣a進行指派時, 要每次都重新指派, 不然a作為一個全局變量, 上一次計算的結果造成對下一次結果的影響,下面附上代碼

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int a[35][35];
void gauss()
{
 int i, j, k;
 for(k=1; k<=30; k++)
   {
     if(a[k][k] == 0)
      {
        for(i=k+1; i<=30; i++)
         if(a[i][k])
          break;
        for(j=k; j<=31; j++)
         swap(a[i][j], a[k][j]);
      }
    for(i=1; i<=30; i++)
      if(a[i][k] && i!=k)
        for(j=k; j<=31; j++)
          a[i][j] ^= a[k][j];
   }
}
int main()
{
  int t, ca = 1;
  scanf("%d", &t);
  while(t--)
   {
     memset(a, 0, sizeof(a));
     for(int i=1; i<=30; i++)   // 之前一直把這個for循環放在while外面
      {                         // 結果上一次的結果影響了下一次
        a[i][i] = 1;
        if(i%6 != 0)
          a[i][i+1] = 1;
         if(i%6 != 1)
          a[i][i-1] = 1;
         if(i>6)
          a[i][i-6] = 1;
         if(i<25)
         a[i][i+6] = 1;
       }
     for(int i=1; i<=30; i++)
       scanf("%d", &a[i][31]);
     gauss();
     printf("PUZZLE #%d\n", ca++);
     for(int i=1; i<=30; i++)
      {
        printf("%d", a[i][31]);
        if(i%6 == 0)
          printf("\n");
        else
          printf(" ");
      }
   }
  return 0;
}