持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第22天，点击查看活动详情

---

2176. 统计数组中相等且可以被整除的数对：

给你一个下标从 0 开始长度为 n 的整数数组 nums 和一个整数 k ，请你返回满足 0 <= i < j < n ，nums[i] == nums[j] 且 (i * j) 能被 k 整除的数对 (i, j) 的 数目 。

样例 1：

输入：
	nums = [3,1,2,2,2,1,3], k = 2
	
输出：
	4
	
解释：
	总共有 4 对数符合所有要求：
	- nums[0] == nums[6] 且 0 * 6 == 0 ，能被 2 整除。
	- nums[2] == nums[3] 且 2 * 3 == 6 ，能被 2 整除。
	- nums[2] == nums[4] 且 2 * 4 == 8 ，能被 2 整除。
	- nums[3] == nums[4] 且 3 * 4 == 12 ，能被 2 整除。

样例 2：

输入：
	nums = [1,2,3,4], k = 1
	
输出：
	0
	
解释：
	由于数组中没有重复数值，所以没有数对 (i,j) 符合所有要求。

提示：

• 1 <= nums.length <= 100
• 1 <= nums[i], k <= 100

---

分析

• 面对这道算法题目，二当家的陷入了沉思。
• 直接双层循环模拟即可。
• 也可以先把相等数字的下标放一起，然后再检查他们的下标是否符合条件；还可以进一步把按照数字分组和统计符合下表条件在一次遍历中一起完成。

---

题解

rust

impl Solution {
    pub fn count_pairs(nums: Vec<i32>, k: i32) -> i32 {
        let mut ans = 0;
        let k = k as usize;
        (0..nums.len() - 1).for_each(|i| {
            (i + 1..nums.len()).for_each(|j| {
                if nums[i] == nums[j]
                    && (i * j) % k == 0 {
                    ans += 1;
                }
            });
        });
        return ans;
    }
}

---

go

func countPairs(nums []int, k int) int {
    ans := 0
    for i := 0; i < len(nums)-1; i++ {
        for j := i + 1; j < len(nums); j++ {
            if nums[i] == nums[j] && (i*j)%k == 0 {
                ans++
            }
        }
    }
    return ans
}

---

typescript

function countPairs(nums: number[], k: number): number {
    let ans = 0;
    for (let i = 0; i < nums.length - 1; ++i) {
        for (let j = i + 1; j < nums.length; ++j) {
            if (nums[i] == nums[j] && (i * j) % k == 0) {
                ++ans;
            }
        }
    }
    return ans;
};

---

python

class Solution:
    def countPairs(self, nums: List[int], k: int) -> int:
        ans = 0
        for i in range(len(nums) - 1):
            for j in range(i + 1, len(nums)):
                if nums[i] == nums[j] and (i * j) % k == 0:
                    ans += 1
        return ans

---

c

int countPairs(int* nums, int numsSize, int k){
    int ans = 0;
    for (int i = 0; i < numsSize - 1; ++i) {
        for (int j = i + 1; j < numsSize; ++j) {
            if (nums[i] == nums[j]
                && (i * j) % k == 0) {
                ++ans;
            }
        }
    }
    return ans;
}

---

c++

class Solution {
public:
    int countPairs(vector<int>& nums, int k) {
        int ans = 0;
        for (int i = 0; i < nums.size() - 1; ++i) {
            for (int j = i + 1; j < nums.size(); ++j) {
                if (nums[i] == nums[j]
                    && (i * j) % k == 0) {
                    ++ans;
                }
            }
        }
        return ans;
    }
};

---

java

class Solution {
    public int countPairs(int[] nums, int k) {
        int ans = 0;
        for (int i = 0; i < nums.length - 1; ++i) {
            for (int j = i + 1; j < nums.length; ++j) {
                if (nums[i] == nums[j]
                        && (i * j) % k == 0) {
                    ++ans;
                }
            }
        }
        return ans;
    }
}

class Solution {
    public int countPairs(int[] nums, int k) {
        List<Integer>[] idxAry = new List[101];
        for (int i = 0; i < nums.length; ++i) {
            int n = nums[i];
            List<Integer> idxList = idxAry[n];
            if (idxList == null) {
                idxList = new ArrayList<>();
                idxAry[n] = idxList;
            }
            idxList.add(i);
        }

        int ans = 0;
        for (List<Integer> idxList : idxAry) {
            if (idxList != null) {
                for (int i = 0; i < idxList.size() - 1; ++i) {
                    for (int j = i + 1; j < idxList.size(); ++j) {
                        if ((idxList.get(i) * idxList.get(j)) % k == 0) {
                            ++ans;
                        }
                    }
                }
            }
        }
        return ans;
    }
}

class Solution {
    public int countPairs(int[] nums, int k) {
        int ans = 0;

        List<Integer>[] idxAry = new List[101];
        for (int i = 0; i < nums.length; ++i) {
            int n = nums[i];
            List<Integer> idxList = idxAry[n];
            if (idxList == null) {
                idxList = new ArrayList<>();
                idxAry[n] = idxList;
            } else {
                for (int idx : idxList) {
                    if ((i * idx) % k == 0) {
                        ++ans;
                    }
                }
            }
            idxList.add(i);
        }

        return ans;
    }
}

---

原题传送门：https://leetcode.cn/problems/count-equal-and-divisible-pairs-in-an-array/

---

非常感谢你阅读本文~
放弃不难，但坚持一定很酷~
希望我们大家都能每天进步一点点~
本文由 二当家的白帽子：https://juejin.cn/user/2771185768884824/posts 博客原创~

---