nyoj--300 Kiki & Little Kiki 2(矩阵快速幂)

最新推荐文章于 2019-04-28 10:00:07 发布

exmy

最新推荐文章于 2019-04-28 10:00:07 发布

阅读量489

点赞数

分类专栏：数学数学--矩阵快速幂文章标签： nyoj math 矩阵快速幂 easy++

本文链接：https://blog.csdn.net/xushao_Movens/article/details/50974873

版权

数学同时被 2 个专栏收录

19 篇文章 0 订阅

订阅专栏

数学--矩阵快速幂

5 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何运用矩阵快速幂的方法解决nyoj 300题目，通过构造矩阵并进行运算，高效处理了高达10^8次的操作，实现了题目的高效求解。

摘要由CSDN通过智能技术生成

nyoj 300

题解

操作函数： $f(i) = f(i) \text{^} f((i - 1 + n) \% n)$

一共有 $m$ 次操作，而 $m$ 可以达到 $10^8$ 的数量级，所以考虑构造一个矩阵加速运算。
构造矩阵: 以 $n = 4$ 为例
$E = \begin{vmatrix} 1 &1& 0 & 0 \\ 0&1&1&0 \\ 0&0&1&1\\1&0&0&1\end{vmatrix}$
每一列的两个1和操作函数相对应。
这样，每次乘上这个矩阵就是一次操作的结果。
设初始矩阵 $A$ ， $A * E^m$ 就是最后的答案。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;

const int N = 102;

struct mat{
    int r, c;
    int M[N][N];
    mat(int r, int c):r(r), c(c) { memset(M, 0, sizeof(M)); }
};

mat mul(mat& A, mat& B)
{
    mat C(A.r, B.c);
    for(int i = 0; i < A.r; ++i)
        for(int j = 0; j < A.c; ++j)
            if(A.M[i][j]){    //优化，只有state为1的时候才需要改变
                for(int k = 0; k < B.r; ++k)
                    if(B.M[j][k])
                        C.M[i][k] ^= A.M[i][j] & B.M[j][k];
            }
    return C;
}


mat pow(mat& A, int k)
{
    mat B(A.r, A.c);
    for(int i = 0; i < A.r; ++i) B.M[i][i] = 1;

    while(k){
        if(k & 1) B = mul(B, A);
        A = mul(A, A);
        k >>= 1;
    }
    return B;
}

int main()
{
    int m;
    char t[105];

    while(scanf("%d %s", &m, t) != EOF)
    {
        int n = strlen(t);
        mat A(1, n);
        mat T(n, n);
        for(int i = 0; i < n; ++i)
        {
            A.M[0][i] = t[i] - '0';
            T.M[i][i] = T.M[i][(i + 1) % n] = 1;
        }

        T = pow(T, m);
        A = mul(A, T);

        for(int i = 0; i < n; ++i)
        {
            printf("%d", A.M[0][i]);
        }
        printf("\n");
    }

    return 0;
}