From d941915829f122e717190d3712d1fba7b3fbde32 Mon Sep 17 00:00:00 2001
From: chengleqi <chengleqi5g@hotmail.com>
Date: Mon, 17 Jan 2022 22:07:48 +0800
Subject: [PATCH] =?UTF-8?q?update=200051.N=E7=9A=87=E5=90=8E.md=20Go?=
 =?UTF-8?q?=E7=89=88=E6=9C=AC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 problems/0051.N皇后.md | 109 +++++++++++++++++----------------------
 1 file changed, 48 insertions(+), 61 deletions(-)

diff --git a/problems/0051.N皇后.md b/problems/0051.N皇后.md
index 51415e88..7eb0d7a0 100644
--- a/problems/0051.N皇后.md
+++ b/problems/0051.N皇后.md
@@ -346,69 +346,56 @@ class Solution {
 
 ### Go 
 ```Go
-import "strings"
-var res [][]string
-
-func isValid(board [][]string, row, col int) (res bool){
-    n := len(board)
-    for i:=0; i < row; i++ {
-        if board[i][col] == "Q" {
-            return false
-        }
-    }
-    for i := 0; i < n; i++{
-        if board[row][i] == "Q" {
-            return false
-        }
-    }
-
-    for i ,j := row, col; i >= 0 && j >=0 ; i, j = i - 1, j- 1{
-        if board[i][j] == "Q"{
-            return false
-        }
-    }
-    for i, j := row, col; i >=0 && j < n; i,j = i-1, j+1 {
-        if board[i][j] == "Q" {
-            return false
-        }
-    }
-    return true
-}
-
-func backtrack(board [][]string, row int) {
-    size := len(board)
-    if row == size{
-        temp := make([]string, size)
-        for i := 0; i<size;i++{
-            temp[i] = strings.Join(board[i],"")
-        }
-        res =append(res,temp)
-        return 
-    }
-    for col := 0; col < size; col++ {
-        if !isValid(board, row, col){
-            continue
-        }
-        board[row][col] = "Q"
-        backtrack(board, row+1)
-        board[row][col] = "."
-    }
-}
-
 func solveNQueens(n int) [][]string {
-    res = [][]string{}
-    board := make([][]string, n)
-    for i := 0; i < n; i++{
-        board[i] = make([]string, n)
-    }
-    for i := 0; i < n; i++{
-        for j := 0; j<n;j++{
-            board[i][j] = "."
-        }
-    }
-    backtrack(board, 0)
+    var res [][]string
+	chessboard := make([][]string, n)
+	for i := 0; i < n; i++ {
+		chessboard[i] = make([]string, n)
+	}
+	for i := 0; i < n; i++ {
+		for j := 0; j < n; j++ {
+			chessboard[i][j] = "."
+		}
+	}
+	var backtrack func(int)
+	backtrack = func(row int) {
+		if row == n {
+			temp := make([]string, n)
+			for i, rowStr := range chessboard {
+				temp[i] = strings.Join(rowStr, "")
+			}
+			res = append(res, temp)
+			return
+		}
+		for i := 0; i < n; i++ {
+			if isValid(n, row, i, chessboard) {
+				chessboard[row][i] = "Q"
+				backtrack(row + 1)
+				chessboard[row][i] = "."
+			}
+		}
+	}
+	backtrack(0)
+	return res
+}
 
-    return res
+func isValid(n, row, col int, chessboard [][]string) bool {
+	for i := 0; i < row; i++ {
+		if chessboard[i][col] == "Q" {
+			return false
+		}
+	}
+	for i, j := row-1, col-1; i >= 0 && j >= 0; i, j = i-1, j-1 {
+		if chessboard[i][j] == "Q" {
+			return false
+		}
+	}
+	for i, j := row-1, col+1; i >= 0 && j < n; i, j = i-1, j+1 {
+		if chessboard[i][j] == "Q" {
+			return false
+		}
+	}
+	return true
 }
 ```
 ### Javascript