zoj 3497 mistwald

咋看起来想图论，其实不是，大概是最短路写太多了吧，一看到地图就来劲。

给出地图，求出是否结点 A 是否可能在第 Q 步时到达 结点 B，矩阵表示图的邻接关系，mat[i][j] 表示有无从 i 到 j 的边，1 有 0 无。

(mat ^ q) [i][j] 表示 i 结点有无可能刚好在第 q 步到达结点 j，1 有 0 无。

以上讨论所谓 A 结点出发，在第 q 步到达结点 B，是允许在达到 B 结点前经过结点 B 的。题目不允许这种情况，只需要去掉结点 B 的所有出边就行了，既在矩阵中把 B 所在行全部设 0 就行了。

需要对矩阵求 Q 次方，Q 很大，不能 1 次方，2 次方，3 次方……一次一次求，把 O(Q) 次乘法降到 O(lgQ) 次，总时间复杂度是 O(n^3 X lgQ)。有 O(n^2) 的矩阵乘法？还没有学习到。n 最大 25，还可以接受。

#include <cstdio> #include <cstdlib> #include <algorithm> #include <cassert> #include <cstring> using namespace std; #undef _DEBUG const int MAX_MAT_SZ = 25; struct Mat { int d[MAX_MAT_SZ][MAX_MAT_SZ]; int sz; const int *operator[](int r) const { assert(r < sz); return d[r]; } int *operator[](int r) { assert(r < sz); return d[r]; } Mat(const Mat &m): sz(m.sz) { memcpy(d, m.d, sizeof(d)); } Mat(int _sz): sz(_sz) { Zero(); } // 零矩阵 void Zero() { memset(d, 0, sizeof(d)); } // 单位矩阵 void One() { memset(d, 0, sizeof(d)); for(int i=0; i<sz; ++i) { d[i][i] = 1; } } Mat &operator=(const Mat &m) { if(&m != this) { memcpy(d, m.d, sizeof(d)); sz = m.sz; } } // 模 2 乘 O(n^3) void Mul(const Mat &m); // 模 2 幂 O(n^3 X lg(p)) void Pow(int p); }; void Mat::Mul(const Mat &m) { assert(m.sz == sz); int c[MAX_MAT_SZ][MAX_MAT_SZ]; for(int i=0; i<sz; ++i) { for(int j=0; j<sz; ++j) { c[i][j] = 0; for(int k=0; k<sz; ++k) { c[i][j] |= (d[i][k] & m.d[k][j]); } } } memcpy(d, c, sizeof(c)); } void Mat::Pow(int p) { Mat t(*this); this->One(); while(p) { if(p & 1) { this->Mul(t); } p >>= 1; t.Mul(t); } } int rows, cols; // 二维 (r, c) 到一维 f 的转化 int f(int r, int c) { return r * cols + c; } int main() { int cs; scanf("%d", &cs); while(cs-- > 0) { scanf("%d%d", &rows, &cols); #ifdef _DEBUG // printf("rows = %d, cols = %d/n", rows, cols); #endif Mat mat(rows * cols); mat.Zero(); for(int r=0; r<rows; ++r) { for(int c=0; c<cols; ++c) { getchar(); int r1, r2, r3, r4, c1, c2, c3, c4; scanf("((%d,%d),(%d,%d),(%d,%d),(%d,%d))", &r1, &c1, &r2, &c2, &r3, &c3, &r4, &c4); --r1, --r2, --r3, --r4, --c1, --c2, --c3, --c4; #ifdef _DEBUG // printf("r1 = %d, c1 = %d, r2 = %d, c2 = %d, r3 = %d, c3 = %d, r4 = %d, c4 = %d/n", // r1, c1, r2, c2, r3, c3, r4, c4); #endif // 最后一行的最后一列不可有出边，忽略之 if(r==rows-1 && c==cols-1) continue; mat[f(r, c)][f(r1, c1)] = 1; mat[f(r, c)][f(r2, c2)] = 1; mat[f(r, c)][f(r3, c3)] = 1; mat[f(r, c)][f(r4, c4)] = 1; } } #ifdef _DEBUG printf("mat ="); mat.Print(); printf("/n"); #endif int lines; scanf("%d", &lines); Mat backup = mat; while(lines-- > 0) { int q; scanf("%d", &q); mat.Pow(q); if(! mat[0][mat.sz - 1]) { printf("False/n"); } else { bool maybe = false; for(int i=0; i<mat.sz-1; ++i) if(mat[0][i]) maybe = true; if(maybe) printf("Maybe/n"); else printf("True/n"); } mat = backup; } printf("/n"); } return 0; }