codility上的问题 （22）
问题描述： 用1 * 1， 1 * 2的矩形覆盖一个n行m列的矩形，问有多少种方法。
数据范围 : n [1..10^6], m [ 1..7]
要求复杂度： 时间 O(log(n) * 8 ^m)) 空间 O(4^m)
分析：这个题跟之前那个木块砌墙问题一样…… 稍作修改即可，又是矩阵乘法。
http://blog.csdn.net/caopengcs/article/details/9928061
代码：
// you can also use includes, for example:
// #include <algorithm>
#include <vector>
vector<vector<int> > a;
const int MOD = 10000007;
int add(int x,int y) {
return ((x += y) >= MOD)?(x - MOD):x;
}
int mul(long long x,long long y) {
return x * y % MOD;
}
vector<vector<int> > mulmatrix(vector<vector<int> > &a,vector<vector<int> > &b) {
vector<vector<int> > c;
int n = a.size(), i ,j, k;
c.resize(n);
for (i = 0; i < n; ++i) {
c[i].resize(n , 0);
for (j = 0; j < n; ++j) {
for (k = 0; k < n; ++k) {
c[i][j] = add(c[i][j], mul(a[i][k],b[k][j]));
}
}
}
return c;
}
void count(int col, int n, int last, int now) {
if (col >= n) {
++a[last][now];
return;
}
count(col + 1, n, last, now);
if (((last & (1 << col)) == 0) && (col + 1 < n) && ((last & (1 << (col + 1))) == 0)) {
count(col + 2, n, last, now | (3 << col));
}
}
int solution(int N, int M) {
// write your code here...
int i,total = 1 << M;
vector<vector<int> > r;
a.resize(total);
r.resize(total);
for (i = 0; i < total; ++i) {
a[i].resize(total, 0);
count(0, M, i, 0);
r[i].resize(total, 0);
r[i][i] = 1;
}
for (; N ; N >>= 1) {
if (N & 1) {
r = mulmatrix(r, a);
}
a = mulmatrix(a, a);
}
return r[0][0];
}