题面:Link
题意:
对n(1<=n<=10000)个数的数组每个数有任意两种操作 +K
or-K
,问最后得到这个新数组里最大值减去最小值的最小可能是?
思路:
开始看到最小可能还以为二分(误),后来想枚举一下就可以。
最暴力做法每一位两种可能2^n
显然不行,继续考虑是不用枚举知道那么多信息,我们只要知道(枚举)最后得到新数组里最大值和最小值可能情况。
于是我们先排个序,枚举每一位A[i]+K之后都可能是新数组里最大值情况,那么最小值就可能是A[0]+K,或者是A[i+1]-K。
最后别漏了还有一种情况是A[n-1]-K是最大值情况,需要继续枚举最小极值。
于是具体见C++代码:
class Solution {
public:
int smallestRangeII(vector<int>& A, int K) {
sort(A.begin(),A.end());
int n = A.size();
int maxx = A[n-1], minn = A[0];
int ans = maxx - minn;
for(int i =0;i<n-1;i++){
if(A[i]+K >= maxx-K){
ans = min(ans,A[i]+K-min(minn+K,A[i+1]-K));
}
}
//考虑 A[n-1]-K是最大的情况,接着枚举最小的情况
minn = A[0]+K;
for(auto p:A){
if(maxx-K <= p+K ){
minn = min(minn, p-K);
}
}
ans = min(ans, maxx-K-minn);
return ans;
}
};
还有一种好写但不难想的思路:
我们枚举数组A[i] 位置i
是新数组里+K 出现结束的情况。
比如A = [1,3,6] K=3
假设在位置i= 0
(数组下标从0开始 ) 结束+K,就变成 [4 ,0, 3 ] 。
在位置i=1
结束+K,新数组就变成[4,6,3]。
那么这样,新数组里极小值只会在A[0] + K, A[i+1] -K,极大值在A[i]+K,A[n-1]-K之间产生。
详见用python语言写的代码。
class Solution:
def smallestRangeII(self, A, K):
"""
:type A: List[int]
:type K: int
:rtype: int
"""
A.sort()
ans = A[-1] - A[0]
for index in range(len(A)-1):
now, next = A[index], A[index+1]
maxx = max(now+K, A[-1]-K)
minn = min(A[0]+K, next-K)
ans = min(ans, maxx - minn)
return ans
两种思路由于排序,时间复杂度都是O(nlogn)
。
这题还是有点意思。