求陣列中兩個字串的最小距離 Python 版
阿新 • • 發佈:2019-01-03
- 題目:
- 給定一個數組 strs,其中的資料都是字串,給定兩個字串 str1,str2。如果這兩個字串都在 strs陣列中,就返回它們之間的最小距離;如果其中任何一個不在裡面,則返回 -1;如果兩個字串相等,則返回 0。
- 例如:給定[‘*’,’3’,’*’,’5’,’10’,’9’,’7’,’1’,’*’],再給定兩個字串’* ‘和’9’,通過函式求得返回值 3。
- 分析:
- 有兩種方法,
方法1:遍歷陣列 strs,分別記錄兩個 str1 和 str2 的位置。求得最小的一個距離數字。這樣做時間複雜度為 o(n^2)。
方法2:
如果查詢的次數非常多,為了提高查詢的效率,構造Hash表,把每次查詢的時間複雜度下降到 o(1)。
Python 的內建 dict 型別就是雜湊表,實現方法也是hash 表,其查詢的時間複雜度就是 o(1)。雜湊表的構造也分很多種:
比如,構造 Hash 表,key值是strs中的每一個字串,value值是一個hash表,裡面存放著該字串到其它字串的最小距離。
寫成程式碼就是:hash_table = {“*”:{“3”:1, “5”:1, “10”:2, “9”:3, “7”:2, “1”:1}}
當然這種方法的空間複雜度是 o(n^2)
程式碼:
def min_distance_1(strs, str1, str2):
'''
陣列中兩個字串的最小距離,這是方法1,時間複雜度 o(n^2)
:param strs: 給定的陣列中存放有多個字串
:param str1: 第一個字串
:param str2: 第二個字串
:return: 如果其中給定的一個字串不在陣列 strs 中,那麼返回-1,否則返回兩個字串之間的最小間距
'''
if str1 not in strs or str2 not in strs:
return -1
if str1 == str2:
return 0
dist, min = 1, len(strs)
pos1, pos2 = 0, len(strs)
for i in xrange(0, len(strs)):
if str1 == strs[i]:
pos1 = i
for j in xrange(0, len(strs)):
if str2 == strs[j]:
pos2 = j
dist = abs(pos1 - pos2)
if dist < min:
min = dist
return min
def min_distance_2(strs, str1, str2):
'''
陣列中兩個字串的最小距離,這是方法2,如果查詢的次數非常多,把每次查詢的時間複雜度下降到 o(1)。
Python 的內建 dict 型別就是雜湊表,實現方法也是hash 表,其查詢的時間複雜度就是 o(1)。雜湊表的構造也分很多種:
比如,構造 Hash 表,key值是strs中的每一個字串,value值是一個hash表,裡面存放著該字串到其它字串的最小距離。
寫成程式碼就是:hash_table = {"*":{"3":1, "5":1, "10":2, "9":3, "7":2, "1":1}}
當然這種方法的空間複雜度是 o(n^2)
:param strs: 給定的陣列中存放有多個字串['*','3','*','5','10','9','7','1','*']
:param str1: 第一個字串, '*'
:param str2: 第二個字串, '9'
:return: 如果其中給定的一個字串不在陣列 strs 中,那麼返回-1,否則返回兩個字串之間的最小間距
'''
if str1 not in strs or str2 not in strs:
return -1
if str1 == str2:
return 0
def create_hash(strs): # 建立 hash 的過程並不在查詢裡
strs_set = list(set(strs))
dist_hash = {}
for i in xrange(0, len(strs_set)):
temp = {}
for j in xrange(0, len(strs_set)):
if strs_set[i] != strs_set[j]:
dist = min_distance_1(strs, strs_set[i], strs_set[j])
temp[strs_set[j]] = dist
dist_hash[strs_set[i]] = temp
return dist_hash
return create_hash(strs)[str1][str2]