回溯

2019-08-25  本文已影响0人  OOM_Killer

如何理解回溯

回溯可以理解成一个人在前进的过程中有无数个岔路口,经过一个岔路口,又有一个岔路口。每在一个岔路口选择一个道都会影响这个人之后的人生。
有的人在每一个岔路口都能做出十分正确的选择,所以这个人的生活和事业都达到了人生巅峰。而有的人一步,步步错,可能就是它最初的选择的那个岔路口就是错的,导致这个人就一致生活坎坷。
经典的回溯算法解决的问题很多,如八皇后、0-1背包问题、图的着色、旅行商问题、全排列问题。
下面挑几个例子来

八皇后问题

这是简化版的三皇后问题,其实感觉有点像穷举,但是不满足的提前被pass掉了。


三皇后问题
代码实现(golang 版本)
type EightQueue struct {
    Column []int
}

func (eq *EightQueue) CalEightQueues(row int) {
    if row == 8 {   // 行数满了
        fmt.Println(eq.Column)  // 打印 当前每行的Q都应该在哪一列
        eq.PrintQueue()         // 打印图
        return
    }
    for column := 0; column < 8;column ++ {  // 每一行有8种摆放方法
        if eq.IsOk(row,column) {      // 将当前 行,列 代入,查看之前已经放入的Q还是否满足8皇后条件
            eq.Column[row] = column
            eq.CalEightQueues(row+1)  // 考察下一行
        }
    }
}

func (eq *EightQueue) IsOk(row,column int) bool {   // 判断摆放位置是否合适
    leftup  := column - 1 // 左斜线上方(因为下方还没放入呢,所以不需要判断)
    rightup := column + 1 // 右斜线上方

    for i := row -1;i >= 0;i-- {  // 从当前行往第0行遍历
        if eq.Column[i] == column {     // 判断该列在某行(下标) 是否存在Q
            return false
        }
        if leftup >= 0 {
            if eq.Column[i] == leftup { // 判断其左上对角线上 是否存在Q
                return false
            }
        }
        if rightup < 8 {
            if eq.Column[i] == rightup { // 判断其右上对角线上 是否存在Q
                return false
            }
        }
        leftup--   // 左列 - 1
        rightup++  // 右列 + 1
    }
    return true
}

func (eq *EightQueue) PrintQueue() {

    for row := 0; row < 8 ;row++ {
        for column := 0; column < 8; column++ {
            if eq.Column[row] == column {
                fmt.Printf(" Q ")
            }else {
                fmt.Printf(" * ")
            }
        }
        fmt.Printf("\n")
    }
}

执行一下结果

func main() {
    eq := EightQueue.EightQueue{
        Column : make([]int ,8 ),
    }
    eq.CalEightQueues(0)  // 调用时,当然是从第0行开始遍历了
}

执行结果:
[0 4 7 5 2 6 1 3]
 Q  *  *  *  *  *  *  * 
 *  *  *  *  Q  *  *  * 
 *  *  *  *  *  *  *  Q 
 *  *  *  *  *  Q  *  * 
 *  *  Q  *  *  *  *  * 
 *  *  *  *  *  *  Q  * 
 *  Q  *  *  *  *  *  * 
 *  *  *  Q  *  *  *  * 
[0 5 7 2 6 3 1 4]
 Q  *  *  *  *  *  *  * 
 *  *  *  *  *  Q  *  * 
 *  *  *  *  *  *  *  Q 
 *  *  Q  *  *  *  *  * 
 *  *  *  *  *  *  Q  * 
 *  *  *  Q  *  *  *  * 
 *  Q  *  *  *  *  *  * 
 *  *  *  *  Q  *  *  * 
... 太多了,只打印这么点就够了

0-1 背包

0-1 背包的经典解法是动态规划,但是这里还是以回溯来说。

有没有发现,这和八皇后问题十分相似,就是先穷尽,并及时pass掉不符合规定的。

代码实现 (golang 版本)

type Knapsack struct {
    MaxWeight int    // 背包的最大承重
    Items     []int        // 待放入背包中的物品
}

type ItemInKnapsack struct {
    itemIndex  int    // 记录放进背包中的物品的下标
    itemWeight int  // 记录被放进背包中物品的重量
}

func (k *Knapsack) PutItemInKnapsack(CurWeight int,n int,items []ItemInKnapsack) {
    if n >= len(k.Items) {    // 表示 items 中的内容被遍历完了
        fmt.Println("共存放的物品个数",len(items))    // 统计一下装进背包里的item
        fmt.Printf("存放的为")
        sum := 0
        for _,v := range items {
            sum += v.itemWeight
            fmt.Printf(" 下标:%d 重量:%d ",v.itemIndex,v.itemWeight)
        }
        fmt.Printf("\n总重量为%d:",sum)
        fmt.Printf("\n-----\n")
        return
    }

    k.PutItemInKnapsack(CurWeight,n+1,items)    // 第n个物品我不放

    if CurWeight + k.Items[n] <= k.MaxWeight {  // 第n个物品,如果放进去还没达到最大重量,我就放
        items = append(items,ItemInKnapsack{n,k.Items[n]})
        k.PutItemInKnapsack(CurWeight + k.Items[n],n+1,items)
    }
}

执行一下

func main() {
       // 最大承重 10,items 是各个物品
    k := Knapsack.Knapsack{MaxWeight:10,Items:[]int{3,2,5,2,1,2,3,1,4,5,5,10}}
    k.PutItemInKnapsack(0,0,make(map[int]int,10))  // 当前重0,从第0个开始遍历
}

执行结果:
共存放的物品个数 0
存放的为
总重量为0:
-----
共存放的物品个数 1
存放的为 下标:11 重量:10 
总重量为10:
-----
共存放的物品个数 1
存放的为 下标:10 重量:5 
总重量为5:
-----
共存放的物品个数 2
存放的为 下标:9 重量:5  下标:10 重量:5 
总重量为10:
-----
共存放的物品个数 1
存放的为 下标:8 重量:4 
总重量为4:
...
共存放的物品个数 3
存放的为 下标:7 重量:1  下标:8 重量:4  下标:10 重量:5 
总重量为10:
-----
共存放的物品个数 3
存放的为 下标:7 重量:1  下标:8 重量:4  下标:9 重量:5 
总重量为10:
-----
共存放的物品个数 1
存放的为 下标:6 重量:3 
总重量为3:
-----

... 太多了,不全打印了

小结

回溯是什么,说白了,就是穷举,但是这个穷举,会在过程中就淘汰掉不符合规范的组合。

上一篇下一篇

猜你喜欢

热点阅读