summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--go/queen-attack/queen_attack.go6
1 files changed, 3 insertions, 3 deletions
diff --git a/go/queen-attack/queen_attack.go b/go/queen-attack/queen_attack.go
index 536ae47..28ea34f 100644
--- a/go/queen-attack/queen_attack.go
+++ b/go/queen-attack/queen_attack.go
@@ -23,9 +23,6 @@ func parseLoc(l string) (int, int, error) {
}
func CanQueenAttack(w, b string) (bool, error) {
- if w == b {
- return false, errors.New("same square")
- }
wx, wy, err := parseLoc(w)
if err != nil {
return false, err
@@ -34,5 +31,8 @@ func CanQueenAttack(w, b string) (bool, error) {
if err != nil {
return false, err
}
+ if wx == bx && wy == by {
+ return false, errors.New("same square")
+ }
return wx == bx || wy == by || abs(wx-bx) == abs(wy-by), nil
}