美数对

/*
 * Leetcode 2613: 美数对
 * 题目思路：
 * 给定一个整数数组 nums 和一个整数 k，求满足以下条件的元素对 (i, j) 的数量：
 * 1. 0 <= i < j < nums.length
 * 2. nums[i] == nums[j]
 * 3. (i * j) % k == 0
 */

import java.util.stream.IntStream;

public class BeautifulPairsStream {
    public int countBeautifulPairs(int[] nums, int k) {
        return (int) IntStream.range(0, nums.length)
            .flatMap(i -> IntStream.range(i + 1, nums.length)
                .filter(j -> nums[i] == nums[j] && (i * j) % k == 0))
            .count();
    }

    public static void main(String[] args) {
        BeautifulPairsStream bp = new BeautifulPairsStream();
        int[] nums = {1, 2, 3, 1, 1, 3};
        int k = 2;
        System.out.println(bp.countBeautifulPairs(nums, k)); // Output: 2
    }
}