算法leetcode2176. 统计数组中相等且可以被整除的数对（rust和go重拳出击）

Posted 2022-08-11 二当家的白帽子

---

文章目录

• 2176. 统计数组中相等且可以被整除的数对：
• 样例 1：
• 样例 2：
• 提示：
• 分析
• 题解
• • rust
  • go
  • typescript
  • python
  • c
  • c++
  • java
• 原题传送门：https://leetcode.cn/problems/count-equal-and-divisible-pairs-in-an-array/

---

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://le-yi.blog.csdn.net/ 博客原创~

---