玩具套圈

/*
 * 思路：
 * 使用Java Stream API实现同样的逻辑。
 * 对于每个玩具，检查它是否在任意一个圈内。
 * 计算玩具中心到圈中心的距离，如果距离加上玩具半径小于等于圈的半径，则玩具被套中。
 * 如果一个玩具被多个圈套中，只计数一次。
 */

import java.util.*;
import java.util.stream.*;

public class ToyCircleGameStream {
    public int countCoveredToys(int[][] toys, int[][] circles, int r) {
        return (int) IntStream.range(0, toys.length)
                .filter(i -> Arrays.stream(circles)
                        .anyMatch(circle -> {
                            double distance = Math.sqrt(Math.pow(toys[i][0] - circle[0], 2) + Math.pow(toys[i][1] - circle[1], 2));
                            return distance + toys[i][2] <= r;
                        }))
                .distinct()
                .count();
    }

    public static void main(String[] args) {
        ToyCircleGameStream game = new ToyCircleGameStream();
        int[][] toys1 = {{3, 3, 1}, {3, 2, 1}};
        int[][] circles1 = {{4, 3}};
        int r1 = 2;
        System.out.println(game.countCoveredToys(toys1, circles1, r1)); // 输出：1

        int[][] toys2 = {{1, 3, 2}, {4, 3, 1}, {7, 1, 2}};
        int[][] circles2 = {{1, 0}, {3, 3}};
        int r2 = 4;
        System.out.println(game.countCoveredToys(toys2, circles2, r2)); // 输出：2
    }
}