leetcode1577. 数的平方等于两数乘积的方法数(java双指针实现)

最新推荐文章于 2024-09-10 15:28:21 发布

haogeya

最新推荐文章于 2024-09-10 15:28:21 发布

阅读量88

点赞数 2

文章标签：算法

本文链接：https://blog.csdn.net/haogaya/article/details/130183418

版权

题目

给你两个整数数组 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;
    }
}