golang力扣leetcode 1706.球会落何处

package main
func findBall(grid [][]int) []int {
  n := len(grid[0])
  ans := make([]int, n)
  for j := range ans {
    col := j // 球的初始列
    for _, row := range grid {
      dir := row[col]                             //要么往左要么往右，刚好是+1，-1
      col += dir                                  // 移动球
      if col < 0 || col == n || row[col] != dir { // 到达侧边或 V 形，row[col]是移动后的位置，移动后的与当前的不一样，那么就是v字
        col = -1
        break
      }
    }
    ans[j] = col // col >= 0 为成功到达底部
  }
  return ans
}