HDOJ 2276 - Kiki & Little Kiki 2

Matrix Multiplication (& Quick Power) 

Description

有n盏灯，编号从1到n。

他们绕成一圈，也就是说，1号灯的左边是n号灯。

如果在第t秒的时候，某盏灯左边的灯是亮着的，那么就在第t+1秒的时候改变这盏灯的状态。

输入m和初始灯的状态。

输出m秒后，所有灯的状态。

Type

Matrix Multiplication

Quick Power

Analysis

矩阵乘法和快速幂。

这题的关键就是构造矩阵。

每盏灯的状态下一秒的状态，和他当前的状态，还有左边的灯的状态有关系。

所以我们可以构造出这么一个矩阵：

然后就好办了，矩阵乘法快速幂一下。

但是要注意，由于是灯的开关问题，要mod 2运算，也可以用位运算，效率会高一点。

mod 2用得不好有可能会TLE。

（话说刚开始题目看错，对矩阵元素开了long long，后来忘记改回来，结果居然是TLE了。）

（顺便吐槽一下题目的标题，是鸡鸡和小鸡鸡2吗？）

Solution

// HDOJ 2276
// Kiki & Little Kiki 2
// by A Code Rabbit

#include <cstdio>
#include <cstring>

const int MAXO = 102;

template <typename T>
struct Matrix {
    T e[MAXO][MAXO];
    int o;
    Matrix(int order) { memset(e, 0, sizeof(e)); o = order; }
    Matrix operator*(const Matrix& one) {
        Matrix res(o);
        for (int i = 0; i < o; i++)
            for (int j = 0; j < o; j++)
                for (int k = 0; k < o; k++)
                    res.e[i][j] += e[i][k] * one.e[k][j];
        return res;
    }
    Matrix operator%(int mod) {
        for (int i = 0; i < o; i++)
            for (int j = 0; j < o; j++)
                e[i][j] %= mod;
        return *this;
    }
};

template <typename T>
T QuickPower(T rdx, int exp, int mod) {
    T res = rdx;
    exp--;
    while (exp) {
        if (exp & 1) res = res * rdx % mod;
        exp >>= 1;
        rdx = rdx * rdx % mod;
    }
    return res;
}

int n, m;
char t[MAXO];

int main() {
    while (scanf("%d", &m) != EOF) {
        getchar();
        // Input. 
        gets(t);
        n = strlen(t);
        // Solve.
        Matrix<int> mat_one(n);
        for (int i = 0; i < n; i++)
            for (int j = 0; j < n; j++)
                mat_one.e[i][j] = (i + 1) % n == j || i == j ? 1 : 0;
        Matrix<int> mat_ans = QuickPower(mat_one, m, 2);
        // Compute and output.
        for (int i = 0; i < n; i++) {
            int sum = 0;
            for (int j = 0; j < n; j++)
                sum ^= (t[j] - '0') & mat_ans.e[j][i];
            printf("%d", sum);
        }
        printf("\n");
    }

    return 0;
}