还是一样的郁闷呢|

gonghr

园龄:4年1个月粉丝:307关注:25

【数据结构与算法】快速幂

快速幂

引入

快速幂是用来解决求幂运算的高效方式。

例如我们要求 x90 次方,一般的方法可以通过一个循环,每次乘一个 x,循环 90 次之后就可以得到答案,时间复杂度为 O(n),效率较低。而通过快速幂,我们可以在 O(log(n)) 的时间复杂度内完成该运算。

具体方法

我们可以通过二进制的视角来看待幂运算。

要计算的是 xn,把 n 以二进制的形式展开。

n=(...b3b2b1b0)2(bi01)

n=b020+b121+b222+b323...=1×b0+2×b1+4×b2+8×b3+...

xn=xb020+b121+b222+b323...=xb020×xb121×xb222×xb323×...

bi01,故{xbi=1(bi=0)xbi=x(bi=1)

所以,只需要使用一个循环求 n 的二进制的每一位,每次一循环中,如果该二进制位为 0,则不需要乘;如果该二进制位为 1,则需要乘 x。且每一次循环中都执行 x *= x,可以一次获取 x 的不同幂次。

代码实现

public static double getPower(double x, int n) {
      if(x == 0) return 0;
      if(n < 0) {     // x^(-a) = (1/x)^a
          x = 1/x;
          n = -n;
      }
      double res = 1.0;
      while(n > 0) {
          if((n & 1) == 1) {
              res *= x;
          }
          x *= x;
          n >>= 1;
      }
      return res;
}

题目

Pow(x, n)

50. Pow(x, n)

实现 pow(x, n) ,即计算 x 的 n 次幂函数(即,xn )。

 

示例 1:

输入:x = 2.00000, n = 10
输出:1024.00000
示例 2:

输入:x = 2.10000, n = 3
输出:9.26100
示例 3:

输入:x = 2.00000, n = -2
输出:0.25000
解释:2-2 = 1/22 = 1/4 = 0.25
 

提示:

-100.0 < x < 100.0
-231 <= n <= 231-1
-104 <= xn <= 104

class Solution {
    public double myPow(double x, int n) {
        long exp = n;              // 特殊处理:补码表示的负数最小值的相反数超过 Integer 表示范围,故提高数据表示范围
        if(x == 0.0) return 0.0; 
        if(n < 0) {
            x = 1/x;
            exp = -exp;
        }
        double res = 1.0;
        while(exp > 0) {
            if((exp & 1) == 1) res *= x;
            x *= x;
            exp >>= 1;
        }
        return res;
    }
}

矩阵快速幂

斐波那契数列

LeetCode 剑指 Offer 10- I. 斐波那契数列

写一个函数,输入 n ,求斐波那契(Fibonacci)数列的第 n 项(即 F(N))。斐波那契数列的定义如下:

F(0) = 0,   F(1) = 1
F(N) = F(N - 1) + F(N - 2), 其中 N > 1.
斐波那契数列由 0 和 1 开始,之后的斐波那契数就是由之前的两数相加而得出。

答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。

 

示例 1:

输入:n = 2
输出:1
示例 2:

输入:n = 5
输出:5
 

提示:

0 <= n <= 100

解:找到一种递推关系,满足矩阵乘法。

f(n) = f(n - 1) + f(n - 2),将其依赖的状态存成列向量

[f(n1)f(n2)]

目标值 f(n) 所在矩阵为:

[f(n)f(n1)]

下面关键就是找到这两个矩阵直接满足的一个关系,知道系数矩阵 mat

{f(n)=1×f(n1)+1×f(n2)f(n1)=1×f(n1)+0×f(n2)

[f(n)f(n1)]=[1110][f(n1)f(n2)]

则令

mat=[1110]

我们就成功找到了系数矩阵。

下面可以求得递推关系式:

[f(n)f(n1)]=[1110][f(n1)f(n2)]=[1110]2[f(n2)f(n3)]=[1110]n1[f(1)f(0)]=matn1×[10]

对于 mat 可以通过快速幂求得结果。

class Solution {
    int mod = (int)1e9+7;
    public int fib(int n) {
        if(n <= 1) return n;
        long[][] mat = new long[][]{
            {1, 1},
            {1, 0}
        };
        long[][] ans = new long[][]{
            {1},
            {0}
        };
        int count =  n - 1;
        while(count > 0) {
            if((count & 1) == 1) ans = mul(mat, ans); // 注意矩阵乘法顺序,不满足交换律
            mat = mul(mat, mat);
            count >>= 1; 
        }
        return (int)(ans[0][0] % mod);
    }
    public long[][] mul(long[][] a, long[][] b) {
        // 矩阵乘法,新矩阵的行数 = a的行数rowa,列数 = b的列数colb
        // a矩阵的列数 = b矩阵的行数 = common
        int rowa = a.length, colb = b[0].length, common = b.length;
        long[][] ans = new long[rowa][colb];
        for (int i = 0; i < rowa; i++) {
            for (int j = 0; j < colb; j++) {
                for (int k = 0; k < common; k++) {
                    ans[i][j] += a[i][k] * b[k][j];
                    ans[i][j] %= mod;
                }
            }
        }
        return ans;
    }
}

第 N 个泰波那契数

LeetCode 1137. 第 N 个泰波那契数

泰波那契序列 Tn 定义如下: 

T0 = 0, T1 = 1, T2 = 1, 且在 n >= 0 的条件下 Tn+3 = Tn + Tn+1 + Tn+2

给你整数 n,请返回第 n 个泰波那契数 Tn 的值。

 

示例 1:

输入:n = 4
输出:4
解释:
T_3 = 0 + 1 + 1 = 2
T_4 = 1 + 1 + 2 = 4
示例 2:

输入:n = 25
输出:1389537
 

提示:

0 <= n <= 37
答案保证是一个 32 位整数,即 answer <= 2^31 - 1。


解:

f(i) 依赖于 f(i - 1)、f(i - 2)、f(i - 3),故将其存为一个列向量

[f(i1)f(i2)f(i3)]

不难得知结果中 f(n) 满足的列向量为

[f(n)f(n1)f(n2)]

下面求系数矩阵,

{f(n)=1f(n1)+1f(n2)+1f(n3)f(n1)=1f(n1)+0f(n2)+0f(n3)f(n2)=0f(n1)+1f(n2)+0f(n3)[f(n)f(n1)f(n2)]=(111100010)[f(n1)f(n2)f(n3)]

mat=(111100010)

变换等式得到

[f(n)f(n1)f(n2)]=(111100010)[f(n1)f(n2)f(n3)]=(111100010)n2[f(2)f(1)f(0)]=mati2[110]

对于 mat 的幂运算可以使用快速幂

class Solution {
    public int tribonacci(int n) {
        if(n == 0) return 0;
        if(n == 1 || n == 2) return 1;
        int[][] mat = new int[][]{
            {1, 1, 1},
            {1, 0, 0},
            {0, 1, 0}
        };
        int[][] ans = new int[][]{
            {1},
            {1},
            {0}
        };
        int count = n - 2;
        while(count > 0) {
            if((count & 1) == 1) ans = mul(mat, ans);
            mat = mul(mat, mat);
            count >>= 1;
        }
        return ans[0][0];
    }
    public int[][] mul(int[][] a, int[][] b) {
        int rowa = a.length;
        int colb = b[0].length;
        int common = b.length;
        int[][] ans = new int[rowa][colb];
        for(int i = 0; i < rowa; i++) {
            for(int j = 0; j < colb; j++) {
                for(int k = 0; k < common; k++) {
                    ans[i][j] += a[i][k] * b[k][j];
                }
            }
        }
        return ans;
    }
}

统计元音字母序列的数目

LeetCode 1220.统计元音字母序列的数目

给你一个整数 n,请你帮忙统计一下我们可以按下述规则形成多少个长度为 n 的字符串:

字符串中的每个字符都应当是小写元音字母('a', 'e', 'i', 'o', 'u')
每个元音 'a' 后面都只能跟着 'e'
每个元音 'e' 后面只能跟着 'a' 或者是 'i'
每个元音 'i' 后面 不能 再跟着另一个 'i'
每个元音 'o' 后面只能跟着 'i' 或者是 'u'
每个元音 'u' 后面只能跟着 'a'
由于答案可能会很大,所以请你返回 模 10^9 + 7 之后的结果。

 

示例 1:

输入:n = 1
输出:5
解释:所有可能的字符串分别是:"a", "e", "i" , "o" 和 "u"。
示例 2:

输入:n = 2
输出:10
解释:所有可能的字符串分别是:"ae", "ea", "ei", "ia", "ie", "io", "iu", "oi", "ou" 和 "ua"。
示例 3:

输入:n = 5
输出:68
 

提示:

1 <= n <= 2 * 10^4

解:题目中给定的字符的下一个字符的规则如下:

字符串中的每个字符都应当是小写元音字母 (‘a’,‘e’,‘i’,‘o’,‘u’);

  • 每个元音 ‘a’ 后面都只能跟着 ‘e’;
  • 每个元音 ‘e’ 后面只能跟着 ‘a’ 或者是 ‘a’;
  • 每个元音 ‘i’ 后面不能再跟着另一个 ‘i’;
  • 每个元音 ‘o’ 后面只能跟着 ‘i’ 或者是 ‘u’;
  • 每个元音 ‘u’ 后面只能跟着 ‘a’;

以上等价于每个字符的前一个字符的规则如下:

  • 元音字母 ‘a’ 前面只能跟着 ‘e’,‘i’,‘u’;
  • 元音字母 ‘e’ 前面只能跟着 ‘a’,‘i’;
  • 每个元音 ‘i’ 前面只能跟着 ‘e’,‘o’;
  • 每个元音 ‘o’ 前面只能跟着 ‘i’;
  • 每个元音 ‘u’ 前面只能跟着 ‘o’,‘i’;

我们设 f[i][j] 代表当前长度为 i 且以字符 j 为结尾的字符串的数目,其中在此 j=0,1,2,3,4 分别代表元音字母 ‘a’,‘e’,‘i’,‘o’,‘u’

{f[i][0]=f[i1][1]+f[i1][2]+f[i1][4]f[i][1]=f[i1][0]+f[i1][2]f[i][2]=f[i1][1]+f[i1][3]f[i][3]=f[i1][2]f[i][4]=f[i1][2]+f[i1][3]

结果为以下列向量的值求和

(f[n][0]f[n][1]f[n][2]f[n][3]f[n][4])

依赖于

(f[n1][0]f[n1][1]f[n1][2]f[n1][3]f[n1][4])

列出递推关系式

(f[n][0]f[n][1]f[n][2]f[n][3]f[n][4])=[0110110100010100010000110](f[n1][0]f[n1][1]f[n1][2]f[n1][3]f[n1][4])=[0110110100010100010000110]n1(f[0][0]f[0][1]f[0][2]f[0][3]f[0][4])

得到系数矩阵

mat=[0110110100010100010000110]

对于系数矩阵使用矩阵快速幂

结果为

ans=i=04f[i][0]

class Solution {
    long mod = 1_000_000_007;
    public int countVowelPermutation(int n) {
        
        long[][] mat =
        {
            {0, 1, 0, 0, 0}, 
            {1, 0, 1, 0, 0}, 
            {1, 1, 0, 1, 1}, 
            {0, 0, 1, 0, 1}, 
            {1, 0, 0, 0, 0}
        };
        long[][] ans = {
            {1},{1},{1},{1},{1}
        };
        int count = n - 1;

        while(count > 0) {
            if((count & 1) == 1) ans = mul(mat, ans);
            mat = mul(mat, mat);
            count >>= 1;
        }
        long res = 0;
        for(int i = 0; i < 5; i++) {
            res += ans[i][0];
        }
        return (int)(res % mod);
    }
    public long[][] mul(long[][] a, long[][] b) {
        int rowa = a.length;
        int colb = b[0].length;
        int common = b.length;
        long[][] ans = new long[rowa][colb];
        for(int i = 0; i < rowa; i++) {
            for(int j = 0; j < colb; j++) {
                for(int k = 0; k < common; k++) {
                    ans[i][j] += a[i][k] * b[k][j];
                    ans[i][j] %= mod;
                }
            }
        }
        return ans;
    }
}

本文作者:GHR

本文链接:https://www.cnblogs.com/gonghr/p/15815126.html

版权声明:本作品采用知识共享署名-非商业性使用-禁止演绎 2.5 中国大陆许可协议进行许可。

posted @   gonghr  阅读(411)  评论(0编辑  收藏  举报
评论
收藏
关注
推荐
深色
回顶
收起
点击右上角即可分享
微信分享提示
  1. 1 Whenever you are ONE OK ROCK
Whenever you are - ONE OK ROCK
00:00 / 00:00
An audio error has occurred.

作词 : TAKA

作曲 : TAKA

I'm telling you

Tonight tonight

You are my angel

爱してるよ

爱してるよ

2人は一つに

Tonight tonight

I just say…

Wherever you are I always make you smile

Wherever you are I'm always by your side

Whatever you say 君を思う気持ち

I promise you forever right now

I don't need a reason

I don't need a reason

I just want you baby

Alright alright

Day after day

この先长いことずっと

この先长いことずっと

どうかこんな仆とずっと

死ぬまで Stay with me

We carry on…

Wherever you are I always make you smile

Wherever you are I'm always by your side

Whatever you say' 君を思う気持ち

I promise you forever right now

Wherever you are I never make you cry

Wherever you are I'never say goodbye

Whatever you say 君を思う気持ち

I promise you forever right now

仆らが出逢った日は2人にとって

一番目の记念すべき日だね

そして今日という日は2人にとって

二番目の记念すべき日だね

心から爱せる人

心から爱せる人

心から爱しい人

この仆の爱の真ん中には

いつも君(きみ)がいるから

Wherever you are I always make you smile

Wherever you are I'm always by your side

Whatever you say 君を思う気持ち

I promise you forever right now

Wherever you are wherever you are

Wherever you are

おわり

おわり

おわり