给你两个下标从 0 开始的整数数组 nums1
和 nums2
,它们分别含有 n
和 m
个元素。请你计算以下两个数值:
answer1
:使得nums1[i]
在nums2
中出现的下标i
的数量。answer2
:使得nums2[i]
在nums1
中出现的下标i
的数量。
返回 [answer1, answer2]
。
示例 1:
输入:nums1 = [2,3,2], nums2 = [1,2]
输出:[2,1]
解释:
示例 2:
输入:nums1 = [4,3,2,3,1], nums2 = [2,2,5,2,3,6]
输出:[3,4]
解释:
nums1
中下标在 1,2,3 的元素在 nums2
中也存在。所以 answer1
为 3。
nums2
中下标在 0,1,3,4 的元素在 nums1
中也存在。所以 answer2
为 4。
示例 3:
输入:nums1 = [3,4,2,3], nums2 = [1,5]
输出:[0,0]
解释:
nums1
和 nums2
中没有相同的数字,所以答案是 [0,0]。
提示:
n == nums1.length
m == nums2.length
1 <= n, m <= 100
1 <= nums1[i], nums2[i] <= 100
问题简要描述:返回两个数组中的公共元素
Java
class Solution {
public int[] findIntersectionValues(int[] nums1, int[] nums2) {
int[] s1 = new int[101];
int[] s2 = new int[101];
for (int x : nums1) {
s1[x] = 1;
}
for (int x : nums2) {
s2[x] = 1;
}
int[] ans = new int[2];
for (int x : nums1) {
ans[0] += s2[x];
}
for (int x : nums2) {
ans[1] += s1[x];
}
return ans;
}
}
Python3
class Solution:
def findIntersectionValues(self, nums1: List[int], nums2: List[int]) -> List[int]:
s1, s2 = set(nums1), set(nums2)
return [sum(x in s2 for x in nums1), sum(x in s1 for x in nums2)]
TypeScript
function findIntersectionValues(nums1: number[], nums2: number[]): number[] {
let s1 = Array(101).fill(0);
let s2 = Array(101).fill(0);
for (const x of nums1) {
s1[x] = 1;
}
for (const x of nums2) {
s2[x] = 1;
}
let ans = Array(2).fill(0);
for (const x of nums1) {
ans[0] += s2[x];
}
for (const x of nums2) {
ans[1] += s1[x];
}
return ans;
};
C++
class Solution {
public:
vector<int> findIntersectionValues(vector<int>& nums1, vector<int>& nums2) {
int s1[101]{};
int s2[102]{};
for (int& x : nums1) {
s1[x] = 1;
}
for (int& x : nums2) {
s2[x] = 1;
}
vector<int> ans(2);
for (int& x : nums1) {
ans[0] += s2[x];
}
for (int& x : nums2) {
ans[1] += s1[x];
}
return ans;
}
};
Go
func findIntersectionValues(nums1 []int, nums2 []int) []int {
s1 := [101]int{}
s2 := [101]int{}
for _, v := range nums1 {
s1[v] = 1
}
for _, v := range nums2 {
s2[v] = 1
}
ans := make([]int, 2)
for _, v := range nums1 {
ans[0] += s2[v]
}
for _, v := range nums2 {
ans[1] += s1[v]
}
return ans
}