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

easy

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

1
2
F(0) = 0,   F(1) = 1
F(N) = F(N - 1) + F(N - 2), 其中 N > 1.

斐波那契数列由 0 和 1 开始,之后的斐波那契数就是由之前的两数相加而得出。

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

示例 1:

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

示例 2:

1
2
输入:n = 5
输出:5

还省去了一张 O(n) 的表,妙哉

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
// DP
func fib2(n int) int {
   const mod int = 1e9 + 7
   if n < 2 {
      return n
   }
   a, b, sum := 0, 0, 1
   for i := 2; i <= n; i++ {
      a = b
      b = sum
      sum = (a + b) % mod
   }
   return sum
}
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
func totalFruit(fruits []int) int {
	if len(fruits) == 0 {
		return 0
	}
	dp := make([]int, len(fruits))
	dp[0] = 1
	a, b, aIndex, bIndex := -1, fruits[0], -1, 0
	maxResult := 1
	for i := 1; i < len(fruits); i++ {
		if fruits[i] == a || fruits[i] == b {
			// 加老果子
			if fruits[i] == a {
				aIndex = i
			}
			if fruits[i] == b {
				bIndex = i
			}
			dp[i] = dp[i-1] + 1
		} else {
			// 换新果子
			var length int
			// 计算最近一类果实的长度 并覆盖掉丢弃果子的种类
			if aIndex > bIndex {
				length = aIndex - bIndex
				b = fruits[i]
				bIndex = i
			} else {
				length = bIndex - aIndex
				a = fruits[i]
				aIndex = i
			}
			dp[i] = length + 1
		}
		if maxResult < dp[i] {
			maxResult = dp[i]
		}
	}
	return maxResult
}

滑动窗口