Skip to content

Instantly share code, notes, and snippets.

@katsuobushiFPGA
Created March 26, 2020 12:54
Show Gist options
  • Save katsuobushiFPGA/f321a3a975e423cb690b84f73310ec69 to your computer and use it in GitHub Desktop.
Save katsuobushiFPGA/f321a3a975e423cb690b84f73310ec69 to your computer and use it in GitHub Desktop.
package main
import (
"bufio"
"fmt"
"math"
"os"
"reflect"
"strconv"
"strings"
// "regexp"
)
type i64 int64
type pair struct {
p1, p2 interface{}
}
const (
initialBufSize = 1e4
maxBufSize = 1e8
INF = 1e8
)
var (
scanner = bufio.NewScanner(os.Stdin)
writer = bufio.NewWriter(os.Stdout)
di = []int64{-1, 0, 1, 0}
dj = []int64{0, -1, 0, 1}
)
func main() {
buf := make([]byte, initialBufSize)
scanner.Buffer(buf, maxBufSize)
scanner.Split(bufio.ScanWords)
R, C := readInt(), readInt()
s := pair{readInt() - 1, readInt() - 1}
g := pair{readInt() - 1, readInt() - 1}
var maze = make([][]string, R)
for i := int64(0); i < R; i++ {
maze[i] = make([]string, C)
maze[i] = readStrings()
}
// mazeの距離
dist := doubleAry(R, C, INF)
// 初期状態
var q []pair
q = append(q, s)
i := s.p1.(int64)
j := s.p2.(int64)
dist[i][j] = 0
dist, q = searchPos(R, C, dist, maze, q)
// 後はぐるぐる回す
for len(q) > 0 {
dist, q = searchPos(R, C, dist, maze, q)
}
gx := g.p1.(int64)
gy := g.p2.(int64)
fmt.Println(dist[gx][gy])
}
// 現在の位置と全体の幅,高さ, 迷路全体を入力とし、探索可能リストを返す
func searchPos(H int64, W int64, dist [][]int64, maze [][]string, q []pair) ([][]int64, []pair) {
x := q[0].p1.(int64)
y := q[0].p2.(int64)
for dir := 0; dir < 4; dir++ {
ni, nj := x+di[dir], y+dj[dir]
if ni < 0 || ni >= H || nj < 0 || nj >= W {
continue
}
if maze[ni][nj] == "#" {
continue
}
// 探索可能リストに入れ, 距離を更新する
if dist[ni][nj] == INF {
dist[ni][nj] = dist[x][y] + 1
q = append(q, pair{ni, nj})
}
}
q = q[1:]
return dist, q
}
/*==========================================
* Library
*==========================================*/
func doubleAry(h int64, w int64, init int64) (res [][]int64) {
res = make([][]int64, h)
for i := int64(0); i < h; i++ {
res[i] = make([]int64, w)
for j := int64(0); j < w; j++ {
res[i][j] = init
}
}
return
}
func aryEq(a []int64, b []int64) bool {
return reflect.DeepEqual(a, b)
}
func clone(n []int64) (r []int64) {
r = make([]int64, len(n))
for i := 0; i < len(n); i++ {
r[i] = n[i]
}
return
}
func write(s string) {
writer.WriteString(s)
}
func print() {
writer.Flush()
}
// scanner.Split(bufio.ScanWords) をコメントアウトしないと使用不可
func readLine() (s string) {
if scanner.Scan() {
s = scanner.Text()
}
return s
}
func readInt() (read int64) {
scanner.Scan()
read, err := strconv.ParseInt(scanner.Text(), 10, 64)
if err != nil {
panic(err)
}
return
}
func readFloat() (read float64) {
scanner.Scan()
read, err := strconv.ParseFloat(scanner.Text(), 64)
if err != nil {
panic(err)
}
return
}
func readRunes() (read []rune) {
scanner.Scan()
for _, v := range scanner.Text() {
read = append(read, v)
}
return
}
func readString() (read string) {
scanner.Scan()
read = scanner.Text()
return
}
func readStrings() (read []string) {
scanner.Scan()
for _, v := range scanner.Text() {
read = append(read, string(v))
}
return
}
func s2i(s string) int64 {
var intVal, e = strconv.ParseInt(s, 10, 64)
if e != nil {
panic(e)
}
return intVal
}
func i2s(i int64) string {
var strVal = strconv.FormatInt(i, 10)
return strVal
}
func s2f(s string) float64 {
var floatVal, e = strconv.ParseFloat(s, 64)
if e != nil {
panic(e)
}
return floatVal
}
func abs(i int64) int64 {
return int64(math.Abs(float64(i)))
}
func sum(i []int64) int64 {
sum := int64(0)
for _, val := range i {
sum += val
}
return sum
}
func split(s string) []string {
return strings.Fields(s)
}
func strAry2intAry(strs []string) []int64 {
var ret = make([]int64, 0, len(strs))
for _, str := range strs {
var intVal = s2i(str)
ret = append(ret, intVal)
}
return ret
}
func intAry2strAry(nums []int64) []string {
var ret = make([]string, 0, len(nums))
for _, num := range nums {
var strVal = i2s(num)
ret = append(ret, strVal)
}
return ret
}
func ary2str(strs []string) string {
return strings.Join(strs, " ")
}
func reverse(res []int64) []int64 {
for i, j := 0, len(res)-1; i < j; i, j = i+1, j-1 {
res[i], res[j] = res[j], res[i]
}
return res
}
func reverseStr(res []string) []string {
for i, j := 0, len(res)-1; i < j; i, j = i+1, j-1 {
res[i], res[j] = res[j], res[i]
}
return res
}
func ini(res []int, init int) {
if len(res) == 0 {
return
}
res[0] = init
for i := 0; i < len(res); i++ {
copy(res[i:], res[:i])
}
}
//
// func regexpExample() {
// // Your code here!
// var str = "13:20"
// r := regexp.MustCompile(`(\d+):(\d+)`)
// fmt.Println(r.FindStringSubmatch(str))
// }
// type Country struct {
// gold int
// silver int
// blonze int
// }
// // 複数ソートのサンプル
// func stableSortExample() []Country{
// var country = []Country {
// {gold:1, silver:2, blonze:3},
// {gold:1, silver:2, blonze:3},
// {gold:1, silver:3, blonze:2},
// {gold:1, silver:3, blonze:3},
// }
// sort.SliceStable(country, func(i, j int) bool { return country[i].blonze > country[j].blonze })
// sort.SliceStable(country, func(i, j int) bool { return country[i].silver > country[j].silver })
// sort.SliceStable(country, func(i, j int) bool { return country[i].gold > country[j].gold })
// return country
// }
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment