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;
}