代码随想录算法训练营第30天|● 332.重新安排行程 ● 51. N皇后 ● 37. 解数独 ● 总结

332. 重新安排行程

困难
相关标签
相关企业
给你一份航线列表 tickets ，其中 tickets[i] = [from(i), to(i)] 表示飞机出发和降落的机场地点。请你对该行程进行重新规划排序。
所有这些机票都属于一个从 JFK（肯尼迪国际机场）出发的先生，所以该行程必须从 JFK 开始。如果存在多种有效的行程，请你按字典排序返回最小的行程组合。

• 例如，行程 [“JFK”, “LGA”] 与 [“JFK”, “LGB”] 相比就更小，排序更靠前。
  假定所有机票至少存在一种合理的行程。且所有的机票 必须都用一次 且 只能用一次。

示例 1：
[图片]
输入：tickets = [[“MUC”,“LHR”],[“JFK”,“MUC”],[“SFO”,“SJC”],[“LHR”,“SFO”]]
输出：[“JFK”,“MUC”,“LHR”,“SFO”,“SJC”]
示例 2：
[图片]
输入：tickets = [[“JFK”,“SFO”],[“JFK”,“ATL”],[“SFO”,“ATL”],[“ATL”,“JFK”],[“ATL”,“SFO”]]
输出：[“JFK”,“ATL”,“JFK”,“SFO”,“ATL”,“SFO”]
解释：另一种有效的行程是 [“JFK”,“SFO”,“ATL”,“JFK”,“ATL”,“SFO”] ，但是它字典排序更大更靠后。

提示：

• 1 <= tickets.length <= 300
• tickets[i].length == 2
• from(i).length == 3
• to(i).length == 3
• from(i) 和 to(i) 由大写英文字母组成
• from(i) != to(i)

思路

• 使用unordered_map<string, map<string, int>> targets，在遍历 unordered_map<出发机场, map<到达机场, 航班次数>> targets的过程中，可以使用"航班次数"这个字段的数字做相应的增减，来标记到达机场是否使用过了。

51. N 皇后

困难
相关标签
相关企业
按照国际象棋的规则，皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。
n 皇后问题 研究的是如何将 n 个皇后放置在 n×n 的棋盘上，并且使皇后彼此之间不能相互攻击。
给你一个整数 n ，返回所有不同的 n 皇后问题 的解决方案。
每一种解法包含一个不同的 n 皇后问题 的棋子放置方案，该方案中 ‘Q’ 和 ‘.’ 分别代表了皇后和空位。

示例 1：
[图片]
输入：n = 4
输出：[[“.Q…”,“…Q”,“Q…”,“…Q.”],[“…Q.”,“Q…”,“…Q”,“.Q…”]]
解释：如上图所示，4 皇后问题存在两个不同的解法。
示例 2：
输入：n = 1
输出：[[“Q”]]

提示：

• 1 <= n <= 9

代码

func solveNQueens(n int) [][]string {
    res := make([][]string, 0)
    path := make([][]string, n)
    for i := 0; i < n; i++ {
        path[i] = make([]string, n)
        for j := 0; j < n; j++ {
            path[i][j] = "."
        }
    }
    dfs_sNQ(n, 0, path, &res)
    return res
}

func dfs_sNQ(n int, row int, path [][]string, res *[][]string) {
    if row == n {
        tmp := make([]string, len(path))
        for i, rowStr := range path {
            tmp[i] = strings.Join(rowStr, "")
        }
        *res = append(*res, tmp)
        return
    }

    for i := 0; i < n; i++ {
        if isValid(row, i, path) {
            path[row][i] = "Q"
            dfs_sNQ(n, row+1, path, res)
            path[row][i] = "."
        }
    }
}

func isValid(row int, col int, path [][]string) bool {
    n := len(path)

    // Check the column
    for i := 0; i < row; i++ {
        if path[i][col] == "Q" {
            return false
        }
    }

    // Check upper-left diagonal
    for i, j := row-1, col-1; i >= 0 && j >= 0; i, j = i-1, j-1 {
        if path[i][j] == "Q" {
            return false
        }
    }

    // Check upper-right diagonal
    for i, j := row-1, col+1; i >= 0 && j < n; i, j = i-1, j+1 {
        if path[i][j] == "Q" {
            return false
        }
    }

    return true
}