ZOJ - 2974 | 矩阵快速幂模板

题目描述：n个杯子，每个杯子(i)每轮将自己的水平分给第i1，i2…个杯子，问m轮后各个杯子内水的多少。 n<=20, m<=1e9。

设一开始各个杯子内容量为x[1][n]数组，分配情况为y[n][n]（y[i][j]，表示i分配给j自己容量的占比）。则x *
(y^m)后，一维数组x内就是各个容器内含量。

代码，附模板：

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<set>
#include<map>
#include<list>
#include<stack>
#include<queue>
#include<vector>
#define cl (k<<1)
#define cr (k<<1|1)
#define Mid ((a[k].l+a[k].r)>>1)
#define mem(a,x) memset(a,x,sizeof(a))
using namespace std;
#define bug printf("???\n")

typedef long long LL;

const int Inf=0x3f3f3f3f;
const double eps=1e-7;
const int maxn=25;

struct Mat{
    int row,col;
    double m[25][25];
    friend Mat operator*(const Mat&x,const Mat&y){
        Mat z;
        z.row = x.row;
        z.col = y.col;
        for(int i=1; i<=x.row; i++)
        for(int j=1; j<=y.col; j++){
            z.m[i][j] = 0;
            for(int k=1; k<=x.col; k++)
                z.m[i][j] += x.m[i][k]*y.m[k][j];
        }
        return z;
    }
};

Mat Pow(Mat x,int m){
    Mat ret;
    ret.row = ret.col = x.row;
    for(int i=1; i<=ret.row; i++)
    for(int j=1; j<=ret.row; j++)
        ret.m[i][j] = i==j? 1 : 0;

    while(m){
        if(m&1)
            ret = ret*x;
        x = x*x;
        m = m>>1;
    }
    return ret;
}

int main()
{
    int T;
    cin>>T;
    while(T--){
        int n;
        cin>>n;
        Mat x,y;
        mem(x.m, 0);    //局部变量注意清空
        mem(y.m, 0);
        x.row = 1;
        x.col = n;
        y.row = y.col = n;

        for(int i=1; i<=n; i++){
            scanf("%lf",&x.m[1][i]);
        }
        for(int i=1; i<=n; i++){
            int k;
            scanf("%d",&k);
            if(k==0){y.m[i][i]=1; continue;}
            for(int j=1; j<=k; j++){
                int tmp;
                scanf("%d",&tmp);
                y.m[i][tmp] = 1.0/k;
            }
        }
        int m;
        cin>>m;
        x = x*Pow(y, m);
        for(int i=1; i<=n; i++){
            printf(i<n ? "%.2lf ":"%.2lf\n",x.m[1][i]);
        }
    }
}