lavos/cap3/work9.go

package cap3

// Work9 蚂蚁爬房，求a~b路线数，动规解法，时间O(n)，空间O(1)
func Work9(a, b int) int {
	if a > b {
		return 0
	}

	length := b - a
	if length <= 2 {
		return 1
	}

	// 从a开始到后两个房间路线分别有一个和两种路线
	//a+1:（a->a+1），a+2：（a->a+1->a+2, a->a+2）
	prev, prev2 := 1, 2
	step := 0
	// 前两步已经走过了，所以len-2
	for i := 0; i < length-2; i++ {
		step = prev + prev2

		// 上两个变成上一个，上一个变成下一个
		prev2 = prev
		prev = step
	}

	return step
}