目录
题目
给你两个整数数组 nums1 和 nums2 ,请你返回根据以下规则形成的三元组的数目(类型 1 和类型 2 ):
类型 1:三元组 (i, j, k) ,如果 nums1[i]2 == nums2[j] * nums2[k] 其中 0 <= i < nums1.length 且 0 <= j < k < nums2.length
类型 2:三元组 (i, j, k) ,如果 nums2[i]2 == nums1[j] * nums1[k] 其中 0 <= i < nums2.length 且 0 <= j < k < nums1.length
示例 1:
输入:nums1 = [7,4], nums2 = [5,2,8,9]
输出:1
解释:类型 1:(1,1,2), nums1[1]^2 = nums2[1] * nums2[2] (4^2 = 2 * 8)
示例 2:输入:nums1 = [1,1], nums2 = [1,1,1]
输出:9
解释:所有三元组都符合题目要求,因为 1^2 = 1 * 1
类型 1:(0,0,1), (0,0,2), (0,1,2), (1,0,1), (1,0,2), (1,1,2), nums1[i]^2 = nums2[j] * nums2[k]
类型 2:(0,0,1), (1,0,1), (2,0,1), nums2[i]^2 = nums1[j] * nums1[k]
示例 3:输入:nums1 = [7,7,8,3], nums2 = [1,2,9,7]
输出:2
解释:有两个符合题目要求的三元组
类型 1:(3,0,2), nums1[3]^2 = nums2[0] * nums2[2]
类型 2:(3,0,1), nums2[3]^2 = nums1[0] * nums1[1]
方法: 双指针
思路:
可以先对nums2进行排序,后用一个for循环遍历nums1的数,再用双指针j指向nums2头部,k指向nums2的尾部,再用一个while循环通过j自增,k自减的方式来找到符合条件的j和k的位置,此时有两种情况。1:j和k中间的数都是同一个数,数目为m个,则方法数+=(k-j)+(k-(j+1))+(k-(j+2))+...+1=(k-j+1)*(k-j)/2,然后跳出循环,程序结束2:j和k位置上的数不相同,此时可以用while内循环统计和j,k位置上相同的数ct1,ct2,,则方法数+=ct1*ct2,注意,此时不会跳出循环,而是while内循环结束后执行while外循环,不断重复这两种情况直到跳出循环。
代码
class Solution {
public int numTriplets(int[] nums1, int[] nums2) {
return (number(nums1,nums2)+number(nums2,nums1));
}
public static int number(int[] nums1,int[] nums2){
int n1=nums1.length,n2=nums2.length;
Arrays.sort(nums2);
int ans=0;
for(int i=0;i<n1;i++){
long t1=(long)nums1[i]*nums1[i];
int j=0,k=n2-1;
while(j<k){
long t2 = (long)nums2[j]*nums2[k];
if(t1>t2) j++;
else if(t1<t2) k--;
else{
if(nums2[j]==nums2[k]){
ans+=(k-j+1)*(k-j)/2;
break;
}
else{
int l = nums2[j], r = nums2[k];
int ct1 = 0, ct2 = 0;
while(nums2[j] == l)
{
ct1++; j++;
}
while(nums2[k] == r)
{
ct2++; k--;
}
ans+=ct1*ct2;
}
}
}
}
return ans;
}
}
结果
执行结果:
通过
显示详情
查看示例代码
添加备注
执行用时:9 ms, 在所有 Java 提交中击败了88.24%的用户
内存消耗:41.1 MB, 在所有 Java 提交中击败了61.76%的用户
通过测试用例:92 / 92
炫耀一下:
写题解,分享我的解题思路