題目如下:
解題思路:我的思路是先找出最大值。對于數組中任意一個元素A[i]來說,如果A[i] + K 是B中的最大值,那么意味著從A[i+1]開始的元素都要減去K,即如果有A[i] + K >= A[-1] - K,那么A[i] + K 就可以作為最大值而存在;如果A[i] + K是最大值,那么最大的最小值是多少呢?因為A[i] 左邊的元素都比A[i]小,所以其左邊的元素都可以加上K,最大的最小值就會在 A[0] + K 和 A[i+1] - K 之間產生。遍歷數組,計算每一個A[i] + K 作為最大值的時候最大值和最小值的差值即可。這里還需要考慮一個特殊情況,就是A[-1] - K 作為最大值。如果是這樣情況,再遍歷一次數組,如果A[i] + K <= A[-1] -K,那么最小值就可能是A[i] + K;如果A[i] + K > A[-1] -K,那么最小值可能是A[i] - K 。
代碼如下:
class Solution(object):def smallestRangeII(self, A, K):""":type A: List[int]:type K: int:rtype: int"""A.sort()res = 20000for i in range(len(A)):# suppose A[i] + K is largestif i == len(A) - 1:res = min(res,A[i] - A[0])elif A[i] + K > A[-1] - K:diff = A[i] + K - min(A[i+1] - K,A[0] + K)res = min(res,diff)#A[-1] - K is largestmaxv = A[-1] - Kminv = A[0] + Kfor i in range(len(A)-1):if A[i] + K <= maxv:continueelse:minv = min(A[i] - K ,minv)res = min(res,maxv-minv)return res if res != 20000 else 0
?