回溯法之n皇后问题

package main

import (
    "fmt"
    "math"
)

//判断第k行的某一列放置是否合法
func check(col []int, k int) int {
    for i := 0; i < k; i++ {
        if col[i] == col[k] || float64(k - i) == math.Abs(float64(col[k] - col[i])) {//与前部分行同列或者列之差的绝对值与两行之差的绝对值相等
            return 0
        }
    }
    return 1
}

//迭代实现， 思想原理同着色问题
func n_queen(k int) []int{
    var i int
    col := make([]int, k)
    for i = 0; i < k; i++ {
        col[i] = 0
    }
    for i = 0; i >= 0; {
        for col[i] < k {
            col[i]++
            if check(col, i) > 0 {
                i++
            }
            if i == k {
                return col
            }
        }
        col[i] = 0
        i--
    }
    return nil
}

func main() {
    k := 29
    res := n_queen(k)
    if res != nil {
        for i := 0; i < k; i++ {
            fmt.Print(res[i], "\t")
        }
    }
}