ZOJ - 2974 Just Pour the Water

表示十分无语，本想写个矩阵类的。同时重载它的乘法运算之类的。但总是异常退出，把类里的矩阵开小了后就没问题了。不知道为什么！！ 我只开了double[22][22]大小的数组啊 T_T

 

最后还是算了， 直接写吧。 只是代码可重用性不怎么好。

 

 

这里在说下这个矩阵的构造：c[i][j] 表示i杯子向j杯子倒入自己原先c[i][j]分之一的水。

 

 

#include <cstdio> #include <algorithm> using namespace std; const int MAXN = 22; struct Matrix { int w, h; double c[MAXN][MAXN]; }; Matrix a, b, ans, t; void matrix_mul(Matrix& a, Matrix& b, Matrix& t) { t.h = a.h; t.w = b.w; fill(t.c[1], t.c[t.h+1], 0.0); for(int i = 1; i <= a.h; i++) { for(int j = 1; j <= b.w; j++) { for(int k = 1; k <= a.w; k++) { t.c[i][j] += a.c[i][k] * b.c[k][j]; } } } } void matrix_cpy(Matrix& a, Matrix& b) { for(int i = 1; i <= b.h; i++) { for(int j = 1; j <= b.w; j++) { a.c[i][j] = b.c[i][j]; } } } void matrix_qm(Matrix& m, int y) { ans.h = ans.w = m.h; fill(ans.c[1], ans.c[m.h+1], 0.0); for(int i = 1; i <= ans.h; i++) { ans.c[i][i] = 1.0; } while(y > 0) { if(y&1) { matrix_mul(ans, m, t); matrix_cpy(ans, t); } matrix_mul(m, m, t); matrix_cpy(m, t); y /= 2; } matrix_cpy(b, ans); } int main(int argc, char* argv[]) { int T, n, x, m, i, j, k; double tmp; scanf("%d", &T); while(T--) { scanf("%d", &n); fill(a.c[1], a.c[2], 0.0); a.h = 1; a.w = n; for(i = 1; i <= a.w; i++) { scanf("%lf", &a.c[1][i]); } fill(b.c[1], b.c[n+1], 0.0); b.h = b.w = n; for(i = 1; i <= b.h; i++) { scanf("%d", &k); if(k == 0) { b.c[i][i] = 1; continue; } tmp = 1.0/k; for(j = 1; j <= k; j++) { scanf("%d", &x); b.c[i][x] = tmp; } } scanf("%d", &m); matrix_qm(b, m); matrix_mul(a, b, t); matrix_cpy(a, t); printf("%.2lf", a.c[1][1]); for(i = 2; i <= a.w; i++) { printf(" %.2lf", a.c[1][i]); } printf("/n"); } return 0; }