标签:blog http io ar os sp for on div
题目链接:BZOJ - 1875
题目分析:
这道题如果去掉“不会立刻沿着刚刚走来的路走回”的限制,直接用邻接矩阵跑矩阵乘法就可以了。然而现在加了这个限制,建图的方式就要做一些改变。如果我们把每一条边看做点建矩阵,那么每次从一条边出发都只会到其他的边,不能仍然在这条边上“停留”,所以这就可以满足题目的限制。将每条边拆成两条单向边,比如一条编号为 4,一条编号为 5。那么 4^1=5, 5^1=4。这样只要不从第 i 条边走到 i 或 i^1 就可以了。初始的矩阵中以 A 为起点的边到达的方案数为 1 ,其余为 0。最后将终点为 B 的边的方案数累加即为答案。、
这种将边与点灵活转化的思想十分巧妙,应注意。
代码如下:
#include <iostream> #include <cstdio> #include <cstdlib> #include <cstring> #include <cmath> #include <algorithm> using namespace std; const int MaxN = 20 + 5, MaxM = 120 + 5, Mod = 45989; int n, m, t, A, B, TopA, TopB, Index, RT, Ans, a, b; int EA[MaxM], EB[MaxM]; struct Edge { int u, v; Edge() {} Edge(int a, int b) { u = a; v = b; } } E[MaxM]; struct Matrix { int x, y, Num[MaxM][MaxM]; void SetXY(int xx, int yy) { x = xx; y = yy; } void Clear(int nn) { for (int i = 0; i < x; ++i) { for (int j = 0; j < y; ++j) { Num[i][j] = nn; } } } } M0, MZ; Matrix Mul(Matrix A, Matrix B) { Matrix ret; ret.SetXY(A.x, B.y); ret.Clear(0); for (int i = 0; i < ret.x; ++i) { for (int j = 0; j < ret.y; ++j) { for (int k = 0; k < A.y; ++k) { ret.Num[i][j] += A.Num[i][k] * B.Num[k][j]; ret.Num[i][j] %= Mod; } } } return ret; } Matrix Pow(Matrix A, int b) { Matrix ret, f; f = A; ret.SetXY(f.x, f.y); for (int i = 0; i <= ret.x; ++i) ret.Num[i][i] = 1; while (b) { if (b & 1) ret = Mul(ret, f); b >>= 1; f = Mul(f, f); } return ret; } int main() { scanf("%d%d%d%d%d", &n, &m, &t, &A, &B); Index = -1; for (int i = 1; i <= m; ++i) { scanf("%d%d", &a, &b); E[++Index] = Edge(a, b); E[++Index] = Edge(b, a); } MZ.SetXY(m * 2, m * 2); MZ.Clear(0); TopA = TopB = 0; for (int i = 0; i <= Index; ++i) { if (E[i].u == A) EA[++TopA] = i; if (E[i].v == B) EB[++TopB] = i; for (int j = 0; j <= Index; ++j) { if (i != j && i != (j ^ 1) && E[i].v == E[j].u) MZ.Num[i][j] = 1; } } M0.SetXY(1, m * 2); M0.Clear(0); for (int i = 1; i <= TopA; ++i) M0.Num[0][EA[i]] = 1; MZ = Pow(MZ, t - 1); M0 = Mul(M0, MZ); Ans = 0; for (int i = 1; i <= TopB; ++i) { Ans += M0.Num[0][EB[i]]; Ans %= Mod; } printf("%d\n", Ans); return 0; }
[BZOJ 1875] [SDOI 2009] HH去散步【矩阵乘法】
标签:blog http io ar os sp for on div
原文地址:http://www.cnblogs.com/F-Magician/p/4158566.html