找到差值为k的数组中对的数量的最佳方法

时间:2013-11-24 06:34:29

标签: java algorithm search optimization

我正在解决hackerrank的问题。我脑子里有两种方法:

输入:未排序的数组(a)和k

第一种方法:

1)对数组进行排序

2)对于每个数组元素a [i],使用二分搜索找到元素a [i] + K.If发现递增计数并打破内循环。

第二种方法:

1)对数组进行排序

2)对于每个数组元素a [i],使用linearsearch找到元素a [i] + K.If发现递增计数并打破内部循环。

我发现第一种方法更好,因为它将解决n(logn)中的问题。但是当解决方案中存在多个测试用例时,方法2需要较少的时间。有人可以解释一下原因吗?

以下是两种方法的代码:

第一种方法代码:

static int pairs(int[] a,int k) {
  /* Complete this function */
    int temp;
    int len=a.length;
    int count=0;
    int beg;
    int mid;
    int end;
    int midVal;
    Arrays.sort(a);
    for(int i=0;i<len-1;i++){
    temp=a[i]+k;
    beg=i+1;  
    end=len-1;
        for(int l=beg;l<len;l++){
            mid=(beg+end)/2;
            midVal=a[mid];
            if(midVal==temp){
                count++;
                break;
            }
            else if(midVal>temp){
                end=mid-1;
            }
            else{
                beg=mid+1;
            }
        }

    }
    return count;
}

第二种方法代码:

static int pairs(int[] a,int k) {
  /* Complete this function */
    int temp;
    int len=a.length;
    int count=0;
    Arrays.sort(a);
    for(int i=0;i<len;i++){
    temp=a[i];
        for(int j=i+1;j<len;j++){
            if(temp-a[j]==-k){
                count++;
                break;
            }
        }
    }
    return count;
}

6 个答案:

答案 0 :(得分:6)

第一种方法是两者中最好的方法,但有比两者更好的方法: -

这是一个更好的方法的伪代码: -

for(i=0;i<Arr.length;i++) {
  Hashmap.add(Arr[i])
}
count=0;
for(j=0;j<Arr.length;j++) {

  if(Hashmap.containsKey(Arr[j]+k))
     count++;

}

时间复杂度:O(N) 而你的方法= O(NlogN)

修改: -

注意: - 我的方法对于哈希表有额外的空间复杂度O(N),而建议的方法就位。

答案 1 :(得分:0)

在您的第一个代码中,您的内部循环终止条件不正确。如果begend指针相互交叉,它应该终止。

将其更改为while (beg <= end)for (; beg <= end; )

此外,对于第二种方法,不需要对数组进行排序。

答案 2 :(得分:0)

您也可以使用set no need to use hashmap

public static void main(String[] args) throws Exception {

    BufferedReader in = new BufferedReader(new InputStreamReader(System.in));

    int[] firstLine = Stream.of(in.readLine().split(" ")).mapToInt(Integer::parseInt).toArray();
    int count = firstLine[0];
    int diff = firstLine[1];           
    Set<Integer> numbers = new HashSet<Integer>(Stream.of(in.readLine().split(" ")).map(Integer::valueOf).collect(Collectors.toList()));

    int pairCount = 0;
    for (Integer number : numbers) {
        pairCount += (numbers.contains(number+diff) ? 1:0); 
    }
    System.out.println(pairCount);                
}

答案 3 :(得分:0)

private static int CountDistinctPairs(int[] nums, int k) {
    // TODO Auto-generated method stub
    HashMap<Integer, Boolean> map = new HashMap<Integer, Boolean>();
    HashSet<String> set = new HashSet<String>();
    for(int i =0;i < nums.length;i++){
        map.put(nums[i],true);
    }
    for (int i = 0 ; i < nums.length; i++){
        if(map.containsKey(nums[i]+k)){
            String a = "";
            if(nums[i]<nums[i]+k){
                a = "("+nums[i]+","+(nums[i]+k)+")";
            }
            else{
                a = "("+(nums[i] + k)+","+nums[i]+")";
            }
            set.add(a);
        }
    }
    System.out.println(set);
    return set.size();
}

答案 4 :(得分:0)

static boolean checkDuplicatesWithinK(int arr[], int k)
    {
        // Creates an empty hashset
        HashSet<Integer> set = new HashSet<>();

        // Traverse the input array
        for (int i=0; i<arr.length; i++)
        {
            // If already present n hash, then we found 
            // a duplicate within k distance
            if (set.contains(arr[i]))
               return true;

            // Add this item to hashset
            set.add(arr[i]);

            // Remove the k+1 distant item
            if (i >= k)
              set.remove(arr[i-k]);
        }
        return false;
    }

public static void main (String[] args)
    {
        int arr[] = {10, 5, 3, 4, 3, 5, 6};
        if (checkDuplicatesWithinK(arr, 3))
           System.out.println("Yes");
        else
           System.out.println("No");
    }

答案 5 :(得分:0)

我的解决方案

System.out.println(&#34; findPair:&#34; + findPair(new int [] {4,7,1,2,5,3,0,7,8,5,2}, 3));

public static int findPair(int[] nums, int target){ 
    Arrays.sort(nums);

    int count = 0;
    String pairText = "";
    for(int i =0; i < nums.length;i++){
        for(int j =i+1; j < nums.length;j++){

            if(nums[j]-nums[i] == target && (!pairText.equals(nums[i]+", "+nums[j]))){

                //System.out.println(nums[i]+", "+nums[j]);
                pairText = nums[i]+", "+nums[j];
                count++;
            }
        }

    }  


    return count;


}