20170831美团笔试:长度为n的数组中最长的和为K的倍数的连续子数组

xiaoxiao2021-02-28  76

看到这道题,第一个想法是遍历,从n个数求和,再到n-1个数求和(这里需要遍历)。再到1个数,没有k的倍数则输出0。但是一看数量级10^5,O(n^2)的复杂度肯定超了,于是想到了以前项目里用到的滑窗法,暂且这么叫吧。当时老师让做图像积分的同学做的。借鉴下。就是m个数的和,可以求出前m个,在减去最开始的,加上最后面数下一个就是下一个积分和。看图:

具体代码如下:

#include <iostream> using namespace std; int resolution(int *p, int n, int k) { int sum = 0; for (int i = 0; i < n; i++) { sum += p[i]; } if (sum%k == 0) { return n; } for (int i = n-1; i > 0; i--) { sum = 0; for (int ii = 0; ii < i; ii++) { sum += p[ii]; } if (sum%k == 0) return i; for (int j = 0; j < n-i; j++) { sum = sum - p[j] + p[j + i]; if (sum%k == 0) return i; } } return 0; } int main() { int n, k; while (cin >> n) { int *p = new int[n]; for (int i = 0; i < n; i++) { cin >> p[i]; } cin >> k; cout << resolution(p, n, k) << endl; delete[] p; } return 0; }

这里要说的是,我的代码有可以修改的地方,每次改变求和个数时是重新计算的和,其实可以每次从头计算时把结果记下来,然后下次计算用上次的和减去少掉的一个元素,就可以了,复杂度又降低了很多。

转载请注明原文地址: https://www.6miu.com/read-57765.html

最新回复(0)