【Leetcode 每日一题】2176. 统计数组中相等且可以被整除的数对

问题背景

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

数据约束

• $1\le nums.length\le 100$
• $1\le nums[i],k\le 100$

解题过程

这题数据范围不大，暴力枚举也能过。
一种更好的做法，是将值像等的元素都保存在链表中，遍历时就只需要判断能否整除了。

具体实现

class Solution {public int countPairs(int[] nums, int k) {int n = nums.length;int res = 0;ListNode[] map = new ListNode[110];for (int i = 0; i < n; i++) {ListNode cur = new ListNode(i);ListNode pre = map[nums[i]];if (pre != null) {cur.next = pre;}while (pre != null) {if (i * pre.index % k == 0) {res++;}pre = pre.next;}map[nums[i]] = cur;}return res;}
}class ListNode {int index;ListNode next;ListNode(int index) {this.index = index;next = null;}
}