データ構造とアルゴリズム学習ノート:高速べき乗&行列高速べき乗
3055 ワード
1.高速べき乗時間複雑度:O(logN)
int pow(int a, int n)
{// a^n
int res = 1;
while(n > 0)
{
if(n & 1) //n % 2 == 1
res = res * a;
a = a * a;
n = n >> 1;
}
return res;
}
2.行列の高速べき乗ここの行列は正方行列に対してです.時間複雑度:O(logN)//
const int N = 4; //
void multi(int matrix1[][N], int matrix2[][N])
{
int res[N][N] = {0};
for(int i = 0; i < N; i++)
for(int j = 0; j < N; j++)
for(int k = 0; k < N; k++)
res[i][k] += matrix1[i][j] * matrix2[j][k];
for(int i = 0; i < N; i++)
for(int j = 0; j < N; j++)
matrix1[i][j] = res[i][j];
}
int ans[N][N];
void pow(int matrix[][N], int n)
{//matrix n
memset(ans, 0, sizeof ans);
for(int i = 0; i < N; i++) ans[i][i] = 1;
while(n > 0)
{
if(n & 1)
multi(ans, matrix); // ans[][] * matrix[][]
multi(matrix, matrix);
n = n >> 1;
}
}
問題を解く:転送行列(「行列の等比数列」)をデリバリー式で見つけ、高速べき乗で解いてください.一般的には、f(n)とf(n-1)をF(n)とし、f(n-1)とf(n-2)をF(n-1)としてマトリックスを探します.