leetcode 2176. Count Equal and Divisible Pairs in an Array(python)

·  阅读 613

「这是我参与2022首次更文挑战的第36天,活动详情查看:2022首次更文挑战

前言

这是 leetcode 中 Biweekly Contest 72 的第一题,难度为 Eazy ,考查的就是对列表的基本操作。其实这次的第 72 场双周赛难度都比较低,我在不到半个小时的时间里都做出来三道题,平时我可是全程下来最多也只能做三道题。

描述

Given a 0-indexed integer array nums of length n and an integer k, return the number of pairs (i, j) where 0 <= i < j < n, such that nums[i] == nums[j] and (i * j) is divisible by k.

Example 1:

Input: nums = [3,1,2,2,2,1,3], k = 2 Output: 4 Explanation: There are 4 pairs that meet all the requirements:

  • nums[0] == nums[6], and 0 * 6 == 0, which is divisible by 2.
  • nums[2] == nums[3], and 2 * 3 == 6, which is divisible by 2.
  • nums[2] == nums[4], and 2 * 4 == 8, which is divisible by 2.
  • nums[3] == nums[4], and 3 * 4 == 12, which is divisible by 2.

Note:

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

解析

根据题意,给定一个长度为 n 的 0 索引整数数组 nums 和一个整数 k,返回 (i, j) 的个数,必须要满足 0 <= i < j < n ,且使得 nums[i] == nums[j] 和 ( i * j) 可被 k 整除。

这题意说的是很清楚了,条件给出的也很明晰,而且限制条件也指出来 nums 的最长的长度为 1000 ,nums 中的元素和 k 的值最大为 100 ,所以基本上是可以使用暴力直接求解的,算是比赛的送分题目吧,这个都做不出来真的完蛋了。

初始化一个结果列表 result ,然后双重 for 循环遍历所有的 nums 中的索引对 (i,j) ,如果满足:

nums[i] == nums[j] and (i*j)%k==0 and [i,j] not in result 
复制代码

就将 (i,j) 加入 result ,遍历结束最后返回 result 的长度即可。

解答

class Solution(object):
    def countPairs(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: int
        """
        result = []
        N = len(nums)
        for i in range(N):
            for j in range(i+1, N):
                if nums[i] == nums[j] and (i*j)%k==0 and [i,j] not in result:
                    result.append([i,j])
        return len(result)
        	      
		
复制代码

运行结果

237 / 237 test cases passed.
Status: Accepted
Runtime: 858 ms
Memory Usage: 13.9 MB
复制代码

原题链接

leetcode.com/contest/biw…

您的支持是我最大的动力

分类:
后端
收藏成功!
已添加到「」, 点击更改