VIJOS1049 送给圣诞夜的礼品 矩阵乘法

/*
题目大意：顺次给出m个置换，反复使用这m个置换对初始序列进行操作，问k次置换后的序列。m<=10, k<2^31。
首先将这m个置换“合并”起来（算出这m个置换的乘积），
然后接下来我们需要执行这个置换k/m次（取整，若有余数则剩下几步模拟即可）。
注意任意一个置换都可以表示成矩阵的形式。
例如，将1 2 3 4置换为3 1 2 4，相当于下面的矩阵乘法：
[ [ 0 0 1 0]       [1]         [3]
  [ 1 0 0 0]       [2]    =    [1]
  [ 0 1 0 0]    *  [3]         [2]
  [ 0 0 0 1] ]     [4]         [4]
置换k/m次就相当于在前面乘以k/m个这样的矩阵.
我们可以二分计算出该矩阵的k/m次方，再乘以初始序列即可。
做出来了别忙着高兴，得意之时就是你灭亡之日，别忘了最后可能还有几个置换需要模拟。

*/

描述

当小精灵们把贺卡都书写好了之后。礼品准备部的小精灵们已经把所有的礼品都制作好了。可是由于精神消耗的缘故，他们所做的礼品的质量越来越小，也就是说越来越不让圣诞老人很满意。可是这又是没有办法的事情。

于是圣诞老人把礼品准备部的小精灵们聚集起来，说明了自己的看法：“现在你们有n个礼品，其质量也就是降序排列的。那么为了使得这个礼品序列保持平均，不像现在这样很有规律的降序，我这里有一个列表。”
“列表共有m行，这m行都称作操作（不是序列），每一行有n个数字，这些数字互不相同而且每个数字都在1到n之间。一开始，礼品的序列就是现在礼品所处的位置，也就是说，一开始礼品的序列就是1、2、3、4……n；那么然后，我们看列表的第一行操作，设这一行操作的第i个数字为a[i]，那么就把原来序列中的第a[i]个礼物放到现在这个序列的第i的位置上，然后组成新的礼物序列。然后，看列表的第二行操作……、第三行操作……一直到最后一行操作，重复上面的操作。当最后一行的操作结束，组成了的序列又按照第一行来操作，然后第二行操作……第三行操作……一直循环下去，直到一共操作了k行为止。最后生成的这个序列就是我们最终礼品送给孩子们的序列了。大家明白了吗？”
“明白了！”
等圣诞老人一个微笑走后，大家却开始忙碌了。因为m值可能很大很大，而小精灵们的操作速度有限。所以可能在圣诞老人去送礼物之前完成不了这个任务。让他们很是恼火……

格式

输入格式

第一行三个数，n，m和k。

接下来m行，每行n个数。

输出格式

一行，一共n个数，表示最终的礼品序列。n个数之间用一个空格隔开，行尾没有空格，需要回车。

样例1

样例输入1[复制]

7 5 8
6 1 3 7 5 2 4
3 2 4 5 6 7 1
7 1 3 4 5 2 6
5 6 7 3 1 2 4
2 7 3 4 6 1 5

样例输出1[复制]

2 4 6 3 5 1 7

限制

各个测试点1s

提示

1<=n<=100；1<=m<=10；1<=k<=2^31-1。

对于50%的数据，保证k<=500。这些数据每个数据点8分，其他的数据每个数据点12分。

 

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cstdlib>

#define MAXN 110

using namespace std;

struct Matrix
{
    int size;
    long element[MAXN][MAXN];
    void setSize(int);
    void setModulo(int);
    Matrix operator* (Matrix);
    Matrix power(int);

};

void Matrix::setSize(int a)
{
    for (int i=0; i<a; i++)
        for (int j=0; j<a; j++)
            element[i][j]=0;
    size = a;
}


Matrix Matrix::operator* (Matrix param)
{
    Matrix product;
    product.setSize(size);
    for (int i=0; i<size; i++)
        for (int j=0; j<size; j++)
            for (int k=0; k<size; k++)
                product.element[i][j]+=element[i][k]*param.element[k][j];
    return product;
}


Matrix Matrix::power(int exp)
{
    Matrix res,A;
    A=*this;
    res.setSize(size);
    for(int i=0;i<size;i++)
        res.element[i][i]=1;
    while(exp)
    {
        if(exp&1)
            res=res*A;
        exp>>=1;
        A=A*A;
    }
    return res;
}

Matrix ans,a,data[110];
int n,k,m;
long squ[110],tsqu[110];
int main()
{
    while(~scanf("%d%d%d",&n,&m,&k))
    {
        ans.setSize(n);
        for(int i=0;i<n;i++)
            ans.element[i][i]=1,squ[i]=i+1;
        for(int i=0;i<m;i++)
        {
            a.setSize(n);
            for(int j=0;j<n;j++)
            {
                long tmp;
                scanf("%ld",&tmp);
                a.element[j][tmp-1]=1;
            }
            data[i]=a;
            ans=a*ans;
        }
        int mod,div;
        mod=k%m;
        div=k/m;
        ans=ans.power(div);
        for(int i=0;i<n;i++)
        {
            long tmp=0;
            for(int j=0;j<n;j++)
                tmp+=(ans.element[i][j]*squ[j]);
            tsqu[i]=tmp;
        }
        memcpy(squ,tsqu,sizeof(tsqu));
        for(int mm=0;mm<mod;mm++)
        {
            memcpy(squ,tsqu,sizeof(tsqu));
            for(int i=0;i<n;i++)
            {
                long tmp=0;
                for(int j=0;j<n;j++)
                    tmp+=data[mm].element[i][j]*squ[j];
                tsqu[i]=tmp;
            }

        }
        for(int i=0;i<n;i++)
            printf("%ld%s",tsqu[i],i!=n-1?" ":"\n");
    }
    return 0;
}