这道题是求解大小为k,且平均值大于等于阀值的子数组的数目,然后求出均值大于threshold的数组。看到题目的第一感觉是,应该先列出所有长度为k的子数组,尝试了两种方法:
第一种是,求出所有长度为k的数组,分别求出这些子数组的均值,如果大于阈值,进行累加。话不多说,直接上代码。
public int getSubArray(int[] array, int k, int threshold) {
int length = array.length;
int tempSum = 0;
int result = 0;
if (length < k) return 0;
for (int i = 0; i <= array.length - k; ++i) {
int[] tempData = new int[k];
System.arraycopy(array, i, tempData, 0, k);
if (tempData.length == 0) return 0;
for (int m : tempData) {
tempSum += m;
}
if (tempSum / k >= threshold) {
result += 1;
}
tempSum = 0;
}
return result;
}
本地运行ok,于是信心满满的提交了,结果没等到AC,却报了一个运行时间超时,觉得应该是两层for循环嵌套,所以运行时间很慢。所以优化一下,先得到所有的子数组,然后再去计算子数组的均值大于阈值的数组的个数(好像也没优化多少),直接上代码,欢迎各位大佬指正。
public int getSubArray1(int[] array, int k, int threshold) {
int length = array.length;
int tempSum = 0;
int result = 0;
List<int[]> tempList = new ArrayList<>();
if (length < k) return 0;
for (int i = 0; i <= array.length - k; ++i) {
int[] tempData = new int[k];
System.arraycopy(array, i, tempData, 0, k);
if (tempData.length == 0) return 0;
tempList.add(tempData);
}
if (tempList.size() == 0) return 0;
for (int[] temp : tempList) {
for (int n: temp) {
tempSum += n;
}
if (tempSum / k >= threshold) {
result += 1;
}
tempSum = 0;
}
return result;
}
还是运行时间超时,好尴尬,好希望AC啊,无奈,去学习了一下大佬的代码,哇咔咔,好生简洁。
public static int numOfSubarrays3(int[] arr, int k, int threshold) {
int sum = 0, result = 0;
int sumTarget = k*threshold;
for (int i = 0; i < k; i++) {
sum += arr[i];
}
int adder = sum - sumTarget;
if (adder >= 0) {
result++;
}
int pos = k;
for (int i = 0; i < arr.length-k; i++) {
adder = adder + arr[pos] - arr[i];
if (adder >= 0) {
result ++;
}
pos++;
}
return result;
}
先解释一下,当时为了找java如何得到固定长度的子数组,还用到了这个函数 arraycopy(Object src,int srcPos,Object dest, int destPos,int length),感兴趣的童鞋可以去了解一下,整个运行效率好像还ok。
再看大佬的代码,压根没有想到得到子数组,直接逆向思维,如果长度为k的数组阈值为threshold,那么如果从从整体上来看,是不是数组的和值大于k*threshold就ok了,牛皮啊,觉得想的很有意思,瞬间简单了不少啊!之后就是常规的指针一点点往前移动了,直到指针移动到数组的长度减去k,然后呢,通过判断是否大于阈值就ok,好生简单。
不过,用我那种笨办法也有一丢丢好处,就是,如果求出所有的满足条件的子数组,可能用我的笨办法好些吧!
向优秀的大佬们致敬!