每日一练--39玩具套圈

力扣

//计算位置x1,y1半径r1的玩具是否在x2,y2处半径位r2的圆圈内
func canGetToy(x1, y1, r1, x2, y2, r2 int) bool {
	//两圆圆心连线距离
	length := math.Sqrt(float64((x2-x1)*(x2-x1)) + float64((y1-y2)*(y1-y2)))
	return length+float64(r1) <= float64(r2)
}

//套圈游戏
func circleGame(toys [][]int, circles [][]int, r int) int {
	var toynum int
	//减少循环次数
	//所有圆圈的半径都是r
	//[[4,3]]
	//xi 小于圆右边
	ciclemap := make(map[int][][]int)
	for ci := 0; ci < len(circles); ci++ {
        //玩具半径最小1
		for i := circles[ci][0] - r + 1; i <= circles[ci][0]+r-1; i++ {
			ciclemap[i] = append(ciclemap[i], circles[ci])
		}
	}

	for i := 0; i < len(toys); i++ {
		//玩具半径大于圈跳过
		if toys[i][2] > r{
			continue
		}
		//获取玩具可能对应的circle
		toycircles:=ciclemap[toys[i][0]]
		for j:=0;j<len(toycircles);j++{
			if canGetToy(toys[i][0], toys[i][1], toys[i][2], toycircles[j][0], toycircles[j][1], r) {
				toynum++
                break
			}
		}
	}
	return toynum
}