1. 程式人生 > >2018搜狗春招後端筆試題(部分)

2018搜狗春招後端筆試題(部分)

/**
 * 搜狗2018春招後端程式設計筆試題一:
 * 給定一個大小為n的正整數集合,要求找出這個陣列中差值為k的數值對的個數。
 * 數值對不能重複。
 * 
 * 限制條件:
* 時間限制:C++ 1秒;JAVA 2秒
 * 空間限制:忘了

 */

下面是我的答案,因為迴圈複雜度太高,超出時間限制,所以沒有通過。但是演算法是對的,就記錄下。

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import java.util.Scanner;

public class Main {


public static void main(String[] args) {

                Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
int k = sc.nextInt();
List<Integer> nums = new ArrayList<Integer>();//n個正整數
Map<Integer,Integer> pair = new HashMap<Integer, Integer>(); 
for(int i = 0; i < n; i++){
nums.add(sc.nextInt());
}
int key = 0, value = 0;
for(int i = 0; i < n; i++){
key = nums.get(i);
int find1 = key + k;
int find2 = (key - k) > 0 ? key - k : 0; 
for(int j = i; j < n; j++){
value = nums.get(j);
if(value == find1 || value == find2){
//放入HashMap中,key要比value小
if(key < value)
pair.put(key, value);
else
pair.put(value, key);
break;
}
}
}
System.out.println(pair.size());
}
}